Make TestPyQgsPostgresProvider.testPktCompositeFloat idempotent

References #45417
qgis · Oct 13, 2021 · 2a4ba7d · 2a4ba7d
1 parent 1221e61
commit 2a4ba7d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/src/python/test_provider_postgres.py b/tests/src/python/test_provider_postgres.py
@@ -902,6 +902,9 @@ def testPktCompositeFloat(self):
         self.assertAlmostEqual(f['pk3'], 3.14159274)
         self.assertEqual(f['value'], 'test 2')
 
+        # Backup test table (will be edited)
+        self.backupTable('qgis_test', 'tb_test_composite_float_pk')
+
         # can we edit a field?
         vl.startEditing()
         vl.changeAttributeValue(f.id(), fields.indexOf('value'), 'Edited Test 2')
@@ -959,6 +962,9 @@ def testPktCompositeFloat(self):
 
         self.assertFalse(got_feature)
 
+        # Restore test table
+        self.restoreTable('qgis_test', 'tb_test_composite_float_pk')
+
     def testPktFloatingPoint(self):
         """
         Check if we can handle floating point/numeric primary keys correctly