diff options
Diffstat (limited to 'tests/modeladmin/test_actions.py')
| -rw-r--r-- | tests/modeladmin/test_actions.py | 75 |
1 files changed, 41 insertions, 34 deletions
diff --git a/tests/modeladmin/test_actions.py b/tests/modeladmin/test_actions.py index b61641c0c9..fa3108ce08 100644 --- a/tests/modeladmin/test_actions.py +++ b/tests/modeladmin/test_actions.py @@ -7,16 +7,23 @@ from .models import Band class AdminActionsTests(TestCase): - @classmethod def setUpTestData(cls): - cls.superuser = User.objects.create_superuser(username='super', password='secret', email='super@example.com') + cls.superuser = User.objects.create_superuser( + username="super", password="secret", email="super@example.com" + ) content_type = ContentType.objects.get_for_model(Band) - Permission.objects.create(name='custom', codename='custom_band', content_type=content_type) - for user_type in ('view', 'add', 'change', 'delete', 'custom'): - username = '%suser' % user_type - user = User.objects.create_user(username=username, password='secret', is_staff=True) - permission = Permission.objects.get(codename='%s_band' % user_type, content_type=content_type) + Permission.objects.create( + name="custom", codename="custom_band", content_type=content_type + ) + for user_type in ("view", "add", "change", "delete", "custom"): + username = "%suser" % user_type + user = User.objects.create_user( + username=username, password="secret", is_staff=True + ) + permission = Permission.objects.get( + codename="%s_band" % user_type, content_type=content_type + ) user.user_permissions.add(permission) setattr(cls, username, user) @@ -25,31 +32,31 @@ class AdminActionsTests(TestCase): pass class BandAdmin(admin.ModelAdmin): - actions = ['custom_action'] + actions = ["custom_action"] @admin.action def custom_action(modeladmin, request, queryset): pass def has_custom_permission(self, request): - return request.user.has_perm('%s.custom_band' % self.opts.app_label) + return request.user.has_perm("%s.custom_band" % self.opts.app_label) ma = BandAdmin(Band, admin.AdminSite()) mock_request = MockRequest() mock_request.GET = {} cases = [ - (None, self.viewuser, ['custom_action']), - ('view', self.superuser, ['delete_selected', 'custom_action']), - ('view', self.viewuser, ['custom_action']), - ('add', self.adduser, ['custom_action']), - ('change', self.changeuser, ['custom_action']), - ('delete', self.deleteuser, ['delete_selected', 'custom_action']), - ('custom', self.customuser, ['custom_action']), + (None, self.viewuser, ["custom_action"]), + ("view", self.superuser, ["delete_selected", "custom_action"]), + ("view", self.viewuser, ["custom_action"]), + ("add", self.adduser, ["custom_action"]), + ("change", self.changeuser, ["custom_action"]), + ("delete", self.deleteuser, ["delete_selected", "custom_action"]), + ("custom", self.customuser, ["custom_action"]), ] for permission, user, expected in cases: with self.subTest(permission=permission, user=user): if permission is None: - if hasattr(BandAdmin.custom_action, 'allowed_permissions'): + if hasattr(BandAdmin.custom_action, "allowed_permissions"): del BandAdmin.custom_action.allowed_permissions else: BandAdmin.custom_action.allowed_permissions = (permission,) @@ -59,7 +66,7 @@ class AdminActionsTests(TestCase): def test_actions_inheritance(self): class AdminBase(admin.ModelAdmin): - actions = ['custom_action'] + actions = ["custom_action"] @admin.action def custom_action(modeladmin, request, queryset): @@ -73,14 +80,14 @@ class AdminActionsTests(TestCase): ma1 = AdminA(Band, admin.AdminSite()) action_names = [name for _, name, _ in ma1._get_base_actions()] - self.assertEqual(action_names, ['delete_selected', 'custom_action']) + self.assertEqual(action_names, ["delete_selected", "custom_action"]) # `actions = None` removes actions from superclasses. ma2 = AdminB(Band, admin.AdminSite()) action_names = [name for _, name, _ in ma2._get_base_actions()] - self.assertEqual(action_names, ['delete_selected']) + self.assertEqual(action_names, ["delete_selected"]) def test_global_actions_description(self): - @admin.action(description='Site-wide admin action 1.') + @admin.action(description="Site-wide admin action 1.") def global_action_1(modeladmin, request, queryset): pass @@ -99,32 +106,32 @@ class AdminActionsTests(TestCase): self.assertEqual( [description for _, _, description in ma._get_base_actions()], [ - 'Delete selected %(verbose_name_plural)s', - 'Site-wide admin action 1.', - 'Global action 2', + "Delete selected %(verbose_name_plural)s", + "Site-wide admin action 1.", + "Global action 2", ], ) def test_actions_replace_global_action(self): - @admin.action(description='Site-wide admin action 1.') + @admin.action(description="Site-wide admin action 1.") def global_action_1(modeladmin, request, queryset): pass - @admin.action(description='Site-wide admin action 2.') + @admin.action(description="Site-wide admin action 2.") def global_action_2(modeladmin, request, queryset): pass - admin.site.add_action(global_action_1, name='custom_action_1') - admin.site.add_action(global_action_2, name='custom_action_2') + admin.site.add_action(global_action_1, name="custom_action_1") + admin.site.add_action(global_action_2, name="custom_action_2") - @admin.action(description='Local admin action 1.') + @admin.action(description="Local admin action 1.") def custom_action_1(modeladmin, request, queryset): pass class BandAdmin(admin.ModelAdmin): - actions = [custom_action_1, 'custom_action_2'] + actions = [custom_action_1, "custom_action_2"] - @admin.action(description='Local admin action 2.') + @admin.action(description="Local admin action 2.") def custom_action_2(self, request, queryset): pass @@ -134,10 +141,10 @@ class AdminActionsTests(TestCase): [ desc for _, name, desc in ma._get_base_actions() - if name.startswith('custom_action') + if name.startswith("custom_action") ], [ - 'Local admin action 1.', - 'Local admin action 2.', + "Local admin action 1.", + "Local admin action 2.", ], ) |