Testing boilerplate unified in a single base class

The class makes a connection always available, allows creating
new connection and closes everything on tear down.

1 files changed, 6 insertions, 9 deletions

diff --git a/tests/test_dates.py b/tests/test_dates.py
index 0d18f66..cc183c5 100755
--- a/tests/test_dates.py
+++ b/tests/test_dates.py
@@ -23,10 +23,9 @@
 # License for more details.
 
 import math
-import unittest
 import psycopg2
 from psycopg2.tz import FixedOffsetTimezone, ZERO
-from testconfig import dsn
+from testutils import unittest, ConnectingTestCase
 
 class CommonDatetimeTestsMixin:
 
@@ -93,20 +92,17 @@ class CommonDatetimeTestsMixin:
         self.assertEqual(value, None)
 
 
-class DatetimeTests(unittest.TestCase, CommonDatetimeTestsMixin):
+class DatetimeTests(ConnectingTestCase, CommonDatetimeTestsMixin):
     """Tests for the datetime based date handling in psycopg2."""
 
     def setUp(self):
-        self.conn = psycopg2.connect(dsn)
+        ConnectingTestCase.setUp(self)
         self.curs = self.conn.cursor()
         self.DATE = psycopg2.extensions.PYDATE
         self.TIME = psycopg2.extensions.PYTIME
         self.DATETIME = psycopg2.extensions.PYDATETIME
         self.INTERVAL = psycopg2.extensions.PYINTERVAL
 
-    def tearDown(self):
-        self.conn.close()
-
     def test_parse_bc_date(self):
         # datetime does not support BC dates
         self.assertRaises(ValueError, self.DATE, '00042-01-01 BC', self.curs)
@@ -311,11 +307,11 @@ if not hasattr(psycopg2.extensions, 'PYDATETIME'):
     del DatetimeTests
 
 
-class mxDateTimeTests(unittest.TestCase, CommonDatetimeTestsMixin):
+class mxDateTimeTests(ConnectingTestCase, CommonDatetimeTestsMixin):
     """Tests for the mx.DateTime based date handling in psycopg2."""
 
     def setUp(self):
-        self.conn = psycopg2.connect(dsn)
+        ConnectingTestCase.setUp(self)
         self.curs = self.conn.cursor()
         self.DATE = psycopg2._psycopg.MXDATE
         self.TIME = psycopg2._psycopg.MXTIME
@@ -557,6 +553,7 @@ class FixedOffsetTimezoneTests(unittest.TestCase):
             self.assertEqual(tz11, tz21)
             self.assertEqual(tz12, tz22)
 
+
 def test_suite():
     return unittest.TestLoader().loadTestsFromName(__name__)