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')
def testRegistry(self):
registry = QgsDataItemProviderRegistry()
initial_providers = registry.providers()
self.assertTrue(initial_providers) # we expect a bunch of default providers
self.assertTrue([p.name() for p in initial_providers if p.name() == 'GDAL'])
# add a new provider
p1 = TestProvider('p1')
registry.addProvider(p1)
self.assertIn(p1, registry.providers())
p2 = TestProvider('p2')
registry.addProvider(p2)
self.assertIn(p1, registry.providers())
self.assertIn(p2, registry.providers())
registry.removeProvider(None)
p3 = TestProvider('p3')
# not in registry yet
registry.removeProvider(p3)
registry.removeProvider(p1)
self.assertNotIn('p1', [p.name() for p in registry.providers()])
self.assertIn(p2, registry.providers())
registry.removeProvider(p2)
self.assertNotIn('p2', [p.name() for p in registry.providers()])
self.assertEqual(registry.providers(), initial_providers)
def testRegistry(self):
registry = QgsDataItemProviderRegistry()
initial_providers = registry.providers()
self.assertTrue(initial_providers) # we expect a bunch of default providers
self.assertTrue([p.name() for p in initial_providers if p.name() == 'GDAL'])
# add a new provider
p1 = TestProvider('p1')
registry.addProvider(p1)
self.assertIn(p1, registry.providers())
p2 = TestProvider('p2')
registry.addProvider(p2)
self.assertIn(p1, registry.providers())
self.assertIn(p2, registry.providers())
registry.removeProvider(None)
p3 = TestProvider('p3')
# not in registry yet
registry.removeProvider(p3)
registry.removeProvider(p1)
self.assertNotIn('p1', [p.name() for p in registry.providers()])
self.assertIn(p2, registry.providers())
registry.removeProvider(p2)
self.assertNotIn('p2', [p.name() for p in registry.providers()])
self.assertEqual(registry.providers(), initial_providers)