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.
55 lines
2.0 KiB
55 lines
2.0 KiB
from django.test.utils import override_settings
|
|
from django.utils.timezone import now
|
|
from hc.api.models import Channel, Check, Flip, Notification, Ping
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
@override_settings(METRICS_KEY="foo")
|
|
class MetricsTestCase(BaseTestCase):
|
|
url = "/api/v1/metrics/"
|
|
|
|
def test_it_returns_num_unprocessed_flips(self):
|
|
check = Check.objects.create(project=self.project, status="down")
|
|
flip = Flip(owner=check)
|
|
flip.created = now()
|
|
flip.old_status = "up"
|
|
flip.new_status = "down"
|
|
flip.save()
|
|
|
|
r = self.client.get(self.url, HTTP_X_METRICS_KEY="foo")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
doc = r.json()
|
|
self.assertEqual(doc["num_unprocessed_flips"], 1)
|
|
|
|
def test_it_returns_max_ping_id(self):
|
|
check = Check.objects.create(project=self.project, status="down")
|
|
Ping.objects.create(owner=check, n=1)
|
|
last_ping = Ping.objects.last()
|
|
|
|
r = self.client.get(self.url, HTTP_X_METRICS_KEY="foo")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
doc = r.json()
|
|
self.assertEqual(doc["max_ping_id"], last_ping.id)
|
|
|
|
def test_it_returns_max_notification_id(self):
|
|
check = Check.objects.create(project=self.project, status="down")
|
|
channel = Channel.objects.create(project=self.project, kind="email")
|
|
Notification.objects.create(owner=check, channel=channel, check_status="down")
|
|
last_notification = Notification.objects.last()
|
|
|
|
r = self.client.get(self.url, HTTP_X_METRICS_KEY="foo")
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
doc = r.json()
|
|
self.assertEqual(doc["max_notification_id"], last_notification.id)
|
|
|
|
@override_settings(METRICS_KEY=None)
|
|
def test_it_handles_unset_metrics_key(self):
|
|
r = self.client.get(self.url, HTTP_X_METRICS_KEY="foo")
|
|
self.assertEqual(r.status_code, 403)
|
|
|
|
def test_it_handles_incorrect_metrics_key(self):
|
|
r = self.client.get(self.url, HTTP_X_METRICS_KEY="bar")
|
|
self.assertEqual(r.status_code, 403)
|