mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

telegram: ensure msg update test uses the correct edit_message_text call

Browse Source
This commit is contained in:
Matthias
2024-02-12 06:31:39 +01:00
committed by Joe Schr
parent 3341964417
commit 0796f49761
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/rpc/test_rpc_telegram.py
+8 -8
View File
@@ -2557,22 +2557,22 @@ async def test_telegram__send_msg(default_conf, mocker, caplog) -> None:
# Test update # Test update
query = MagicMock() query = MagicMock()
query.edit_message_text = AsyncMock()
await telegram._send_msg('test', callback_path="DeadBeef", query=query, reload_able=True) await telegram._send_msg('test', callback_path="DeadBeef", query=query, reload_able=True)
edit_message_text = telegram._app.bot.edit_message_text assert query.edit_message_text.call_count == 1
assert edit_message_text.call_count == 1 assert "Updated: " in query.edit_message_text.call_args_list[0][1]['text']
assert "Updated: " in edit_message_text.call_args_list[0][1]['text']
telegram._app.bot.edit_message_text = AsyncMock(side_effect=BadRequest("not modified")) query.edit_message_text = AsyncMock(side_effect=BadRequest("not modified"))
await telegram._send_msg('test', callback_path="DeadBeef", query=query) await telegram._send_msg('test', callback_path="DeadBeef", query=query)
assert telegram._app.bot.edit_message_text.call_count == 1 assert query.edit_message_text.call_count == 1
assert not log_has_re(r"TelegramError: .*", caplog) assert not log_has_re(r"TelegramError: .*", caplog)
telegram._app.bot.edit_message_text = AsyncMock(side_effect=BadRequest("")) query.edit_message_text = AsyncMock(side_effect=BadRequest(""))
await telegram._send_msg('test2', callback_path="DeadBeef", query=query) await telegram._send_msg('test2', callback_path="DeadBeef", query=query)
assert telegram._app.bot.edit_message_text.call_count == 1 assert query.edit_message_text.call_count == 1
assert log_has_re(r"TelegramError: .*", caplog) assert log_has_re(r"TelegramError: .*", caplog)
telegram._app.bot.edit_message_text = AsyncMock(side_effect=TelegramError("DeadBEEF")) query.edit_message_text = AsyncMock(side_effect=TelegramError("DeadBEEF"))
await telegram._send_msg('test3', callback_path="DeadBeef", query=query) await telegram._send_msg('test3', callback_path="DeadBeef", query=query)
assert log_has_re(r"TelegramError: DeadBEEF! Giving up.*", caplog) assert log_has_re(r"TelegramError: DeadBEEF! Giving up.*", caplog)