Add tests for QgsAttributeTableModel

tests/src/python/CMakeLists.txt

@@ -45,6 +45,7 @@ ADD_PYTHON_TEST(PyQgsGraduatedSymbolRendererV2 test_qgsgraduatedsymbolrendererv2
 ADD_PYTHON_TEST(PyQgsNetworkContentFetcher test_qgsnetworkcontentfetcher.py)
 ADD_PYTHON_TEST(PyQgsEditWidgets test_qgseditwidgets.py)
 ADD_PYTHON_TEST(PyQgsRangeWidgets test_qgsrangewidgets.py)
+ADD_PYTHON_TEST(PyQgsAttributeTableModel test_qgsattributetablemodel.py)
 IF (ENABLE_PGTEST)
   ADD_PYTHON_TEST(PyQgsPostgresProvider test_provider_postgres.py)
 ENDIF (ENABLE_PGTEST)

tests/src/python/test_qgsattributetablemodel.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+"""QGIS Unit tests for the attribute table model.
+
+.. note:: This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation; either version 2 of the License, or
+(at your option) any later version.
+"""
+__author__ = 'Matthias Kuhn'
+__date__ = '27/05/2015'
+__copyright__ = 'Copyright 2015, The QGIS Project'
+# This will get replaced with a git SHA1 when you do a git archive
+__revision__ = '$Format:%H$'
+
+from qgis.gui import QgsAttributeTableModel, QgsEditorWidgetRegistry
+from qgis.core import QgsFeature, QgsGeometry, QgsPoint, QgsVectorLayer, QgsVectorLayerCache, NULL
+
+from utilities import (unitTestDataPath,
+                       getQgisTestApp,
+                       TestCase,
+                       unittest
+                       )
+QGISAPP, CANVAS, IFACE, PARENT = getQgisTestApp()
+
+
+class TestQgsAttributeTableModel(TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        QgsEditorWidgetRegistry.initEditors()
+
+    def setUp(self):
+        self.layer = self.createLayer()
+        self.cache = QgsVectorLayerCache(self.layer, 100)
+        self.am = QgsAttributeTableModel(self.cache)
+        self.am.loadLayer()
+        self.am.loadAttributes()
+
+    def tearDown(self):
+        del self.am
+        del self.cache
+        del self.layer
+
+    def createLayer(self):
+        layer = QgsVectorLayer("Point?field=fldtxt:string&field=fldint:integer",
+                               "addfeat", "memory")
+        pr = layer.dataProvider()
+        features = list()
+        for i in range(10):
+            f = QgsFeature()
+            f.setAttributes(["test", i])
+            f.setGeometry(QgsGeometry.fromPoint(QgsPoint(100 * i, 2 ^ i)))
+            features.append(f)
+
+        assert pr.addFeatures(features)
+        return layer
+
+    def testLoad(self):
+        assert self.am.rowCount() == 10, self.am.rowCount()
+        assert self.am.columnCount() == 2, self.am.columnCount()
+
+    def testRemove(self):
+        self.layer.startEditing()
+        self.layer.deleteFeature(5)
+        assert self.am.rowCount() == 9, self.am.rowCount()
+        self.layer.setSelectedFeatures([1, 3, 6, 7])
+        self.layer.deleteSelectedFeatures()
+        assert self.am.rowCount() == 5, self.am.rowCount()
+
+    def testAdd(self):
+        self.layer.startEditing()
+
+        f = QgsFeature()
+        f.setAttributes(["test", 8])
+        f.setGeometry(QgsGeometry.fromPoint(QgsPoint(100,200)))
+        self.layer.addFeature(f)
+
+        assert self.am.rowCount() == 11, self.am.rowCount()
+
+    def testRemoveColumns(self):
+        self.layer.startEditing()
+
+        self.layer.deleteAttribute(1)
+
+        assert self.am.columnCount() == 1, self.am.columnCount()
+
+if __name__ == '__main__':
+    unittest.main()