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)
def create_registry_with_registered_enum():
registry = PermissionRegistry()
registry.register_enum(ItemPermission)
return registry
