Add some unit tests for layer notes

qgis · May 4, 2021 · 7ec1abb · 7ec1abb
1 parent 3bdb2bb
commit 7ec1abb
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/tests/src/python/test_qgsmaplayer.py b/tests/src/python/test_qgsmaplayer.py
@@ -21,7 +21,8 @@
                        QgsVectorLayer,
                        QgsRasterLayer,
                        QgsProject,
-                       QgsLayerMetadata)
+                       QgsLayerMetadata,
+                       QgsLayerNotesUtils)
 from qgis.testing import start_app, unittest
 from qgis.PyQt.QtXml import QDomDocument
 from qgis.PyQt.QtCore import QTemporaryDir
@@ -77,6 +78,25 @@ def testGettersSetters(self):
         self.assertFalse(layer.hasAutoRefreshEnabled())
         self.assertEqual(layer.autoRefreshInterval(), 0)
 
+    def testLayerNotes(self):
+        """
+        Test layer notes
+        """
+        layer = QgsVectorLayer("Point?field=fldtxt:string",
+                               "layer", "memory")
+        self.assertFalse(QgsLayerNotesUtils.layerHasNotes(layer))
+        self.assertFalse(QgsLayerNotesUtils.layerNotes(layer))
+
+        QgsLayerNotesUtils.setLayerNotes(layer, 'my notes')
+        self.assertTrue(QgsLayerNotesUtils.layerHasNotes(layer))
+        self.assertEqual(QgsLayerNotesUtils.layerNotes(layer), 'my notes')
+        QgsLayerNotesUtils.setLayerNotes(layer, 'my notes 2')
+        self.assertEqual(QgsLayerNotesUtils.layerNotes(layer), 'my notes 2')
+
+        QgsLayerNotesUtils.removeNotes(layer)
+        self.assertFalse(QgsLayerNotesUtils.layerHasNotes(layer))
+        self.assertFalse(QgsLayerNotesUtils.layerNotes(layer))
+
     def testSaveRestoreAutoRefresh(self):
         """ test saving/restoring auto refresh to xml """
         layer = QgsVectorLayer("Point?field=fldtxt:string",