healthchecks/hc/api/tests/test_check_going_down_after.py

from datetime import timedelta as td

from django.test import TestCase
from django.utils import timezone
from hc.api.models import Check


class CheckModelTestCase(TestCase):
    def test_it_handles_new_check(self):
        check = Check()
        self.assertEqual(check.going_down_after(), None)

    def test_it_handles_paused_check(self):
        check = Check(status="paused")
        check.last_ping = timezone.now() - td(days=2)
        self.assertEqual(check.going_down_after(), None)

    def test_it_handles_up(self):
        check = Check(status="up")
        check.last_ping = timezone.now() - td(hours=1)
        expected_aa = check.last_ping + td(days=1, hours=1)
        self.assertEqual(check.going_down_after(), expected_aa)

    def test_it_handles_paused_then_started_check(self):
        check = Check(status="paused")
        check.last_start = timezone.now() - td(days=2)

        expected_aa = check.last_start + td(hours=1)
        self.assertEqual(check.going_down_after(), expected_aa)

    def test_it_handles_down(self):
        check = Check(status="down")
        check.last_ping = timezone.now() - td(hours=1)
        self.assertEqual(check.going_down_after(), None)

    def test_it_handles_down_then_started_check(self):
        check = Check(status="down")
        check.last_start = timezone.now() - td(minutes=10)

        self.assertEqual(check.going_down_after(), None)