- add some connection cleanup

1 files changed, 3 insertions, 1 deletions

diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py
index f1b2c2d..908eec6 100644
--- a/tests/test_postgresql.py
+++ b/tests/test_postgresql.py
@@ -319,9 +319,10 @@ class PostgresqlDetectSerialTest(TestBase):
     @classmethod
     def setup_class(cls):
         cls.bind = config.db
+        cls.conn = cls.bind.connect()
         staging_env()
         context = MigrationContext.configure(
-            connection=cls.bind.connect(),
+            connection=cls.conn,
             opts={
                 'compare_type': True,
                 'compare_server_default': True
@@ -343,6 +344,7 @@ class PostgresqlDetectSerialTest(TestBase):
 
     @classmethod
     def teardown_class(cls):
+        cls.conn.close()
         clear_staging_env()
 
     @provide_metadata