test: Fix FATAL: sorry, too many clients already

Dispose the SQLAlchemy engine object after use within test utility functions.
MeltanoLabs · Dec 19, 2023 · 8b3ea4f · 8b3ea4f
1 parent c0d4a89
commit 8b3ea4f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/target_postgres/tests/test_target_postgres.py b/target_postgres/tests/test_target_postgres.py
@@ -132,6 +132,7 @@ def verify_data(
                 sqlalchemy.text(f"SELECT COUNT(*) FROM {full_table_name}")
             )
             assert result.first()[0] == number_of_rows
+    engine.dispose()
 
 
 def verify_schema(
@@ -169,6 +170,7 @@ def verify_schema(
                     f"Column '{column.name}' (with type '{column.type}') "
                     f"does not match expected type: {column_type_expected}"
                 )
+    engine.dispose()
 
 
 def test_sqlalchemy_url_config(postgres_config_no_ssl):