Pytest fixtures for tests involving databases managed by alembic migrations
Project description
Description
Pytest fixture to simplify writing tests against databases managed with alembic
.
Before each test run, alembic migrations apply schema changes which then allows tests to only care about data.
This way your application code, and the database migrations get executed by the test.
Only tested with PosgreSQL. However, code may work with other databases as well.
Installation
Install with pip install alembic-migration-fixtures
or any other dependency manager.
Afterwards, create a pytest fixture called database_engine
returning an SQLAlchemy Engine
instance.
WARNING
Do not specify the production / development / any other database where data is important in the engine fixture. If you do so, the tests WILL truncate all tables and data loss WILL occur.
Usage
This library provides a pytest fixture called test_db_session
.
Use this to replace the normal SQLAlchemy session used within the application, or else tests may not be independent
of one another.
How the fixture works with your tests:
- Fixture recreates (wipes) the database schema based on the engine provided for the test session
- Fixture runs alembic migrations (equivalent to
alembic upgrade heads
) - Fixture creates a test database session within a transaction for the test
- Your test sets up data and runs the test using the session (including
COMMIT
ing transactions) - Your test verifies data is in the database
- Fixture rolls back the transaction (and any inner
COMMIT
ed transactions in the test)
This two-level transaction strategy makes it so any test is independent of one another, since the database is empty after each test. Since the database schema only gets re-created once per session, the test speed is only linearly dependent on the number of migrations.
Development
This library uses the poetry package manager, which has to be installed before installing
other dependencies. Afterwards, run poetry install
to create a virtualenv and install all dependencies.
To then activate that environment, use poetry shell
. To run a command in the environment without activating it,
use poetry run <command>
.
Black is used (and enforced via workflows) to format all code. Poetry will install it
automatically, but running it is up to the user. To format the entire project, run black .
inside the virtualenv.
Contributing
This project uses the Apache 2.0 license and is maintained by the data science team @ Barbora. All contribution are welcome in the form of PRs or raised issues.
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 alembic_migration_fixtures-0.2.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 24cdf4bf9fc1aa6532ca98ce466240f89351015002b058c0487c63e94cb1a521 |
|
MD5 | 32869096c62e4fbe3b59fde83a22492b |
|
BLAKE2b-256 | d3b7ead52e216f43bd01262a4e9368eb3d709e471db1a3b72e737c78437d830d |
Hashes for alembic_migration_fixtures-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a93faf52b986613239e4e3f9cea0455cac4bf6e42a0822581395a8a866efd5c |
|
MD5 | 5df7528d8942a30d12c878a90f76d923 |
|
BLAKE2b-256 | b4e589b326a871ee68d40720b5b3bf27b2f22a2a6e677bac25acecdbcadec414 |