ephemeral-postgres is a python package for easily setting up, testing, and finally tearing down postgres instances. It can be easily integrated in most test suites but has been primarily used with pytest.
pip install ephemeral-postgres
import ephemeral_postgres
import psycopg2
with ephemeral_postgres.postgres() as uri:
con = psycopg2.connect(uri)
cur = con.cursor()
cur.execute('select * from pg_database')
for row in cur.fetchall():
print(row)
Alternatively, if you would like to run an external Postgres server, you can
override ephemeral_postgres from starting a new instance by setting the
environment variable EPHEMERAL_POSTGRES_URI
accordingly. For example:
export EPHEMERAL_POSTGRES_URI="postgresql://postgres:postgres@localhost:5432/postgres"
python "import ephemeral_postgres; print(ephemeral_postgres.postgres()[0])"
pip install -e .[test]
# run unit tests
pytest