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.
48 lines
1.9 KiB
48 lines
1.9 KiB
import json
|
|
from datetime import timedelta as td
|
|
|
|
from hc.api.models import Check, User
|
|
from hc.test import BaseTestCase
|
|
from hc.accounts.models import Profile
|
|
|
|
class ListChecksTestCase(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(ListChecksTestCase, self).setUp()
|
|
self.profile = Profile(user=self.alice, api_key="abc")
|
|
self.profile.save()
|
|
self.checks = [
|
|
Check(user=self.alice, name="Alice 1", timeout=td(seconds=3600), grace=td(seconds=900)),
|
|
Check(user=self.alice, name="Alice 2", timeout=td(seconds=86400), grace=td(seconds=3600)),
|
|
]
|
|
for check in self.checks:
|
|
check.save()
|
|
|
|
def get(self, url, data):
|
|
return self.client.generic('GET', url, json.dumps(data), 'application/json')
|
|
|
|
def test_it_works(self):
|
|
r = self.get("/api/v1/checks/", { "api_key": "abc" })
|
|
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertTrue("checks" in r.json())
|
|
self.assertEqual(len(r.json()["checks"]), 2)
|
|
|
|
checks = { check["name"]: check for check in r.json()["checks"] }
|
|
self.assertEqual(checks["Alice 1"]["timeout"], 3600)
|
|
self.assertEqual(checks["Alice 1"]["grace"], 900)
|
|
self.assertEqual(checks["Alice 1"]["ping_url"], self.checks[0].url())
|
|
self.assertEqual(checks["Alice 2"]["timeout"], 86400)
|
|
self.assertEqual(checks["Alice 2"]["grace"], 3600)
|
|
self.assertEqual(checks["Alice 2"]["ping_url"], self.checks[1].url())
|
|
|
|
def test_it_shows_only_users_checks(self):
|
|
bob = User(username="bob", email="[email protected]")
|
|
bob.save()
|
|
bob_check = Check(user=bob, name="Bob 1")
|
|
|
|
r = self.get("/api/v1/checks/", { "api_key": "abc" })
|
|
|
|
self.assertEqual(len(r.json()["checks"]), 2)
|
|
checks = { check["name"]: check for check in r.json()["checks"] }
|
|
self.assertNotIn("Bob 1", checks)
|