diff --git a/modoboa/admin/tests/test_domain.py b/modoboa/admin/tests/test_domain.py index a9e8cb394..b3eeabfeb 100644 --- a/modoboa/admin/tests/test_domain.py +++ b/modoboa/admin/tests/test_domain.py @@ -308,6 +308,14 @@ class DomainTestCase(ModoTestCase): self.assertEqual(response.status_code, 200) self.assertIn("Usage", response.content.decode()) + user = User.objects.get(username="user@test.com") + self.client.force_login(user) + domain = Domain.objects.get(name="test2.com") + url = reverse("admin:domain_detail", args=[domain.pk]) + response = self.client.get(url) + self.assertEqual(response.status_code, 302) + self.assertTrue(response.url.startswith("/accounts/login")) + def test_domain_quota_list_view(self): """Test quota list view.""" factories.DomainFactory(name="test3.com", quota=100) diff --git a/modoboa/admin/views/domain.py b/modoboa/admin/views/domain.py index 30adcecd1..8fada95f7 100644 --- a/modoboa/admin/views/domain.py +++ b/modoboa/admin/views/domain.py @@ -205,7 +205,7 @@ class DomainDetailView( def get_queryset(self): """Add some prefetching.""" return ( - super(DomainDetailView, self).get_queryset() + Domain.objects.get_for_admin(self.request.user) .prefetch_related("domainalias_set", "mailbox_set", "alias_set") )