You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.8 KiB
64 lines
1.8 KiB
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)
|