fix: run integration tests in session (#3801)

TobikoData · Feb 7, 2025 · ebad540 · ebad540
1 parent c6af34b
commit ebad540
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/core/engine_adapter/integration/conftest.py b/tests/core/engine_adapter/integration/conftest.py
@@ -89,7 +89,8 @@ def ctx(
     ctx = TestContext(test_type, engine_adapter, mark, gateway, is_remote=is_remote)
     ctx.init()
 
-    yield ctx
+    with ctx.engine_adapter.session({}):
+        yield ctx
 
     try:
         ctx.cleanup()