forked from GithubBackups/healthchecks
77 lines
2.4 KiB
Python
77 lines
2.4 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 AddTelegramTestCase(BaseTestCase):
|
|
url = "/integrations/add_telegram/"
|
|
|
|
def test_instructions_work(self):
|
|
self.client.login(username="alice@example.org", password="password")
|
|
r = self.client.get(self.url)
|
|
self.assertContains(r, "start@ExampleBot")
|
|
|
|
def test_it_shows_confirmation(self):
|
|
payload = signing.dumps((123, "group", "My Group"))
|
|
|
|
self.client.login(username="alice@example.org", password="password")
|
|
r = self.client.get(self.url + "?" + payload)
|
|
self.assertContains(r, "My Group")
|
|
|
|
def test_it_works(self):
|
|
payload = signing.dumps((123, "group", "My Group"))
|
|
|
|
self.client.login(username="alice@example.org", password="password")
|
|
r = self.client.post(self.url + "?" + payload, {})
|
|
self.assertRedirects(r, "/integrations/")
|
|
|
|
c = Channel.objects.get()
|
|
self.assertEqual(c.kind, "telegram")
|
|
self.assertEqual(c.telegram_id, 123)
|
|
self.assertEqual(c.telegram_type, "group")
|
|
self.assertEqual(c.telegram_name, "My Group")
|
|
|
|
@patch("hc.api.transports.requests.request")
|
|
def test_it_sends_invite(self, mock_get):
|
|
data = {
|
|
"message": {
|
|
"chat": {
|
|
"id": 123,
|
|
"title": "My Group",
|
|
"type": "group"
|
|
},
|
|
"text": "/start"
|
|
}
|
|
}
|
|
r = self.client.post("/integrations/telegram/bot/", json.dumps(data),
|
|
content_type="application/json")
|
|
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertTrue(mock_get.called)
|
|
|
|
@patch("hc.api.transports.requests.request")
|
|
def test_bot_handles_bad_message(self, mock_get):
|
|
samples = ["", "{}"]
|
|
|
|
# text is missing
|
|
samples.append(json.dumps({
|
|
"message": {"chat": {"id": 123, "type": "group"}}
|
|
}))
|
|
|
|
# bad type
|
|
samples.append(json.dumps({
|
|
"message": {
|
|
"chat": {"id": 123, "type": "invalid"},
|
|
"text": "/start"
|
|
}
|
|
}))
|
|
|
|
for sample in samples:
|
|
r = self.client.post("/integrations/telegram/bot/", sample,
|
|
content_type="application/json")
|
|
|
|
self.assertEqual(r.status_code, 400)
|