def test__send_text_message_shoul_call_sendMessage_api_correctly(self):
tcb = TelegramController()
tcb.post_worker = MagicMock()
tcb.TOKEN = "banana"
tcb.CHAT_ID = "to_banana"
tcb._send_http = MagicMock()
tcb._send_text_message("hello~ banana")
tcb.post_worker.post_task.assert_called_once_with(ANY)
task = tcb.post_worker.post_task.call_args[0][0]
tcb.post_worker.post_task.call_args[0][0]["runnable"](task)
tcb._send_http.assert_called_once_with(
"https://api.telegram.org/banana/sendMessage?chat_id=to_banana&text=hello%7E%20banana"
)
def test__handle_message_call__execute_command_with_correct_commands(self):
tcb = TelegramController()
tcb.CHAT_ID = 1234567890
tcb._execute_command = MagicMock()
tcb._get_updates = MagicMock(
return_value={
"ok":
True,
"result": [
{
"update_id": 402107588,
"message": {
"message_id": 11,
"from": {
"id": 1234567890,
"is_bot": False,
"first_name": "msaltnet",
"language_code": "ko",
},
"chat": {
"id": 1234567890,
"first_name": "msaltnet",
"type": "private"
},
"date": 1627694419,
"text": "3",
},
},
{
"update_id": 402107589,
"message": {
"message_id": 12,
"from": {
"id": 1234567891,
"is_bot": False,
"first_name": "msaltnet",
"language_code": "ko",
},
"chat": {
"id": 1234567891,
"first_name": "msaltnet",
"type": "private"
},
"date": 1627694420,
"text": "4",
},
},
{
"update_id": 402107590,
"message": {
"message_id": 13,
"from": {
"id": 1234567890,
"is_bot": False,
"first_name": "msaltnet",
"language_code": "ko",
},
"chat": {
"id": 1234567890,
"first_name": "msaltnet",
"type": "private"
},
"date": 1627694420,
"text": "5",
},
},
],
})
tcb._handle_message()
self.assertEqual(tcb._execute_command.call_args_list[0][0][0], "3")
self.assertEqual(tcb._execute_command.call_args_list[1][0][0], "5")
self.assertEqual(tcb.last_update_id, 402107590)