# coding: utf-8 from datetime import timedelta as td from unittest.mock import patch from django.utils.timezone import now from hc.api.models import Channel, Check, Notification from hc.test import BaseTestCase class NotifyLineTestCase(BaseTestCase): def setUp(self): super().setUp() self.check = Check(project=self.project) self.check.name = "Foo" self.check.status = "down" self.check.last_ping = now() - td(minutes=61) self.check.save() self.channel = Channel(project=self.project) self.channel.kind = "linenotify" self.channel.value = "fake-token" self.channel.save() self.channel.checks.add(self.check) @patch("hc.api.transports.requests.request") def test_it_works(self, mock_post): mock_post.return_value.status_code = 200 self.channel.notify(self.check) assert Notification.objects.count() == 1 args, kwargs = mock_post.call_args headers = kwargs["headers"] params = kwargs["params"] self.assertEqual(headers["Authorization"], "Bearer fake-token") self.assertIn("""The check "Foo" is DOWN""", params["message"])