add try block

Eventual-Inc · Feb 23, 2024 · f0edfde · f0edfde
1 parent f93ae9a
commit f0edfde
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integration/sql/conftest.py b/tests/integration/sql/conftest.py
@@ -14,5 +14,9 @@
 )
 @pytest.fixture(scope="session")
 def check_db_server_initialized() -> None:
-    with sqlalchemy.create_engine(TRINO_URL).connect() as conn:
-        conn.execute(sqlalchemy.text("SELECT 1"))
+    try:
+        with sqlalchemy.create_engine(TRINO_URL).connect() as conn:
+            conn.execute(sqlalchemy.text("SELECT 1"))
+    except Exception as e:
+        print(f"Connection failed with exception: {e}")
+        raise