Fix feature count for categorized renderer (NULLs)

Fixes #45280
qgis · Oct 4, 2021 · f6d7ed7 · f6d7ed7
1 parent 924eb4f
commit f6d7ed7
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tests/src/python/test_qgsrulebasedrenderer.py b/tests/src/python/test_qgsrulebasedrenderer.py
@@ -503,6 +503,38 @@ def testConvertFromEmbedded(self):
         self.assertEqual(rule_2.label(), 'All other features')
         self.assertEqual(rule_2.symbol().color().name(), '#ff00ff')
 
+    def testNullsCount(self):
+
+        vl = QgsVectorLayer('Point?crs=epsg:4326&field=number:int',
+                            'test', 'memory')
+
+        f = QgsFeature(vl.fields())
+        f.setAttribute(0, 0)
+        f.setGeometry(QgsGeometry.fromWkt('point(7 45)'))
+        vl.dataProvider().addFeatures([f])
+        f = QgsFeature(vl.fields())
+        f.setGeometry(QgsGeometry.fromWkt('point(7 45)'))
+        f.setAttribute(0, 1)
+        vl.dataProvider().addFeatures([f])
+        f.setGeometry(QgsGeometry.fromWkt('point(7 45)'))
+        f = QgsFeature(vl.fields())
+        vl.dataProvider().addFeatures([f])
+
+        cats = []
+        cats.append(QgsRendererCategory(1, QgsMarkerSymbol(), 'one'))
+        cats.append(QgsRendererCategory(0, QgsMarkerSymbol(), 'zero'))
+        cats.append(QgsRendererCategory(None, QgsMarkerSymbol(), 'NULL'))
+        renderer = QgsCategorizedSymbolRenderer("number", cats)
+
+        vl.setRenderer(renderer)
+
+        counter = vl.countSymbolFeatures()
+        counter.waitForFinished()
+
+        self.assertEqual(counter.featureCount('0'), 1)
+        self.assertEqual(counter.featureCount('1'), 1)
+        self.assertEqual(counter.featureCount('2'), 1)
+
 
 if __name__ == '__main__':
     unittest.main()