|
|
- 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)
|
