@@ 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 |