Some more test checks

qgis · Feb 13, 2021 · d4e8658 · d4e8658
1 parent e89c163
commit d4e8658
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/src/python/test_qgsvectorlayereditbuffer.py b/tests/src/python/test_qgsvectorlayereditbuffer.py
@@ -565,30 +565,37 @@ def _check_feature(wkt):
             field = QgsField('attr1', QVariant.String)
             self.assertTrue(layer_a.addAttribute(field))
             self.assertNotEqual(layer_a.fields().lookupField(field.name()), -1)
+            self.assertEqual(buffer.addedAttributes(), [field])
 
             layer_a.undoStack().undo()
             self.assertEqual(layer_a.fields().lookupField(field.name()), -1)
+            self.assertEqual(buffer.addedAttributes(), [])
 
             layer_a.undoStack().redo()
             self.assertNotEqual(layer_a.fields().lookupField(field.name()), -1)
+            self.assertEqual(buffer.addedAttributes(), [field])
 
             self.assertTrue(layer_a.commitChanges())
 
             ###########################################
             # Remove attribute
 
             self.assertTrue(layer_a.startEditing())
+            buffer = layer_a.editBuffer()
 
             attr_idx = layer_a.fields().lookupField(field.name())
             self.assertNotEqual(attr_idx, -1)
 
             self.assertTrue(layer_a.deleteAttribute(attr_idx))
+            self.assertEqual(buffer.deletedAttributeIds(), [2])
             self.assertEqual(layer_a.fields().lookupField(field.name()), -1)
 
             layer_a.undoStack().undo()
+            self.assertEqual(buffer.deletedAttributeIds(), [])
             self.assertEqual(layer_a.fields().lookupField(field.name()), attr_idx)
 
             layer_a.undoStack().redo()
+            self.assertEqual(buffer.deletedAttributeIds(), [2])
             self.assertEqual(layer_a.fields().lookupField(field.name()), -1)
 
             self.assertTrue(layer_a.rollBack())