def test_add(self):
data_store_registry = io.DataStoreRegistry()
data_store_registry.add_data_store(self.c1)
self.assertEqual(1, len(data_store_registry))
data_store_registry.add_data_store(self.c2)
self.assertEqual(2, len(data_store_registry))
data_store_registry.add_data_store(self.c2)
self.assertEqual(2, len(data_store_registry))
def test_get_data_stores(self):
data_store_registry = io.DataStoreRegistry()
data_store_registry.add_data_store(self.c1)
data_store_registry.add_data_store(self.c2)
self.assertEqual(2, len(data_store_registry))
data_stores = data_store_registry.get_data_stores()
self.assertIsNotNone(data_stores)
self.assertEqual(2, len(data_stores))
def test_remove(self):
data_store_registry = io.DataStoreRegistry()
data_store_registry.add_data_store(self.c1)
data_store_registry.add_data_store(self.c2)
self.assertEqual(2, len(data_store_registry))
data_store_registry.remove_data_store('c1')
self.assertEqual(1, len(data_store_registry))
with self.assertRaises(KeyError):
data_store_registry.remove_data_store('c0')
self.assertEqual(1, len(data_store_registry))
def test_get_data_store(self):
data_store_registry = io.DataStoreRegistry()
data_store_registry.add_data_store(self.c1)
data_store_registry.add_data_store(self.c2)
self.assertEqual(2, len(data_store_registry))
rc2 = data_store_registry.get_data_store('c2')
self.assertIsNotNone(rc2)
self.assertIs(self.c2, rc2)
self.assertEqual(2, len(data_store_registry))
rc0 = data_store_registry.get_data_store('c0')
self.assertIsNone(rc0)
def test_init(self):
data_store_registry = io.DataStoreRegistry()
self.assertEqual(0, len(data_store_registry))