fix: unclosed socket warning in db containers (#378)

This PR fixes #379
testcontainers · Mar 9, 2024 · cd90aa7 · cd90aa7
1 parent 902a5a3
commit cd90aa7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/testcontainers/core/generic.py b/core/testcontainers/core/generic.py
@@ -36,7 +36,10 @@ def _connect(self) -> None:
         import sqlalchemy
 
         engine = sqlalchemy.create_engine(self.get_connection_url())
-        engine.connect()
+        try:
+            engine.connect()
+        finally:
+            engine.dispose()
 
     def get_connection_url(self) -> str:
         raise NotImplementedError