1 files changed, 22 insertions, 4 deletions
diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
index bc974d9..dd6abb4 100644
--- a/tests/unit/conftest.py
+++ b/tests/unit/conftest.py
@@ -1,11 +1,13 @@
 """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
+from yoyo import get_backend, read_migrations
 from yoyo.migrations import Migration, MigrationList
 
 from gn3.app import create_app
@@ -25,16 +27,21 @@ def client():
         # Clean up after ourselves
         testdb.unlink(missing_ok=True)
 
-@pytest.fixture()
+@pytest.fixture(scope="session")
 def test_app_config(client): # pylint: disable=redefined-outer-name
     """Return the test application's configuration object"""
     return client.application.config
 
-@pytest.fixture()
-def auth_testdb(test_app_config): # pylint: disable=redefined-outer-name
+@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]))
@@ -42,3 +49,14 @@ def apply_single_migration(db_uri: Union[Path, str], migration: 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)