Make TestPyQgsPostgresProvider.testPktUpdateBigintPk idempotent

References #45417
qgis · Oct 5, 2021 · 41f1924 · 41f1924
1 parent bb30046
commit 41f1924
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
@@ -720,6 +720,9 @@ def testPktUpdateBigintPk(self):
             "bigint_pk", "postgres")
         flds = vl.fields()
 
+        # Backup test table (will be edited)
+        self.backupTable('qgis_test', 'bigint_pk')
+
         self.assertTrue(vl.isValid())
 
         vl.startEditing()
@@ -763,6 +766,9 @@ def testPktUpdateBigintPk(self):
                 statuses[3] = 1
         self.assertTrue(all(x == 1 for x in statuses))
 
+        # Restore test table
+        self.restoreTable('qgis_test', 'bigint_pk')
+
     def testPktUpdateBigintPkNonFirst(self):
         """Test if we can update objects with positive, zero and negative bigint PKs in tables whose PK is not the first field"""
         vl = QgsVectorLayer('{} sslmode=disable srid=4326 key="pk" table="qgis_test".{} (geom)'.format(self.dbconn,