Use SQLAlchemy's testing framework fully

w/ 1.1 as the minimum version we should be able
to remove the majority of the testing suite code

Change-Id: I04c4ab45e2ce47f20613d6111895c6230b946f76

1 files changed, 8 insertions, 6 deletions

diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py
index 13ced30..28de487 100644
--- a/tests/test_postgresql.py
+++ b/tests/test_postgresql.py
@@ -625,19 +625,21 @@ class PostgresqlDetectSerialTest(TestBase):
     @classmethod
     def setup_class(cls):
         cls.bind = config.db
-        cls.conn = cls.bind.connect()
         staging_env()
-        cls.migration_context = MigrationContext.configure(
-            connection=cls.conn,
-            opts={"compare_type": True, "compare_server_default": True},
-        )
 
     def setUp(self):
+        self.conn = self.bind.connect()
+        self.migration_context = MigrationContext.configure(
+            connection=self.conn,
+            opts={"compare_type": True, "compare_server_default": True},
+        )
         self.autogen_context = api.AutogenContext(self.migration_context)
 
+    def tearDown(self):
+        self.conn.close()
+
     @classmethod
     def teardown_class(cls):
-        cls.conn.close()
         clear_staging_env()
 
     @provide_metadata