Skip to main content

Pytest fixtures for tests involving databases managed by alembic migrations

Project description

Formatter PyPI version PyPI - Python Version

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:

  1. Fixture recreates (wipes) the database schema based on the engine provided for the test session
  2. Fixture runs alembic migrations (equivalent to alembic upgrade heads)
  3. Fixture creates a test database session within a transaction for the test
  4. Your test sets up data and runs the test using the session (including COMMITing transactions)
  5. Your test verifies data is in the database
  6. Fixture rolls back the transaction (and any inner COMMITed 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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

alembic_migration_fixtures-0.2.0.tar.gz (7.5 kB view details)

Uploaded Source

Built Distribution

File details

Details for the file alembic_migration_fixtures-0.2.0.tar.gz.

File metadata

  • Download URL: alembic_migration_fixtures-0.2.0.tar.gz
  • Upload date:
  • Size: 7.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.4.2 CPython/3.11.3 Linux/5.15.0-1037-azure

File hashes

Hashes for alembic_migration_fixtures-0.2.0.tar.gz
Algorithm Hash digest
SHA256 24cdf4bf9fc1aa6532ca98ce466240f89351015002b058c0487c63e94cb1a521
MD5 32869096c62e4fbe3b59fde83a22492b
BLAKE2b-256 d3b7ead52e216f43bd01262a4e9368eb3d709e471db1a3b72e737c78437d830d

See more details on using hashes here.

File details

Details for the file alembic_migration_fixtures-0.2.0-py3-none-any.whl.

File metadata

File hashes

Hashes for alembic_migration_fixtures-0.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 3a93faf52b986613239e4e3f9cea0455cac4bf6e42a0822581395a8a866efd5c
MD5 5df7528d8942a30d12c878a90f76d923
BLAKE2b-256 b4e589b326a871ee68d40720b5b3bf27b2f22a2a6e677bac25acecdbcadec414

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page