one test missing

fractal-analytics-platform · Jun 13, 2024 · a22c85b · a22c85b
1 parent bb0275d
commit a22c85b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/v1/03_db/test_unit_db_models.py b/tests/v1/03_db/test_unit_db_models.py
@@ -155,7 +155,7 @@ async def test_project_and_workflows(db):
     await db.delete(db_project)
 
     DB_ENGINE = Inject(get_settings).DB_ENGINE
-    if DB_ENGINE == "postgres":
+    if DB_ENGINE in ["postgres", "postgres-psycopg"]:
         with pytest.raises(IntegrityError):
             # Workflow.project_id violates fk-contraint in Postgres
             await db.commit()
@@ -310,7 +310,7 @@ async def test_project_and_datasets(db):
     await db.delete(db_project)
 
     DB_ENGINE = Inject(get_settings).DB_ENGINE
-    if DB_ENGINE == "postgres":
+    if DB_ENGINE in ["postgres", "postgres-psycopg"]:
         with pytest.raises(IntegrityError):
             # Dataset.project_id violates fk-contraint in Postgres
             await db.commit()