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)
