Skip to content
Permalink
Browse files

Add tests for QgsAttributeTableModel

  • Loading branch information
m-kuhn committed May 27, 2015
1 parent d4b72a2 commit bb9d41374a252b3a62f63915ceba0e39a71d715b
Showing with 89 additions and 0 deletions.
  1. +1 −0 tests/src/python/CMakeLists.txt
  2. +88 −0 tests/src/python/test_qgsattributetablemodel.py
@@ -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)
@@ -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()

0 comments on commit bb9d413

Please sign in to comment.