def setUp(self):
self.registry = PermissionHandlerRegistry()
self.model = MagicMock()
self.model._meta = MagicMock()
self.model._meta.abstract = False
self.handler = PermissionHandler
def setUp(self):
self.registry = PermissionHandlerRegistry()
self.model = MagicMock()
self.model._meta = MagicMock()
self.model._meta.abstract = False
self.handler = PermissionHandler
class PermissionUtilsHandlersTestCase(TestCase):
def setUp(self):
self.registry = PermissionHandlerRegistry()
self.model = MagicMock()
self.model._meta = MagicMock()
self.model._meta.abstract = False
self.handler = PermissionHandler
def test_register(self):
self.registry.register(self.model, self.handler)
self.assertTrue(self.model in self.registry._registry)
self.assertTrue(isinstance(self.registry._registry[self.model],
self.handler))
def test_register_without_specifing_handler(self):
self.registry.register(self.model)
self.assertTrue(self.model in self.registry._registry)
self.assertTrue(isinstance(self.registry._registry[self.model],
self.handler))
def test_register_with_abstract_model(self):
from django.core.exceptions import ImproperlyConfigured
abstract_model = MagicMock()
abstract_model._meta = MagicMock()
abstract_model._meta.abstract = True
self.assertRaises(ImproperlyConfigured,
self.registry.register,
abstract_model, self.handler)
def test_register_duplicate(self):
self.registry.register(self.model, self.handler)
self.assertRaises(KeyError,
self.registry.register,
self.model, self.handler)
def test_register_permission_handler_instance(self):
handler_instance = self.handler(self.model)
self.assertRaises(AttributeError,
self.registry.register,
self.model, handler_instance)
def test_register_non_permission_handler(self):
self.assertRaises(AttributeError,
self.registry.register,
self.model, self.__class__)
def test_unregister(self):
self.registry.register(self.model, self.handler)
self.registry.unregister(self.model)
self.assertFalse(self.model in self.registry._registry)
def test_unregister_absence(self):
self.assertRaises(KeyError,
self.registry.unregister,
self.model)
def test_get_handlers(self):
results = self.registry.get_handlers()
self.assertTrue(isinstance(results, tuple))
self.assertTrue(len(results) == 0)
self.registry.register(self.model, self.handler)
results = self.registry.get_handlers()
self.assertTrue(isinstance(results, tuple))
self.assertTrue(len(results) == 1)
self.assertTrue(isinstance(results[0], PermissionHandler))
class PermissionUtilsHandlersTestCase(TestCase):
def setUp(self):
self.registry = PermissionHandlerRegistry()
self.model = MagicMock()
self.model._meta = MagicMock()
self.model._meta.abstract = False
self.handler = PermissionHandler
def test_register(self):
self.registry.register(self.model, self.handler)
self.assertTrue(self.model in self.registry._registry)
self.assertTrue(
isinstance(self.registry._registry[self.model], self.handler))
def test_register_without_specifing_handler(self):
self.registry.register(self.model)
self.assertTrue(self.model in self.registry._registry)
self.assertTrue(
isinstance(self.registry._registry[self.model], self.handler))
def test_register_with_abstract_model(self):
from django.core.exceptions import ImproperlyConfigured
abstract_model = MagicMock()
abstract_model._meta = MagicMock()
abstract_model._meta.abstract = True
self.assertRaises(ImproperlyConfigured, self.registry.register,
abstract_model, self.handler)
def test_register_duplicate(self):
self.registry.register(self.model, self.handler)
self.assertRaises(KeyError, self.registry.register, self.model,
self.handler)
def test_register_permission_handler_instance(self):
handler_instance = self.handler(self.model)
self.assertRaises(AttributeError, self.registry.register, self.model,
handler_instance)
def test_register_non_permission_handler(self):
self.assertRaises(AttributeError, self.registry.register, self.model,
self.__class__)
def test_unregister(self):
self.registry.register(self.model, self.handler)
self.registry.unregister(self.model)
self.assertFalse(self.model in self.registry._registry)
def test_unregister_absence(self):
self.assertRaises(KeyError, self.registry.unregister, self.model)
def test_get_handlers(self):
results = self.registry.get_handlers()
self.assertTrue(isinstance(results, tuple))
self.assertTrue(len(results) == 0)
self.registry.register(self.model, self.handler)
results = self.registry.get_handlers()
self.assertTrue(isinstance(results, tuple))
self.assertTrue(len(results) == 1)
self.assertTrue(isinstance(results[0], PermissionHandler))
