Add tests for item provider registry new methods

qgis · Mar 11, 2020 · a3f4911 · a3f4911
1 parent 016f7ed
commit a3f4911
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/src/python/test_qgsdataitemproviderregistry.py b/tests/src/python/test_qgsdataitemproviderregistry.py
@@ -72,6 +72,14 @@ def testRegistry(self):
         self.assertNotIn('p2', [p.name() for p in registry.providers()])
         self.assertEqual(registry.providers(), initial_providers)
 
+    def testProviderKey(self):
+        """Tests finding provider by name and return dataProviderKey"""
+
+        registry = QgsDataItemProviderRegistry()
+        self.assertIsNotNone(registry.provider('PostGIS'))
+        self.assertIsNone(registry.provider('paper_and_pencil'))
+        self.assertEqual(registry.provider('PostGIS').dataProviderKey(), 'postgres')
+
 
 if __name__ == '__main__':
     unittest.main()