Add regression test for #390

test/integration/regression/test_regression_bug390.py

@@ -0,0 +1,34 @@
+from inspect import cleandoc
+
+pytest_plugins = "pytester"
+
+
+def test_connection_with_block_cleans_up_properly(pytester, itde):
+    config = itde.db
+    # Because the error only occurs on process shutdown we need to run a test within a test
+    # (We require the result (stderr) of a terminated process triggering the failure.
+    pytester.makepyfile(
+        # fmt: off
+        cleandoc(
+        f"""
+        from sqlalchemy import create_engine
+        
+        def test():
+            url = "exa+websocket://{{user}}:{{pw}}@{{host}}:{{port}}/{{schema}}?SSLCertificate=SSL_VERIFY_NONE"
+            url = url.format(
+                user="{config.username}",
+                pw="{config.password}",
+                host="{config.host}",
+                port={config.port},
+                schema="{itde.itde.schemas[0]}",
+            )
+            engine = create_engine(url)
+            query = "SELECT 42;"
+            with engine.connect() as con:
+                result = con.execute(query).fetchall()
+        """
+        ),
+        # fmt: on
+    )
+    r = pytester.runpytest_subprocess()
+    assert str(r.stderr) == ""