diff options
Diffstat (limited to 'tests/admin_views/test_adminsite.py')
| -rw-r--r-- | tests/admin_views/test_adminsite.py | 72 |
1 files changed, 39 insertions, 33 deletions
diff --git a/tests/admin_views/test_adminsite.py b/tests/admin_views/test_adminsite.py index 4874bc48c8..68a32567d8 100644 --- a/tests/admin_views/test_adminsite.py +++ b/tests/admin_views/test_adminsite.py @@ -12,71 +12,76 @@ site.register(User) site.register(Article) urlpatterns = [ - path('test_admin/admin/', site.urls), + path("test_admin/admin/", site.urls), ] -@override_settings(ROOT_URLCONF='admin_views.test_adminsite') +@override_settings(ROOT_URLCONF="admin_views.test_adminsite") class SiteEachContextTest(TestCase): """ Check each_context contains the documented variables and that available_apps context variable structure is the expected one. """ + request_factory = RequestFactory() @classmethod def setUpTestData(cls): - cls.u1 = User.objects.create_superuser(username='super', password='secret', email='super@example.com') + cls.u1 = User.objects.create_superuser( + username="super", password="secret", email="super@example.com" + ) def setUp(self): - request = self.request_factory.get(reverse('test_adminsite:index')) + request = self.request_factory.get(reverse("test_adminsite:index")) request.user = self.u1 self.ctx = site.each_context(request) def test_each_context(self): ctx = self.ctx - self.assertEqual(ctx['site_header'], 'Django administration') - self.assertEqual(ctx['site_title'], 'Django site admin') - self.assertEqual(ctx['site_url'], '/') - self.assertIs(ctx['has_permission'], True) + self.assertEqual(ctx["site_header"], "Django administration") + self.assertEqual(ctx["site_title"], "Django site admin") + self.assertEqual(ctx["site_url"], "/") + self.assertIs(ctx["has_permission"], True) def test_each_context_site_url_with_script_name(self): - request = self.request_factory.get(reverse('test_adminsite:index'), SCRIPT_NAME='/my-script-name/') + request = self.request_factory.get( + reverse("test_adminsite:index"), SCRIPT_NAME="/my-script-name/" + ) request.user = self.u1 - self.assertEqual(site.each_context(request)['site_url'], '/my-script-name/') + self.assertEqual(site.each_context(request)["site_url"], "/my-script-name/") def test_available_apps(self): ctx = self.ctx - apps = ctx['available_apps'] + apps = ctx["available_apps"] # we have registered two models from two different apps self.assertEqual(len(apps), 2) # admin_views.Article admin_views = apps[0] - self.assertEqual(admin_views['app_label'], 'admin_views') - self.assertEqual(len(admin_views['models']), 1) - article = admin_views['models'][0] - self.assertEqual(article['object_name'], 'Article') - self.assertEqual(article['model'], Article) + self.assertEqual(admin_views["app_label"], "admin_views") + self.assertEqual(len(admin_views["models"]), 1) + article = admin_views["models"][0] + self.assertEqual(article["object_name"], "Article") + self.assertEqual(article["model"], Article) # auth.User auth = apps[1] - self.assertEqual(auth['app_label'], 'auth') - self.assertEqual(len(auth['models']), 1) - user = auth['models'][0] - self.assertEqual(user['object_name'], 'User') - self.assertEqual(user['model'], User) + self.assertEqual(auth["app_label"], "auth") + self.assertEqual(len(auth["models"]), 1) + user = auth["models"][0] + self.assertEqual(user["object_name"], "User") + self.assertEqual(user["model"], User) - self.assertEqual(auth['app_url'], '/test_admin/admin/auth/') - self.assertIs(auth['has_module_perms'], True) + self.assertEqual(auth["app_url"], "/test_admin/admin/auth/") + self.assertIs(auth["has_module_perms"], True) - self.assertIn('perms', user) - self.assertIs(user['perms']['add'], True) - self.assertIs(user['perms']['change'], True) - self.assertIs(user['perms']['delete'], True) - self.assertEqual(user['admin_url'], '/test_admin/admin/auth/user/') - self.assertEqual(user['add_url'], '/test_admin/admin/auth/user/add/') - self.assertEqual(user['name'], 'Users') + self.assertIn("perms", user) + self.assertIs(user["perms"]["add"], True) + self.assertIs(user["perms"]["change"], True) + self.assertIs(user["perms"]["delete"], True) + self.assertEqual(user["admin_url"], "/test_admin/admin/auth/user/") + self.assertEqual(user["add_url"], "/test_admin/admin/auth/user/add/") + self.assertEqual(user["name"], "Users") class SiteActionsTests(SimpleTestCase): @@ -86,11 +91,12 @@ class SiteActionsTests(SimpleTestCase): def test_add_action(self): def test_action(): pass + self.site.add_action(test_action) - self.assertEqual(self.site.get_action('test_action'), test_action) + self.assertEqual(self.site.get_action("test_action"), test_action) def test_disable_action(self): - action_name = 'delete_selected' + action_name = "delete_selected" self.assertEqual(self.site._actions[action_name], delete_selected) self.site.disable_action(action_name) with self.assertRaises(KeyError): @@ -98,7 +104,7 @@ class SiteActionsTests(SimpleTestCase): def test_get_action(self): """AdminSite.get_action() returns an action even if it's disabled.""" - action_name = 'delete_selected' + action_name = "delete_selected" self.assertEqual(self.site.get_action(action_name), delete_selected) self.site.disable_action(action_name) self.assertEqual(self.site.get_action(action_name), delete_selected) |