Make TestPyQgsPostgresProvider.testGeneratedColumns idempotent

Closes #45414
qgis · Oct 5, 2021 · 989b651 · 989b651
1 parent 9f49c0d
commit 989b651
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/src/python/providertestbase.py b/tests/src/python/providertestbase.py
@@ -1233,3 +1233,9 @@ def testGeneratedColumns(self):
         self.assertEqual(feature.attribute(1), "test:8")
         self.assertFalse(QgsVectorLayerUtils.fieldIsEditable(vl, 1, feature))
         self.assertFalse(QgsVectorLayerUtils.fieldIsEditable(vl, 0, feature))
+
+        # CLEANUP: delete features added during test (cleanup)
+        vl.startEditing()
+        self.assertTrue(vl.deleteFeature(10))
+        self.assertTrue(vl.commitChanges())
+        # TODO: further cleanups in case attributes have been changed