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.
256 lines
8.5 KiB
256 lines
8.5 KiB
from datetime import timedelta as td
|
|
|
|
from django.test import Client
|
|
from django.test.utils import override_settings
|
|
from django.utils.timezone import now
|
|
from hc.api.models import Check, Flip, Ping
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class PingTestCase(BaseTestCase):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.check = Check.objects.create(project=self.project)
|
|
self.url = f"/ping/{self.check.code}"
|
|
|
|
def test_it_works(self):
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(r.headers["Access-Control-Allow-Origin"], "*")
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "up")
|
|
expected_aa = self.check.last_ping + td(days=1, hours=1)
|
|
self.assertEqual(self.check.alert_after, expected_aa)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.scheme, "http")
|
|
self.assertEqual(ping.kind, None)
|
|
self.assertEqual(ping.created, self.check.last_ping)
|
|
self.assertIsNone(ping.exitstatus)
|
|
|
|
def test_it_changes_status_of_paused_check(self):
|
|
self.check.status = "paused"
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "up")
|
|
|
|
def test_it_clears_last_start(self):
|
|
self.check.last_start = now()
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.last_start, None)
|
|
|
|
def test_post_works(self):
|
|
csrf_client = Client(enforce_csrf_checks=True)
|
|
r = csrf_client.post(self.url, "hello world", content_type="text/plain")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.method, "POST")
|
|
self.assertEqual(ping.body, "hello world")
|
|
|
|
def test_head_works(self):
|
|
csrf_client = Client(enforce_csrf_checks=True)
|
|
r = csrf_client.head(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(Ping.objects.count(), 1)
|
|
|
|
def test_it_handles_bad_uuid(self):
|
|
r = self.client.get("/ping/not-uuid/")
|
|
self.assertEqual(r.status_code, 404)
|
|
|
|
def test_it_rejects_alternative_uuid_formats(self):
|
|
# This uuid is missing separators. uuid.UUID() would accept it.
|
|
r = self.client.get("/ping/07c2f54898504b27af5d6c9dc157ec02/")
|
|
self.assertEqual(r.status_code, 404)
|
|
|
|
def test_it_handles_missing_check(self):
|
|
r = self.client.get("/ping/07c2f548-9850-4b27-af5d-6c9dc157ec02/")
|
|
self.assertEqual(r.status_code, 404)
|
|
self.assertEqual(r.content.decode(), "not found")
|
|
|
|
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"
|
|
)
|
|
|
|
r = self.client.get(self.url, HTTP_USER_AGENT=ua)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.ua, ua)
|
|
|
|
def test_it_truncates_long_ua(self):
|
|
ua = "01234567890" * 30
|
|
|
|
r = self.client.get(self.url, HTTP_USER_AGENT=ua)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(len(ping.ua), 200)
|
|
assert ua.startswith(ping.ua)
|
|
|
|
def test_it_reads_forwarded_ip(self):
|
|
ip = "1.1.1.1"
|
|
r = self.client.get(self.url, HTTP_X_FORWARDED_FOR=ip)
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(ping.remote_addr, "1.1.1.1")
|
|
|
|
def test_it_reads_first_forwarded_ip(self):
|
|
ip = "1.1.1.1, 2.2.2.2"
|
|
r = self.client.get(self.url, HTTP_X_FORWARDED_FOR=ip, REMOTE_ADDR="3.3.3.3",)
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(ping.remote_addr, "1.1.1.1")
|
|
|
|
def test_it_reads_forwarded_protocol(self):
|
|
r = self.client.get(self.url, HTTP_X_FORWARDED_PROTO="https")
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(ping.scheme, "https")
|
|
|
|
def test_it_never_caches(self):
|
|
r = self.client.get(self.url)
|
|
assert "no-cache" in r.get("Cache-Control")
|
|
|
|
def test_it_updates_confirmation_flag(self):
|
|
payload = "Please Confirm ..."
|
|
r = self.client.post(self.url, data=payload, content_type="text/plain")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertTrue(self.check.has_confirmation_link)
|
|
|
|
def test_fail_endpoint_works(self):
|
|
r = self.client.get(self.url + "/fail")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "down")
|
|
self.assertEqual(self.check.alert_after, None)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.kind, "fail")
|
|
|
|
flip = Flip.objects.get()
|
|
self.assertEqual(flip.owner, self.check)
|
|
self.assertEqual(flip.new_status, "down")
|
|
|
|
def test_start_endpoint_works(self):
|
|
last_ping = now() - td(hours=2)
|
|
self.check.last_ping = last_ping
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url + "/start")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertTrue(self.check.last_start)
|
|
self.assertEqual(self.check.last_ping, last_ping)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.kind, "start")
|
|
|
|
def test_start_does_not_change_status_of_paused_check(self):
|
|
self.check.status = "paused"
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url + "/start")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertTrue(self.check.last_start)
|
|
self.assertEqual(self.check.status, "paused")
|
|
|
|
def test_it_sets_last_duration(self):
|
|
self.check.last_start = now() - td(seconds=10)
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertTrue(self.check.last_duration.total_seconds() >= 10)
|
|
|
|
def test_it_requires_post(self):
|
|
self.check.methods = "POST"
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "new")
|
|
self.assertIsNone(self.check.last_ping)
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.scheme, "http")
|
|
self.assertEqual(ping.kind, "ign")
|
|
|
|
@override_settings(PING_BODY_LIMIT=5)
|
|
def test_it_chops_long_body(self):
|
|
self.client.post(self.url, "hello world", content_type="text/plain")
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.method, "POST")
|
|
self.assertEqual(ping.body, "hello")
|
|
|
|
@override_settings(PING_BODY_LIMIT=None)
|
|
def test_it_allows_unlimited_body(self):
|
|
self.client.post(self.url, "A" * 20000, content_type="text/plain")
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(len(ping.body), 20000)
|
|
|
|
def test_it_handles_manual_resume_flag(self):
|
|
self.check.status = "paused"
|
|
self.check.manual_resume = True
|
|
self.check.save()
|
|
|
|
r = self.client.get(self.url)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "paused")
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.scheme, "http")
|
|
self.assertEqual(ping.kind, "ign")
|
|
|
|
def test_zero_exit_status_works(self):
|
|
r = self.client.get(self.url + "/0")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "up")
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.kind, None)
|
|
self.assertEqual(ping.exitstatus, 0)
|
|
|
|
def test_nonzero_exit_status_works(self):
|
|
r = self.client.get(self.url + "/123")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.check.refresh_from_db()
|
|
self.assertEqual(self.check.status, "down")
|
|
|
|
ping = Ping.objects.get()
|
|
self.assertEqual(ping.kind, "fail")
|
|
self.assertEqual(ping.exitstatus, 123)
|
|
|
|
def test_it_rejects_exit_status_over_255(self):
|
|
r = self.client.get(self.url + "/256")
|
|
self.assertEqual(r.status_code, 400)
|