|
|
- import json
-
- from hc.api.models import Channel, Check
- from hc.test import BaseTestCase
-
-
- class EditWebhookTestCase(BaseTestCase):
- def setUp(self):
- super().setUp()
-
- self.check = Check.objects.create(project=self.project)
-
- definition = {
- "method_down": "GET",
- "url_down": "http://example.org/down",
- "body_down": "$NAME is down",
- "headers_down": {"User-Agent": "My-Custom-UA"},
- "method_up": "GET",
- "url_up": "http://example.org/up",
- "body_up": "$NAME is up",
- "headers_up": {},
- }
-
- self.channel = Channel(project=self.project, kind="webhook")
- self.channel.name = "Call example.org"
- self.channel.value = json.dumps(definition)
- self.channel.save()
-
- self.url = f"/integrations/{self.channel.code}/edit/"
-
- def test_it_shows_form(self):
- self.client.login(username="[email protected]", password="password")
- r = self.client.get(self.url)
- self.assertContains(r, "Webhook Settings")
-
- self.assertContains(r, "Call example.org")
-
- # down
- self.assertContains(r, "http://example.org/down")
- self.assertContains(r, "My-Custom-UA")
- self.assertContains(r, "$NAME is down")
-
- # up
- self.assertContains(r, "http://example.org/up")
- self.assertContains(r, "$NAME is up")
-
- def test_it_saves_form_and_redirects(self):
- form = {
- "name": "Call foo.com / bar.com",
- "method_down": "POST",
- "url_down": "http://foo.com",
- "headers_down": "X-Foo: 1\nX-Bar: 2",
- "body_down": "going down",
- "method_up": "POST",
- "url_up": "https://bar.com",
- "headers_up": "Content-Type: text/plain",
- "body_up": "going up",
- }
-
- self.client.login(username="[email protected]", password="password")
- r = self.client.post(self.url, form)
- self.assertRedirects(r, self.channels_url)
-
- self.channel.refresh_from_db()
- self.assertEqual(self.channel.name, "Call foo.com / bar.com")
-
- down_spec = self.channel.down_webhook_spec
- self.assertEqual(down_spec["method"], "POST")
- self.assertEqual(down_spec["url"], "http://foo.com")
- self.assertEqual(down_spec["body"], "going down")
- self.assertEqual(down_spec["headers"], {"X-Foo": "1", "X-Bar": "2"})
-
- up_spec = self.channel.up_webhook_spec
- self.assertEqual(up_spec["method"], "POST")
- self.assertEqual(up_spec["url"], "https://bar.com")
- self.assertEqual(up_spec["body"], "going up")
- self.assertEqual(up_spec["headers"], {"Content-Type": "text/plain"})
-
- # Make sure it does not call assign_all_checks
- self.assertFalse(self.channel.checks.exists())
-
- def test_it_requires_kind_webhook(self):
- self.channel.kind = "shell"
- self.channel.save()
-
- self.client.login(username="[email protected]", password="password")
- r = self.client.get(self.url)
- self.assertEqual(r.status_code, 400)
-
- def test_it_requires_rw_access(self):
- self.bobs_membership.role = "r"
- self.bobs_membership.save()
-
- self.client.login(username="[email protected]", password="password")
- r = self.client.post(self.url, {})
- self.assertEqual(r.status_code, 403)
|
