from datetime import timedelta
|
|
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from hc.api.models import Check
|
|
|
|
|
|
class CheckModelTestCase(TestCase):
|
|
|
|
def test_it_strips_tags(self):
|
|
check = Check()
|
|
|
|
check.tags = " foo bar "
|
|
self.assertEquals(check.tags_list(), ["foo", "bar"])
|
|
|
|
check.tags = " "
|
|
self.assertEquals(check.tags_list(), [])
|
|
|
|
def test_in_grace_period_handles_new_check(self):
|
|
check = Check()
|
|
self.assertFalse(check.in_grace_period())
|
|
|
|
def test_status_works_with_grace_period(self):
|
|
check = Check()
|
|
check.status = "up"
|
|
check.last_ping = timezone.now() - timedelta(days=1, minutes=30)
|
|
|
|
self.assertTrue(check.in_grace_period())
|
|
self.assertEqual(check.get_status(), "up")
|