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_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)