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" )