Add unit test

qgis · Jun 7, 2021 · 155fd5c · 155fd5c
1 parent 65a50af
commit 155fd5c
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/src/python/test_qgsauxiliarystorage.py b/tests/src/python/test_qgsauxiliarystorage.py
@@ -27,6 +27,7 @@
                        QgsPalLayerSettings,
                        QgsSymbolLayer,
                        QgsVectorLayerSimpleLabeling,
+                       QgsField,
                        QgsCallout,
                        QgsSimpleLineCallout,
                        NULL)
@@ -534,6 +535,27 @@ def testQgdCreation(self):
         qgd = newpath + '.qgd'
         self.assertTrue(os.path.exists(qgd))
 
+    def testInvalidPrimaryKey(self):
+        # create layer
+        vl = QgsVectorLayer(
+            'Point?crs=epsg:4326&field=pk:integer&key=pk',
+            'test', 'memory')
+        assert (vl.isValid())
+
+        # add a field with an invalid typename
+        field = QgsField(name="invalid_pk", type=QVariant.Int, typeName="xsd:int")
+        vl.startEditing()
+        vl.addAttribute(field)
+        vl.commitChanges()
+
+        # create auxiliary storage based on the invalid field
+        s = QgsAuxiliaryStorage()
+        pkf = vl.fields().field(vl.fields().indexOf('invalid_pk'))
+        al = s.createAuxiliaryLayer(pkf, vl)
+
+        self.assertEqual(al, None)
+        self.assertTrue("CREATE TABLE IF NOT EXISTS" in s.errorString())
+
 
 if __name__ == '__main__':
     unittest.main()