healthchecks/hc/front/tests/test_add_check.py
2016-12-21 17:22:48 +02:00

28 lines
913 B
Python

from hc.api.models import Check
from hc.test import BaseTestCase
class AddCheckTestCase(BaseTestCase):
def test_it_works(self):
url = "/checks/add/"
self.client.login(username="alice@example.org", password="password")
r = self.client.post(url)
self.assertRedirects(r, "/checks/")
assert Check.objects.count() == 1
def test_team_access_works(self):
url = "/checks/add/"
self.client.login(username="bob@example.org", password="password")
self.client.post(url)
check = Check.objects.get()
# Added by bob, but should belong to alice (bob has team access)
self.assertEqual(check.user, self.alice)
def test_it_rejects_get(self):
url = "/checks/add/"
self.client.login(username="alice@example.org", password="password")
r = self.client.get(url)
self.assertEqual(r.status_code, 400)