test_tasks.py 文件源码

def test_send_message_to_chat(self, mock_requests):
        parsed_update = parse_update(TelegramUpdates.TEXT_OK_ID_OK_TEXT)
        parsed_update['parse_mode'] = 'HTML'
        parsed_update['text'] = 'Sentiment'

        result = send_message_to_chat.delay(parsed_update).get(timeout=5)

        # on success the Message sent to the chat is returned
        self.assertEqual(result['text'], parsed_update['text'])
        self.assertEqual(result['parse_mode'], parsed_update['parse_mode'])
        self.assertEqual(result['reply_to_message_id'], parsed_update['reply_to_message_id'])
        self.assertEqual(result['chat_id'], parsed_update['chat_id'])

        mock_requests.assert_called()
        self.assertIn(call(current_app.config['TELEGRAM_URL'] + 'sendMessage',
                      json=parsed_update,
                      timeout=current_app.config['TELEGRAM_REQUEST_TIMEOUT_SEC']),
                      mock_requests.call_args_list)