test: Fix FATAL: sorry, too many clients already

Dispose the SQLAlchemy engine object after use within `PostgresConnector`.
MeltanoLabs · Dec 19, 2023 · 723e1fa · 723e1fa
1 parent a9d1796
commit 723e1fa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/target_postgres/connector.py b/target_postgres/connector.py
@@ -180,8 +180,10 @@ def copy_table_structure(
 
     @contextmanager
     def _connect(self) -> t.Iterator[sqlalchemy.engine.Connection]:
-        with self._engine.connect().execution_options() as conn:
+        engine = self._engine
+        with engine.connect().execution_options() as conn:
             yield conn
+        engine.dispose()
 
     def drop_table(
         self, table: sqlalchemy.Table, connection: sqlalchemy.engine.Connection