aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/unit/auth/conftest.py51
-rw-r--r--tests/unit/auth/test_migration_create_users_table.py2
-rw-r--r--tests/unit/conftest.py35
3 files changed, 52 insertions, 36 deletions
diff --git a/tests/unit/auth/conftest.py b/tests/unit/auth/conftest.py
new file mode 100644
index 0000000..faecd32
--- /dev/null
+++ b/tests/unit/auth/conftest.py
@@ -0,0 +1,51 @@
+"""Fixtures for auth tests."""
+import sqlite3
+from pathlib import Path
+from typing import Union
+from contextlib import closing
+
+import pytest
+from yoyo import get_backend, read_migrations
+from yoyo.migrations import Migration, MigrationList
+
+from gn3.migrations import apply_migrations, rollback_migrations
+
+@pytest.fixture(scope="session")
+def auth_testdb_path(test_app_config): # pylint: disable=redefined-outer-name
+ """Get the test application's auth database file"""
+ return test_app_config["AUTH_DB"]
+
+@pytest.fixture(scope="session")
+def auth_migrations_dir(test_app_config): # pylint: disable=redefined-outer-name
+ """Get the test application's auth database file"""
+ return test_app_config["AUTH_MIGRATIONS"]
+
+def apply_single_migration(db_uri: Union[Path, str], migration: Migration):
+ """Utility to apply a single migration"""
+ apply_migrations(get_backend(f"sqlite:///{db_uri}"), MigrationList([migration]))
+
+def rollback_single_migration(db_uri: Union[Path, str], migration: Migration):
+ """Utility to rollback a single migration"""
+ rollback_migrations(get_backend(f"sqlite:///{db_uri}"), MigrationList([migration]))
+
+@pytest.fixture(scope="session")
+def backend(auth_testdb_path): # pylint: disable=redefined-outer-name
+ return get_backend(f"sqlite:///{auth_testdb_path}")
+
+@pytest.fixture(scope="session")
+def all_migrations(auth_migrations_dir): # pylint: disable=redefined-outer-name
+ return read_migrations(auth_migrations_dir)
+
+@pytest.fixture(scope="function")
+def conn_after_auth_migrations(backend, auth_testdb_path, all_migrations): # pylint: disable=redefined-outer-name
+ """Run all migrations and return a connection to the database after"""
+ apply_migrations(backend, all_migrations)
+ with closing(sqlite3.connect(auth_testdb_path)) as conn:
+ yield conn
+
+ rollback_migrations(backend, all_migrations)
+
+def migrations_up_to(migration, migrations_dir):
+ migrations = read_migrations(migrations_dir)
+ index = [mig.path for mig in migrations].index(migration)
+ return MigrationList(migrations[0:index])
diff --git a/tests/unit/auth/test_migration_create_users_table.py b/tests/unit/auth/test_migration_create_users_table.py
index c97c050..74f2020 100644
--- a/tests/unit/auth/test_migration_create_users_table.py
+++ b/tests/unit/auth/test_migration_create_users_table.py
@@ -5,7 +5,7 @@ import pytest
import sqlite3
from gn3.migrations import get_migration, apply_migrations, rollback_migrations
-from tests.unit.conftest import (
+from tests.unit.auth.conftest import (
apply_single_migration, rollback_single_migration)
migration_path = "migrations/auth/20221103_01_js9ub-initialise-the-auth-entic-oris-ation-database.py"
diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
index dd6abb4..a2b562e 100644
--- a/tests/unit/conftest.py
+++ b/tests/unit/conftest.py
@@ -1,17 +1,11 @@
"""Fixtures for unit tests."""
-import sqlite3
-from typing import Union
from pathlib import Path
from datetime import datetime
-from contextlib import closing
from tempfile import TemporaryDirectory
import pytest
-from yoyo import get_backend, read_migrations
-from yoyo.migrations import Migration, MigrationList
from gn3.app import create_app
-from gn3.migrations import apply_migrations, rollback_migrations
@pytest.fixture(scope="session")
def client():
@@ -31,32 +25,3 @@ def client():
def test_app_config(client): # pylint: disable=redefined-outer-name
"""Return the test application's configuration object"""
return client.application.config
-
-@pytest.fixture(scope="session")
-def auth_testdb_path(test_app_config): # pylint: disable=redefined-outer-name
- """Get the test application's auth database file"""
- return test_app_config["AUTH_DB"]
-
-@pytest.fixture(scope="session")
-def auth_migrations_dir(test_app_config): # pylint: disable=redefined-outer-name
- """Get the test application's auth database file"""
- return test_app_config["AUTH_MIGRATIONS"]
-
-def apply_single_migration(db_uri: Union[Path, str], migration: Migration):
- """Utility to apply a single migration"""
- apply_migrations(get_backend(f"sqlite:///{db_uri}"), MigrationList([migration]))
-
-def rollback_single_migration(db_uri: Union[Path, str], migration: Migration):
- """Utility to rollback a single migration"""
- rollback_migrations(get_backend(f"sqlite:///{db_uri}"), MigrationList([migration]))
-
-@pytest.fixture(scope="function")
-def conn_after_auth_migrations(auth_testdb_path, auth_migrations_dir):
- """Run all migrations and return a connection to the database after"""
- backend = get_backend(f"sqlite:///{auth_testdb_path}")
- migrations = read_migrations(auth_migrations_dir)
- apply_migrations(backend, migrations)
- with closing(sqlite3.connect(auth_testdb_path)) as conn:
- yield conn
-
- rollback_migrations(backend, migrations)