From 723e1fa699b188a3d8e3ccd5589e6f64cfacdea3 Mon Sep 17 00:00:00 2001
From: Andreas Motl <andreas.motl@crate.io>
Date: Tue, 19 Dec 2023 21:31:43 +0100
Subject: [PATCH] test: Fix `FATAL: sorry, too many clients already`

Dispose the SQLAlchemy engine object after use within
`PostgresConnector`.
---
 target_postgres/connector.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/target_postgres/connector.py b/target_postgres/connector.py
index d6730539..369eb462 100644
--- 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