def testSendMessage(self): bot = TelegramBot('FAKE-TOKEN') params = { 'chat_id': 'a', 'text': 'b'} bot.telegramBot = Mock() bot.telegramBot.sendMessage = Mock() self.assertIsNone(bot.sendMessage(**params)) bot.telegramBot.sendMessage.assert_called_once_with(**params)
def testGetInfo(self): bot = TelegramBot('FAKE-TOKEN') info = Mock() bot.telegramBot = Mock() bot.telegramBot.getMe = Mock(return_value=info) infoResult = bot.getInfo() self.assertEqual(infoResult, info)