Updates tests for source select registry

elpaso · elpaso · commit fd7bcb368dca · 2017-09-04T15:16:20.000+02:00
diff --git a/tests/src/python/test_qgssourceselectprovider.py b/tests/src/python/test_qgssourceselectprovider.py
@@ -120,13 +120,28 @@ def _testRegistry(self, registry):
         # Get not existent by key
         self.assertEqual(len(registry.providersByKey('Oh This Is Missing!')), 0)
 
+    def testRemoveProvider(self):
+        registry = QgsSourceSelectProviderRegistry()
+        registry.addProvider(ConcreteSourceSelectProvider())
+        registry.addProvider(ConcreteSourceSelectProvider2())
+        self.assertEqual(['MyTestProviderKey', 'MyName'], [p.name() for p in registry.providers() if p.providerKey().startswith('MyTestProviderKey')])
+
+        self.assertTrue(registry.removeProvider(registry.providerByName('MyName')))
+        self.assertEqual(['MyTestProviderKey'], [p.name() for p in registry.providers() if p.providerKey().startswith('MyTestProviderKey')])
+
+        self.assertTrue(registry.removeProvider(registry.providerByName('MyTestProviderKey')))
+        self.assertEqual([], [p.name() for p in registry.providers() if p.providerKey().startswith('MyTestProviderKey')])
+
     def testRegistry(self):
         registry = QgsSourceSelectProviderRegistry()
         self._testRegistry(registry)
 
     def testRegistrySingleton(self):
         registry = QgsGui.sourceSelectProviderRegistry()
         self._testRegistry(registry)
+        # Check that at least OGR and GDAL are here
+        self.assertTrue(registry.providersByKey('ogr'))
+        self.assertTrue(registry.providersByKey('gdal'))
 
 
 if __name__ == '__main__':
