Add tests for QgsSymbol flags

nyalldawson · nyalldawson · commit faa8be243eeb · 2021-05-26T18:26:44.000+10:00
diff --git a/tests/src/python/test_qgssymbol.py b/tests/src/python/test_qgssymbol.py
@@ -55,7 +55,8 @@
                        QgsSymbolLayerUtils,
                        QgsMarkerLineSymbolLayer,
                        QgsArrowSymbolLayer,
-                       QgsSymbol
+                       QgsSymbol,
+                       Qgis
                        )
 
 from qgis.testing import unittest, start_app
@@ -168,6 +169,27 @@ def testSymbolTypeForGeometryType(self):
         self.assertEqual(QgsSymbol.symbolTypeForGeometryType(QgsWkbTypes.NullGeometry), QgsSymbol.Hybrid)
         self.assertEqual(QgsSymbol.symbolTypeForGeometryType(QgsWkbTypes.UnknownGeometry), QgsSymbol.Hybrid)
 
+    def testFlags(self):
+        """
+        Test symbol flags
+        """
+        s = QgsLineSymbol.createSimple({})
+        self.assertEqual(s.flags(), Qgis.SymbolFlags())
+
+        s.setFlags(Qgis.SymbolFlag.RendererShouldUseSymbolLevels)
+        self.assertEqual(s.flags(), Qgis.SymbolFlag.RendererShouldUseSymbolLevels)
+
+        s2 = s.clone()
+        self.assertEqual(s2.flags(), Qgis.SymbolFlag.RendererShouldUseSymbolLevels)
+
+        # test that flags are saved/restored via XML
+        doc = QDomDocument()
+        context = QgsReadWriteContext()
+        element = QgsSymbolLayerUtils.saveSymbol('test', s, doc, context)
+
+        s2 = QgsSymbolLayerUtils.loadSymbol(element, context)
+        self.assertEqual(s2.flags(), Qgis.SymbolFlag.RendererShouldUseSymbolLevels)
+
     def testCanCauseArtifactsBetweenAdjacentTiles(self):
         """
         Test canCauseArtifactsBetweenAdjacentTiles()
