def _initialize_notifiers(self): if global_config_map.get("telegram_enabled").value: # TODO: refactor to use single instance if not any([isinstance(n, TelegramNotifier) for n in self.notifiers]): self.notifiers.append(TelegramNotifier(token=global_config_map["telegram_token"].value, chat_id=global_config_map["telegram_chat_id"].value, hb=self)) for notifier in self.notifiers: notifier.start()
class TelegramNotifierTests(TestCase): def setUp(self) -> None: super().setUp() self.app = MagicMock() self.telegram = TelegramNotifier(token="000:token", chat_id="", hb=None) @patch("telegram.ext.Updater.stop") @patch("telegram.ext.Updater.start_polling") def test_start_after_stop(self, start_polling: MagicMock, stop: MagicMock): self.telegram.start() start_polling.assert_called_once() start_polling.reset_mock() self.telegram.stop() stop.assert_called_once() self.telegram.start() start_polling.assert_called_once()
def setUp(self) -> None: super().setUp() self.app = MagicMock() self.telegram = TelegramNotifier(token="000:token", chat_id="", hb=None)