forked from GithubBackups/healthchecks
264 lines
9.3 KiB
Python
264 lines
9.3 KiB
Python
from datetime import datetime, timedelta
|
|
from unittest.mock import Mock, patch
|
|
|
|
from django.utils import timezone
|
|
from hc.api.models import Check, Flip
|
|
from hc.test import BaseTestCase
|
|
|
|
CURRENT_TIME = datetime(2020, 1, 15, tzinfo=timezone.utc)
|
|
MOCK_NOW = Mock(return_value=CURRENT_TIME)
|
|
|
|
|
|
class CheckModelTestCase(BaseTestCase):
|
|
def test_it_strips_tags(self):
|
|
check = Check()
|
|
|
|
check.tags = " foo bar "
|
|
self.assertEqual(check.tags_list(), ["foo", "bar"])
|
|
|
|
check.tags = " "
|
|
self.assertEqual(check.tags_list(), [])
|
|
|
|
def test_get_status_handles_new_check(self):
|
|
check = Check()
|
|
self.assertEqual(check.get_status(), "new")
|
|
|
|
def test_status_works_with_grace_period(self):
|
|
check = Check()
|
|
check.status = "up"
|
|
check.last_ping = timezone.now() - timedelta(days=1, minutes=30)
|
|
|
|
self.assertEqual(check.get_status(), "grace")
|
|
|
|
def test_get_status_handles_paused_check(self):
|
|
check = Check()
|
|
check.status = "paused"
|
|
check.last_ping = timezone.now() - timedelta(days=1, minutes=30)
|
|
self.assertEqual(check.get_status(), "paused")
|
|
|
|
def test_status_works_with_cron_syntax(self):
|
|
dt = timezone.make_aware(datetime(2000, 1, 1), timezone=timezone.utc)
|
|
|
|
# Expect ping every midnight, default grace is 1 hour
|
|
check = Check()
|
|
check.kind = "cron"
|
|
check.schedule = "0 0 * * *"
|
|
check.status = "up"
|
|
check.last_ping = dt
|
|
|
|
# 23:59pm
|
|
now = dt + timedelta(hours=23, minutes=59)
|
|
self.assertEqual(check.get_status(now), "up")
|
|
|
|
# 00:00am
|
|
now = dt + timedelta(days=1)
|
|
self.assertEqual(check.get_status(now), "grace")
|
|
|
|
# 1:30am
|
|
now = dt + timedelta(days=1, minutes=60)
|
|
self.assertEqual(check.get_status(now), "down")
|
|
|
|
def test_status_works_with_timezone(self):
|
|
dt = timezone.make_aware(datetime(2000, 1, 1), timezone=timezone.utc)
|
|
|
|
# Expect ping every day at 10am, default grace is 1 hour
|
|
check = Check()
|
|
check.kind = "cron"
|
|
check.schedule = "0 10 * * *"
|
|
check.status = "up"
|
|
check.last_ping = dt
|
|
check.tz = "Australia/Brisbane" # UTC+10
|
|
|
|
# 10:30am
|
|
now = dt + timedelta(hours=23, minutes=59)
|
|
self.assertEqual(check.get_status(now), "up")
|
|
|
|
# 10:30am
|
|
now = dt + timedelta(days=1)
|
|
self.assertEqual(check.get_status(now), "grace")
|
|
|
|
# 11:30am
|
|
now = dt + timedelta(days=1, minutes=60)
|
|
self.assertEqual(check.get_status(now), "down")
|
|
|
|
def test_get_status_handles_past_grace(self):
|
|
check = Check()
|
|
check.status = "up"
|
|
check.last_ping = timezone.now() - timedelta(days=2)
|
|
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_get_status_obeys_down_status(self):
|
|
check = Check()
|
|
check.status = "down"
|
|
check.last_ping = timezone.now() - timedelta(minutes=1)
|
|
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_get_status_handles_started(self):
|
|
check = Check()
|
|
check.last_ping = timezone.now() - timedelta(hours=2)
|
|
# Last start was 5 minutes ago, display status should be "started"
|
|
check.last_start = timezone.now() - timedelta(minutes=5)
|
|
for status in ("new", "paused", "up", "down"):
|
|
check.status = status
|
|
self.assertEqual(check.get_status(with_started=True), "started")
|
|
|
|
def test_get_status_handles_down_then_started_and_expired(self):
|
|
check = Check(status="down")
|
|
# Last ping was 2 days ago
|
|
check.last_ping = timezone.now() - timedelta(days=2)
|
|
# Last start was 2 hours ago - the check is past its grace time
|
|
check.last_start = timezone.now() - timedelta(hours=2)
|
|
|
|
self.assertEqual(check.get_status(with_started=True), "down")
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_get_status_handles_up_then_started(self):
|
|
check = Check(status="up")
|
|
# Last ping was 2 hours ago, so is still up
|
|
check.last_ping = timezone.now() - timedelta(hours=2)
|
|
# Last start was 5 minutes ago
|
|
check.last_start = timezone.now() - timedelta(minutes=5)
|
|
|
|
self.assertEqual(check.get_status(with_started=True), "started")
|
|
# A started check still is considered "up":
|
|
self.assertEqual(check.get_status(), "up")
|
|
|
|
def test_get_status_handles_up_then_started_and_expired(self):
|
|
check = Check(status="up")
|
|
# Last ping was 3 hours ago, so is still up
|
|
check.last_ping = timezone.now() - timedelta(hours=3)
|
|
# Last start was 2 hours ago - the check is past its grace time
|
|
check.last_start = timezone.now() - timedelta(hours=2)
|
|
|
|
self.assertEqual(check.get_status(with_started=True), "down")
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_get_status_handles_paused_then_started_and_expired(self):
|
|
check = Check(status="paused")
|
|
# Last start was 2 hours ago - the check is past its grace time
|
|
check.last_start = timezone.now() - timedelta(hours=2)
|
|
|
|
self.assertEqual(check.get_status(with_started=True), "down")
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_get_status_handles_started_and_mia(self):
|
|
check = Check()
|
|
check.last_start = timezone.now() - timedelta(hours=2)
|
|
|
|
self.assertEqual(check.get_status(with_started=True), "down")
|
|
self.assertEqual(check.get_status(), "down")
|
|
|
|
def test_next_ping_with_cron_syntax(self):
|
|
dt = timezone.make_aware(datetime(2000, 1, 1), timezone=timezone.utc)
|
|
|
|
# Expect ping every round hour
|
|
check = Check(project=self.project)
|
|
check.kind = "cron"
|
|
check.schedule = "0 * * * *"
|
|
check.status = "up"
|
|
check.last_ping = dt
|
|
# Need to save it for M2M relations to work:
|
|
check.save()
|
|
|
|
d = check.to_dict()
|
|
self.assertEqual(d["next_ping"], "2000-01-01T01:00:00+00:00")
|
|
|
|
@patch("hc.api.models.timezone.now", MOCK_NOW)
|
|
def test_downtimes_handles_no_flips(self):
|
|
check = Check(project=self.project)
|
|
check.created = datetime(2019, 1, 1, tzinfo=timezone.utc)
|
|
|
|
nov, dec, jan = check.downtimes(3)
|
|
|
|
# Nov. 2019
|
|
self.assertEqual(nov[0].strftime("%m-%Y"), "11-2019")
|
|
self.assertEqual(nov[1], timedelta())
|
|
self.assertEqual(nov[2], 0)
|
|
|
|
# Dec. 2019
|
|
self.assertEqual(dec[0].strftime("%m-%Y"), "12-2019")
|
|
self.assertEqual(dec[1], timedelta())
|
|
self.assertEqual(dec[2], 0)
|
|
|
|
# Jan. 2020
|
|
self.assertEqual(jan[0].strftime("%m-%Y"), "01-2020")
|
|
self.assertEqual(jan[1], timedelta())
|
|
self.assertEqual(jan[2], 0)
|
|
|
|
@patch("hc.api.models.timezone.now", MOCK_NOW)
|
|
def test_downtimes_handles_currently_down_check(self):
|
|
check = Check(project=self.project, status="down")
|
|
check.created = datetime(2019, 1, 1, tzinfo=timezone.utc)
|
|
|
|
r = check.downtimes(10)
|
|
self.assertEqual(len(r), 10)
|
|
for dt, downtime, outages in r:
|
|
self.assertEqual(outages, 1)
|
|
|
|
@patch("hc.api.models.timezone.now", MOCK_NOW)
|
|
def test_downtimes_handles_flip_one_day_ago(self):
|
|
check = Check.objects.create(project=self.project, status="down")
|
|
check.created = datetime(2019, 1, 1, tzinfo=timezone.utc)
|
|
|
|
flip = Flip(owner=check)
|
|
flip.created = datetime(2020, 1, 14, tzinfo=timezone.utc)
|
|
flip.old_status = "up"
|
|
flip.new_status = "down"
|
|
flip.save()
|
|
|
|
r = check.downtimes(10)
|
|
self.assertEqual(len(r), 10)
|
|
for dt, downtime, outages in r:
|
|
if dt.month == 1:
|
|
self.assertEqual(downtime.total_seconds(), 86400)
|
|
self.assertEqual(outages, 1)
|
|
else:
|
|
self.assertEqual(downtime.total_seconds(), 0)
|
|
self.assertEqual(outages, 0)
|
|
|
|
@patch("hc.api.models.timezone.now", MOCK_NOW)
|
|
def test_downtimes_handles_flip_two_months_ago(self):
|
|
check = Check.objects.create(project=self.project, status="down")
|
|
check.created = datetime(2019, 1, 1, tzinfo=timezone.utc)
|
|
|
|
flip = Flip(owner=check)
|
|
flip.created = datetime(2019, 11, 15, tzinfo=timezone.utc)
|
|
flip.old_status = "up"
|
|
flip.new_status = "down"
|
|
flip.save()
|
|
|
|
r = check.downtimes(10)
|
|
self.assertEqual(len(r), 10)
|
|
for dt, downtime, outages in r:
|
|
if dt.month == 11:
|
|
self.assertEqual(outages, 1)
|
|
elif dt.month == 12:
|
|
self.assertEqual(downtime.total_seconds(), 31 * 86400)
|
|
self.assertEqual(outages, 1)
|
|
elif dt.month == 1:
|
|
self.assertEqual(outages, 1)
|
|
else:
|
|
self.assertEqual(downtime.total_seconds(), 0)
|
|
self.assertEqual(outages, 0)
|
|
|
|
@patch("hc.api.models.timezone.now", MOCK_NOW)
|
|
def test_downtimes_handles_months_when_check_did_not_exist(self):
|
|
check = Check(project=self.project)
|
|
check.created = datetime(2020, 1, 1, tzinfo=timezone.utc)
|
|
|
|
nov, dec, jan = check.downtimes(3)
|
|
|
|
# Nov. 2019
|
|
self.assertIsNone(nov[1])
|
|
self.assertIsNone(nov[2])
|
|
|
|
# Dec. 2019
|
|
self.assertIsNone(dec[1])
|
|
self.assertIsNone(dec[2])
|
|
|
|
# Jan. 2020
|
|
self.assertEqual(jan[1], timedelta())
|
|
self.assertEqual(jan[2], 0)
|