diff options
Diffstat (limited to 'tests/admin_utils')
| -rw-r--r-- | tests/admin_utils/admin.py | 16 | ||||
| -rw-r--r-- | tests/admin_utils/test_logentry.py | 29 | ||||
| -rw-r--r-- | tests/admin_utils/urls.py | 3 |
3 files changed, 46 insertions, 2 deletions
diff --git a/tests/admin_utils/admin.py b/tests/admin_utils/admin.py index 967f37faf3..a7cfcfa6b9 100644 --- a/tests/admin_utils/admin.py +++ b/tests/admin_utils/admin.py @@ -35,3 +35,19 @@ site = admin.AdminSite(name="admin") site.register(Article) site.register(ArticleProxy) site.register(Site, SiteAdmin) + + +class CustomAdminSite(admin.AdminSite): + def get_log_entries(self, request): + from django.contrib.contenttypes.models import ContentType + + log_entries = super().get_log_entries(request) + return log_entries.filter( + content_type__in=ContentType.objects.get_for_models( + *self._registry.keys() + ).values() + ) + + +custom_site = CustomAdminSite(name="custom_admin") +custom_site.register(Article) diff --git a/tests/admin_utils/test_logentry.py b/tests/admin_utils/test_logentry.py index dfa7962b1e..b700fe54b9 100644 --- a/tests/admin_utils/test_logentry.py +++ b/tests/admin_utils/test_logentry.py @@ -10,7 +10,7 @@ from django.urls import reverse from django.utils import translation from django.utils.html import escape -from .models import Article, ArticleProxy, Site +from .models import Article, ArticleProxy, Car, Site @override_settings(ROOT_URLCONF="admin_utils.urls") @@ -318,3 +318,30 @@ class LogEntryTests(TestCase): with self.subTest(action_flag=action_flag): log = LogEntry(action_flag=action_flag) self.assertEqual(log.get_action_flag_display(), display_name) + + def test_hook_get_log_entries(self): + LogEntry.objects.log_action( + self.user.pk, + ContentType.objects.get_for_model(Article).pk, + self.a1.pk, + "Article changed", + CHANGE, + change_message="Article changed message", + ) + c1 = Car.objects.create() + LogEntry.objects.log_action( + self.user.pk, + ContentType.objects.get_for_model(Car).pk, + c1.pk, + "Car created", + ADDITION, + change_message="Car created message", + ) + response = self.client.get(reverse("admin:index")) + self.assertContains(response, "Article changed") + self.assertContains(response, "Car created") + + # site "custom_admin" only renders log entries of registered models + response = self.client.get(reverse("custom_admin:index")) + self.assertContains(response, "Article changed") + self.assertNotContains(response, "Car created") diff --git a/tests/admin_utils/urls.py b/tests/admin_utils/urls.py index 1b6a97e4bd..69db6afc46 100644 --- a/tests/admin_utils/urls.py +++ b/tests/admin_utils/urls.py @@ -1,7 +1,8 @@ from django.urls import path -from .admin import site +from .admin import custom_site, site urlpatterns = [ path("test_admin/admin/", site.urls), + path("test_admin/custom_admin/", custom_site.urls), ] |
