No project description provided
Project description
PG Docker
A python package to provide containerized postgres databases in python
Why would you need this?
If you are using postgres and want to write tests that run against a real database, then this will make your life easier.
Installation
Install via pip:
pip install pg-docker
You will also need to have docker.
Usage
Note: This package is mainly built with pytest in mind, but you can use the context managers documented below with other testing frameworks as well.
Example
With pytest:
import psycopg2
pytest_plugins = ["pg_docker"]
def test_using_a_database(pg_database):
db_connection = psycopg2.connect(**pg_database.connection_kwargs())
cursor = db_connection.cursor()
cursor.execute("SELECT 'hello world!'")
assert cursor.fetchone() == ("hello world!",)
Usage with pytest
You first need to enable the plugin. To do this add a conftest.py
to the root directory of your tests and add:
pytest_plugins = ["pg_docker"]
You can find more details on how to activate plugins in the pytest docs
The plugin The following fixtures:
pg_database
:DatabaseParams
for a clean database.pg_database_pool
: ADatabasePool
instance. Use this if you need more than one database in your tests at a time.
Configuring Database Migrations
Use the below template in your conftest.py
to configure how your databases are set up.
def setup_db(pg_params):
"""Add any setup logic for your database in here."""
pass
@pytest.fixture(scope="session")
def pg_setup_db():
return setup_db
Note: You might be inclined to edit the above code to nest the setup_db function inside of the fixture function. This will not work, because the fixture result needs to be pickleable!
Advanced Usage (and other testing frameworks)
For other use cases you can use the database_pool
context manager:
with database_pool() as db_pool:
with db_pool.database as db_params:
connection = psycopg2.connect(**db_params.connection_kwargs())
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for pg_docker-0.4.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fc24b3a8559e6abd583a0911547b81178b77b0a02887c8cb6e757a32d909195a |
|
MD5 | 8dc948bccf1e04bf53b52e44f0f2c57c |
|
BLAKE2b-256 | 78c5dfae50d5c21ee015947abdcb4cca6a0c2a51f61b7dcb08c02f67a78112c9 |