def setUp(self):
self.registry = ComponentRegistry()
self.instance = FakeClass()
class ComponentRegistryTestCase(test.TestCase):
"""
Tests for registry
"""
def setUp(self):
self.registry = ComponentRegistry()
self.instance = FakeClass()
def test_register(self):
self.registry.register("component", "test", self.instance)
self.assertTrue("component" in self.registry)
def test_get(self):
self.registry.register("component", "test", self.instance)
self.assertEqual(1, len(self.registry.get("component")))
self.assertTrue("test" in self.registry.get("component"))
def test_register_multiple(self):
self.registry.register("component", "test", self.instance)
instance_two = FakeClass()
self.registry.register("component", "test_two", instance_two)
self.assertEqual(len(self.registry.get("component")), 2)
def test_reregister(self):
"""
Registry shouldn't allow a registered key's value to be overwritten.
"""
self.registry.register("component", "test", self.instance)
instance_two = FakeClass()
self.assertRaises(
ComponentRegistry.AlreadyRegistered,
self.registry.register,
"component",
"test",
instance_two
)
def test_get_not_registered(self):
self.assertRaises(
ComponentRegistry.NotRegistered,
self.registry.get,
"test"
)
