|
|
- import json
-
- from hc.api.models import Channel
- from hc.test import BaseTestCase
-
-
- class ChannelModelTestCase(BaseTestCase):
- def test_webhook_spec_handles_mixed(self):
- c = Channel(kind="webhook")
- c.value = json.dumps(
- {
- "method_down": "GET",
- "url_down": "http://example.org",
- "body_down": "",
- "headers_down": {"X-Status": "X"},
- "method_up": "POST",
- "url_up": "http://example.org/up/",
- "body_up": "hello world",
- "headers_up": {"X-Status": "OK"},
- }
- )
-
- self.assertEqual(
- c.down_webhook_spec,
- {
- "method": "GET",
- "url": "http://example.org",
- "body": "",
- "headers": {"X-Status": "X"},
- },
- )
-
- self.assertEqual(
- c.up_webhook_spec,
- {
- "method": "POST",
- "url": "http://example.org/up/",
- "body": "hello world",
- "headers": {"X-Status": "OK"},
- },
- )
-
- def test_it_handles_legacy_opsgenie_value(self):
- c = Channel(kind="opsgenie", value="foo123")
- self.assertEqual(c.opsgenie_key, "foo123")
- self.assertEqual(c.opsgenie_region, "us")
-
- def test_it_handles_json_opsgenie_value(self):
- c = Channel(kind="opsgenie")
- c.value = json.dumps({"key": "abc", "region": "eu"})
- self.assertEqual(c.opsgenie_key, "abc")
- self.assertEqual(c.opsgenie_region, "eu")
-
- def test_it_handles_legacy_sms_value(self):
- c = Channel(kind="sms", value="+123123123")
- self.assertTrue(c.sms_notify_down)
- self.assertFalse(c.sms_notify_up)
-
- def test_it_handles_legacy_sms_json_value(self):
- c = Channel(kind="sms", value=json.dumps({"value": "+123123123"}))
- self.assertTrue(c.sms_notify_down)
- self.assertFalse(c.sms_notify_up)
|
