add test case with specified raster column

qgis · Jan 22, 2025 · 8372493 · 8372493
1 parent f8c9a40
commit 8372493
Showing 1 changed file with 74 additions and 0 deletions.
diff --git a/tests/src/python/test_provider_postgresraster.py b/tests/src/python/test_provider_postgresraster.py
@@ -1100,6 +1100,80 @@ def testStyle(self):
         self.assertEqual(desclist, ["default test style"])
         self.assertFalse(errmsg)
 
+        # Raster layer with defined column
+        rl_1 = QgsRasterLayer(
+            self.dbconn
+            + ' sslmode=disable srid=3035  table="public"."raster_tiled_3035" (rast) sql=',
+            "test",
+            "postgresraster",
+        )
+
+        self.assertTrue(rl_1.isValid())
+
+        # not style yet for layer
+        res, err = QgsProviderRegistry.instance().styleExists(
+            "postgresraster", rl_1.source(), ""
+        )
+        self.assertFalse(res)
+        self.assertFalse(err)
+
+        related_count, idlist, namelist, desclist, errmsg = rl_1.listStylesInDatabase()
+        self.assertEqual(related_count, 0)
+        self.assertEqual(idlist, [])
+        self.assertEqual(namelist, [])
+        self.assertEqual(desclist, [])
+        self.assertFalse(errmsg)
+
+        # Save style twice, one as as default
+        errmsg = rl_1.saveStyleToDatabase(
+            "related raster style", "test style", False, ""
+        )
+        self.assertEqual(errmsg, "")
+
+        related_count, idlist, namelist, desclist, errmsg = rl_1.listStylesInDatabase()
+        self.assertEqual(related_count, 1)
+        self.assertEqual(idlist, ["3"])
+        self.assertEqual(namelist, ["related raster style"])
+        self.assertEqual(desclist, ["test style"])
+        self.assertFalse(errmsg)
+
+        errmsg = rl_1.saveStyleToDatabase(
+            "related raster style default", "default test style", True, ""
+        )
+        self.assertEqual(errmsg, "")
+
+        # check style exist
+        res, err = QgsProviderRegistry.instance().styleExists(
+            "postgresraster", rl_1.source(), "related raster style default"
+        )
+        self.assertTrue(res)
+        self.assertFalse(err)
+
+        qml, errmsg = rl_1.getStyleFromDatabase("3")
+        self.assertTrue(qml)
+        self.assertEqual(errmsg, "")
+
+        related_count, idlist, namelist, desclist, errmsg = rl_1.listStylesInDatabase()
+        self.assertEqual(related_count, 2)
+        self.assertEqual(idlist, ["4", "3"])
+        self.assertEqual(
+            namelist, ["related raster style default", "related raster style"]
+        )
+        self.assertEqual(desclist, ["default test style", "test style"])
+        self.assertFalse(errmsg)
+
+        # Remove these style
+        res, errmsg = rl_1.deleteStyleFromDatabase("3")
+        self.assertTrue(res)
+        self.assertFalse(errmsg)
+
+        related_count, idlist, namelist, desclist, errmsg = rl_1.listStylesInDatabase()
+        self.assertEqual(related_count, 1)
+        self.assertEqual(idlist, ["4"])
+        self.assertEqual(namelist, ["related raster style default"])
+        self.assertEqual(desclist, ["default test style"])
+        self.assertFalse(errmsg)
+
 
 if __name__ == "__main__":
     unittest.main()