Add test for OGR

qgis · Jan 24, 2022 · 39f3040 · 39f3040
1 parent 1be383a
commit 39f3040
Showing 1 changed file with 45 additions and 0 deletions.
diff --git a/tests/src/python/test_provider_ogr_gpkg.py b/tests/src/python/test_provider_ogr_gpkg.py
@@ -572,6 +572,13 @@ def testStyle(self):
 
         self.assertFalse(vl.dataProvider().isSaveAndLoadStyleToDatabaseSupported())
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', '/idont/exist.gpkg', '')
+        self.assertFalse(res)
+        self.assertTrue(err)
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', '/idont/exist.gpkg', 'a style')
+        self.assertFalse(res)
+        self.assertTrue(err)
+
         related_count, idlist, namelist, desclist, errmsg = vl.listStylesInDatabase()
         self.assertEqual(related_count, -1)
         self.assertEqual(idlist, [])
@@ -614,6 +621,14 @@ def testStyle(self):
 
         self.assertTrue(vl.dataProvider().isSaveAndLoadStyleToDatabaseSupported())
 
+        # style tables don't exist yet
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), '')
+        self.assertFalse(res)
+        self.assertFalse(err)
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl2.source(), 'a style')
+        self.assertFalse(res)
+        self.assertFalse(err)
+
         related_count, idlist, namelist, desclist, errmsg = vl.listStylesInDatabase()
         self.assertEqual(related_count, 0)
         self.assertEqual(idlist, [])
@@ -631,6 +646,16 @@ def testStyle(self):
         errorMsg = vl.saveStyleToDatabase("name", "description", False, "")
         self.assertEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), '')
+        self.assertFalse(res)
+        self.assertFalse(err)
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'a style')
+        self.assertFalse(res)
+        self.assertFalse(err)
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'name')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         qml, errmsg = vl.getStyleFromDatabase("not_existing")
         self.assertEqual(qml, "")
         self.assertNotEqual(errmsg, "")
@@ -656,6 +681,10 @@ def testStyle(self):
         errorMsg = vl.saveStyleToDatabase("name", "description_bis", False, "")
         self.assertNotEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'name')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         related_count, idlist, namelist, desclist, errmsg = vl.listStylesInDatabase()
         self.assertEqual(related_count, 1)
         self.assertEqual(errmsg, "")
@@ -669,6 +698,10 @@ def testStyle(self):
         errorMsg = vl.saveStyleToDatabase("name", "description_bis", False, "")
         self.assertEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'name')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         related_count, idlist, namelist, desclist, errmsg = vl.listStylesInDatabase()
         self.assertEqual(related_count, 1)
         self.assertEqual(errmsg, "")
@@ -679,12 +712,24 @@ def testStyle(self):
         errorMsg = vl2.saveStyleToDatabase("name_test2", "description_test2", True, "")
         self.assertEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl2.source(), 'name_test2')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         errorMsg = vl.saveStyleToDatabase("name2", "description2", True, "")
         self.assertEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'name2')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         errorMsg = vl.saveStyleToDatabase("name3", "description3", True, "")
         self.assertEqual(errorMsg, "")
 
+        res, err = QgsProviderRegistry.instance().styleExists('ogr', vl.source(), 'name3')
+        self.assertTrue(res)
+        self.assertFalse(err)
+
         related_count, idlist, namelist, desclist, errmsg = vl.listStylesInDatabase()
         self.assertEqual(related_count, 3)
         self.assertEqual(errmsg, "")