add psycopg3 dialect to connection string.

GSA · Aug 17, 2023 · 7bb874c · 7bb874c · github-actions · Aug 17, 2023
1 parent a1cfc7f
commit 7bb874c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/db/sqlalchemy/conftest.py b/tests/db/sqlalchemy/conftest.py
@@ -15,8 +15,9 @@ def engine():
     user = os.getenv("POSTGRES_USER")
     password = os.getenv("POSTGRES_PASSWORD")
     db = os.getenv("POSTGRES_DB")
+    dialect = "psycopg"
 
-    engine = create_engine(f"postgresql://{user}:{password}@{host}/{db}")
+    engine = create_engine(f"postgresql+{dialect}://{user}:{password}@{host}/{db}")
     yield engine
     engine.dispose()