Merge pull request #10 from andersinno/test-migration-fix

tests: Fix migration test workdir issue

Author: suutari-ai

tests/db/test_migrate.py

@@ -1,33 +1,47 @@
 """
 Test Alembic migrations
 """
-import subprocess
+import os
+from contextlib import contextmanager
+from os import fspath
+from pathlib import Path
+from typing import Iterator
 
-from passari_workflow.db.models import Base
-from alembic.config import Config
 from alembic import command
+from alembic.config import Config
 
-from pathlib import Path
+from passari_workflow.db.models import Base
+
+ROOT_PATH = Path(__file__).resolve().parent.parent.parent
 
 
 def test_migrate(session, engine, database):
     """
     Test migrations by running all migrations and then downgrading
     """
-    config = Config(
-        str(Path(__file__).resolve().parent.parent.parent / "alembic.ini")
-    )
+    config = Config(fspath(ROOT_PATH / "alembic.ini"))
 
     try:
         Base.metadata.drop_all(engine)
 
-        # Upgrade the database
-        command.upgrade(config, "head")
+        with run_in_path(ROOT_PATH):
+            # Upgrade the database
+            command.upgrade(config, "head")
 
-        # Downgrade the database
-        command.downgrade(config, "base")
+            # Downgrade the database
+            command.downgrade(config, "base")
     finally:
         # Ensure that the tables are the same after the test, even if
         # something fails
         Base.metadata.drop_all(engine)
         Base.metadata.create_all(engine)
+
+
+@contextmanager
+def run_in_path(path: Path) -> Iterator[None]:
+    orig_cwd = os.getcwd()
+    try:
+        os.chdir(path)
+        yield
+    finally:
+        os.chdir(orig_cwd)