|
|
- from django.test import Client, TestCase
-
- from hc.api.models import Check, Ping
-
-
- class PingTestCase(TestCase):
-
- def test_it_works(self):
- check = Check()
- check.save()
-
- r = self.client.get("/ping/%s/" % check.code)
- assert r.status_code == 200
-
- same_check = Check.objects.get(code=check.code)
- assert same_check.status == "up"
-
- pings = list(Ping.objects.all())
- assert pings[0].scheme == "http"
-
- def test_post_works(self):
- check = Check()
- check.save()
-
- csrf_client = Client(enforce_csrf_checks=True)
- r = csrf_client.post("/ping/%s/" % check.code)
- assert r.status_code == 200
-
- def test_head_works(self):
- check = Check()
- check.save()
-
- csrf_client = Client(enforce_csrf_checks=True)
- r = csrf_client.head("/ping/%s/" % check.code)
- assert r.status_code == 200
- assert Ping.objects.count() == 1
-
- def test_it_handles_bad_uuid(self):
- r = self.client.get("/ping/not-uuid/")
- assert r.status_code == 400
-
- def test_it_handles_120_char_ua(self):
- ua = ("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_4) "
- "AppleWebKit/537.36 (KHTML, like Gecko) "
- "Chrome/44.0.2403.89 Safari/537.36")
-
- check = Check()
- check.save()
- r = self.client.get("/ping/%s/" % check.code, HTTP_USER_AGENT=ua)
- assert r.status_code == 200
-
- pings = list(Ping.objects.all())
- assert pings[0].ua == ua
-
- def test_it_truncates_long_ua(self):
- ua = "01234567890" * 30
-
- check = Check()
- check.save()
- r = self.client.get("/ping/%s/" % check.code, HTTP_USER_AGENT=ua)
- assert r.status_code == 200
-
- pings = list(Ping.objects.all())
- assert len(pings[0].ua) == 200
- assert ua.startswith(pings[0].ua)
|
