class PermissionRegistryTestCase(TestCase): def setUp(self): self.registry = PermissionRegistry() self.registry.register_enum(ItemPermission) def test_lookup_of_existing_enum_member(self): permission = create_permission('item.create') self.assert_get_enum_member_equals(permission, ItemPermission.create) def test_lookup_of_nonexistent_member_of_existing_enum(self): permission = create_permission('item.delete') self.assert_get_enum_member_equals(permission, None) def test_lookup_of_member_of_nonexistent_enum(self): permission = create_permission('article.create') self.assert_get_enum_member_equals(permission, None) def assert_get_enum_member_equals(self, permission, expected): actual = self.registry.get_enum_member(permission) self.assertEqual(actual, expected)
def create_registry_with_registered_enum(): registry = PermissionRegistry() registry.register_enum(ItemPermission) return registry
def setUp(self): self.registry = PermissionRegistry() self.registry.register_enum(ItemPermission)