Working factories

app/lib/sqlalchemy_plugin.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import sys
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, cast
 from uuid import UUID
@@ -73,15 +74,26 @@ def create_db_engine(connection_settings: DBConnectionSettings) -> AsyncEngine:
     )
 
 
-engine = create_db_engine(
-    connection_settings=DBConnectionSettings(
-        username=settings.db.USER,
-        password=settings.db.PASSWORD,
-        host=settings.db.HOST,
-        port=settings.db.PORT,
-        database=settings.db.NAME,
+if "pytest" in sys.modules:
+    engine = create_db_engine(
+        connection_settings=DBConnectionSettings(
+            username=settings.testing.DB_USER,
+            password=settings.testing.DB_PASSWORD,
+            host=settings.testing.DB_HOST,
+            port=settings.testing.DB_PORT,
+            database=settings.testing.DB_NAME,
+        )
+    )
+else:
+    engine = create_db_engine(
+        connection_settings=DBConnectionSettings(
+            username=settings.db.USER,
+            password=settings.db.PASSWORD,
+            host=settings.db.HOST,
+            port=settings.db.PORT,
+            database=settings.db.NAME,
+        )
     )
-)
 
 
 """Configure via DatabaseSettings.