|
@@ 74-88 (lines=15) @@
|
| 71 |
|
LogicalPermissionHandler)) |
| 72 |
|
|
| 73 |
|
def test_remove_permission_logic_registry_with_class(self): |
| 74 |
|
m = self.mock_logic |
| 75 |
|
m2 = self.mock_logic2 |
| 76 |
|
add_permission_logic(Article, m) |
| 77 |
|
add_permission_logic(Article, m2) |
| 78 |
|
self.assertEqual(Article._permission_logics, set([m, m2])) |
| 79 |
|
self.assertTrue(isinstance(registry._registry[Article], |
| 80 |
|
LogicalPermissionHandler)) |
| 81 |
|
|
| 82 |
|
# permission_logics should be changed but registry |
| 83 |
|
# should not be changed |
| 84 |
|
remove_permission_logic(Article, PermissionLogic) |
| 85 |
|
self.assertEqual(Article._permission_logics, set()) |
| 86 |
|
self.assertTrue(isinstance(registry._registry[Article], |
| 87 |
|
LogicalPermissionHandler)) |
| 88 |
|
|
| 89 |
|
def test_remove_permission_logic_exception(self): |
| 90 |
|
m = self.mock_logic |
| 91 |
|
add_permission_logic(Article, m) |
|
@@ 60-72 (lines=13) @@
|
| 57 |
|
self.assertTrue(hasattr(Article, '_permission_handler')) |
| 58 |
|
|
| 59 |
|
def test_remove_permission_logic_registry(self): |
| 60 |
|
m = self.mock_logic |
| 61 |
|
add_permission_logic(Article, m) |
| 62 |
|
self.assertEqual(Article._permission_logics, set([m])) |
| 63 |
|
self.assertTrue(isinstance(registry._registry[Article], |
| 64 |
|
LogicalPermissionHandler)) |
| 65 |
|
|
| 66 |
|
# permission_logics should be changed but registry |
| 67 |
|
# should not be changed |
| 68 |
|
remove_permission_logic(Article, m) |
| 69 |
|
self.assertEqual(Article._permission_logics, set()) |
| 70 |
|
self.assertTrue(isinstance(registry._registry[Article], |
| 71 |
|
LogicalPermissionHandler)) |
| 72 |
|
|
| 73 |
|
def test_remove_permission_logic_registry_with_class(self): |
| 74 |
|
m = self.mock_logic |
| 75 |
|
m2 = self.mock_logic2 |