forked from GithubBackups/healthchecks
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
import json
|
|
|
|
from django.core import signing
|
|
from hc.api.models import Channel
|
|
from hc.test import BaseTestCase
|
|
from mock import patch
|
|
|
|
|
|
class AddHipChatTestCase(BaseTestCase):
|
|
url = "/integrations/add_hipchat/"
|
|
|
|
def test_instructions_work(self):
|
|
self.client.login(username="alice@example.org", password="password")
|
|
r = self.client.get(self.url)
|
|
self.assertContains(r, "appropriate HipChat room")
|
|
|
|
def test_instructions_work_when_logged_out(self):
|
|
r = self.client.get(self.url)
|
|
self.assertContains(r, "Before adding HipChat integration, please")
|
|
|
|
def test_it_redirects_to_addons_install(self):
|
|
self.client.login(username="alice@example.org", password="password")
|
|
r = self.client.post(self.url)
|
|
self.assertEqual(r.status_code, 302)
|
|
|
|
def test_it_returns_capabilities(self):
|
|
r = self.client.get("/integrations/hipchat/capabilities/")
|
|
self.assertContains(r, "callbackUrl")
|
|
|
|
@patch("hc.api.models.Channel.refresh_hipchat_access_token")
|
|
def test_callback_works(self, mock_refresh):
|
|
state = signing.TimestampSigner().sign("alice")
|
|
payload = json.dumps({"relayState": state, "foo": "foobar"})
|
|
|
|
r = self.client.post("/integrations/hipchat/callback/", payload,
|
|
content_type="application/json")
|
|
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
c = Channel.objects.get()
|
|
self.assertEqual(c.kind, "hipchat")
|
|
self.assertTrue("foobar" in c.value)
|
|
|
|
@patch("hc.api.models.Channel.refresh_hipchat_access_token")
|
|
def test_callback_rejects_bad_signature(self, mock_refresh):
|
|
payload = json.dumps({"relayState": "alice:bad:sig", "foo": "foobar"})
|
|
|
|
r = self.client.post("/integrations/hipchat/callback/", payload,
|
|
content_type="application/json")
|
|
|
|
self.assertEqual(r.status_code, 400)
|