def test_edit_get(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
with self.assertRaises(Http404):
admin.domain_admin_edit(request, 0)
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.context_data["form"].instance, self.domain)
def test_edit_get(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
with self.assertRaises(Http404):
admin.domain_admin_edit(request, 0)
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.context_data["form"].instance, self.domain)
def test_edit_post_invalid_owner(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
request.method = "POST"
request.POST = {"domain": self.domain.domain, "enabled": False, "owner": 0}
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 200)
self.domain.refresh_from_db()
self.assertEqual(self.domain.enabled, True)
self.assertEqual(self.domain.owner, None)
self.assertEqual(models.Domain.objects.count(), 1)
def test_edit_post(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
request.method = "POST"
request.POST = {"domain": self.domain.domain, "enabled": False}
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 302)
self.assertEqual(response["Location"], urlresolvers.reverse("admin:domains:index"))
self.domain.refresh_from_db()
self.assertEqual(self.domain.enabled, False)
self.assertEqual(models.Domain.objects.count(), 1)
def test_edit_post(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
request.method = "POST"
request.POST = {"domain": self.domain.domain, "enabled": False}
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 302)
self.assertEqual(response["Location"],
urlresolvers.reverse("admin:domains:index"))
self.domain.refresh_from_db()
self.assertEqual(self.domain.enabled, False)
self.assertEqual(models.Domain.objects.count(), 1)
def test_edit_post_invalid_owner(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
request.method = "POST"
request.POST = {
"domain": self.domain.domain,
"enabled": False,
"owner": 0
}
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 200)
self.domain.refresh_from_db()
self.assertEqual(self.domain.enabled, True)
self.assertEqual(self.domain.owner, None)
self.assertEqual(models.Domain.objects.count(), 1)
def test_edit_post_blank(self):
request = MockRequest(self.user, has_otp=True, has_sudo=True)
request.method = "POST"
request.POST = {}
self.domain.owner = self.user
self.domain.save()
response = admin.domain_admin_edit(request, self.domain.pk)
self.assertEqual(response.status_code, 302)
self.assertEqual(response["Location"],
urls.reverse("admin:domains:index"))
self.assertEqual(models.Domain.objects.count(), 1)
self.domain.refresh_from_db()
self.assertEqual(self.domain.owner, self.user)
