def test__start_get_updates_loop_start_thread_correctly(self, mock_thread):
dummy_thread = MagicMock()
mock_thread.return_value = dummy_thread
tcb = TelegramController()
tcb._handle_message = MagicMock()
tcb.terminating = True # for Test
tcb._start_get_updates_loop()
dummy_thread.start.assert_called()
self.assertEqual(mock_thread.call_args[1]["name"], "get updates")
self.assertEqual(mock_thread.call_args[1]["daemon"], True)
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)