Don't fail tests if the dummy table is already present.

wulczer · wulczer · commit 2efe1424b27f · 2015-05-04T09:56:20.000+02:00
On Postgres versions that support it, DROP TABLE IF EXISTS is used to clean up
the dummy table before running the tests.
diff --git a/test/test_txpostgres.py b/test/test_txpostgres.py
@@ -449,7 +449,7 @@ class _SimpleDBSetupMixin(object):
     def setUp(self):
         d = self.restoreConnection(None)
         d.addCallback(lambda _: self.conn.cursor())
-        return d.addCallback(lambda c: c.execute(simple_table_schema))
+        return d.addCallback(self.createInitialSchema)
 
     def tearDown(self):
         c = self.conn.cursor()
@@ -467,6 +467,12 @@ def restoreConnection(self, res):
                               host=DB_HOST, database=DB_NAME)
         return d.addCallback(lambda _: res)
 
+    def createInitialSchema(self, c):
+        d = defer.succeed(None)
+        if self.conn.server_version > 80200:
+            d.addCallback(lambda _: c.execute("drop table if exists simple"))
+        return d.addCallback(lambda _: c.execute(simple_table_schema))
+
 
 class TxPostgresManualQueryTestCase(_SimpleDBSetupMixin, Psycopg2TestCase):
 
