lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update telegram "cleanup" test

Browse Source
This commit is contained in:
Matthias 2023-04-16 16:58:12 +02:00
parent 69f61ef767
commit cf0b37057c
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/rpc/test_rpc_telegram.py
View File

@ -2,6 +2,7 @@
# pragma pylint: disable=protected-access, unused-argument, invalid-name # pragma pylint: disable=protected-access, unused-argument, invalid-name
# pragma pylint: disable=too-many-lines, too-many-arguments # pragma pylint: disable=too-many-lines, too-many-arguments
import asyncio
import logging import logging
import re import re
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone
@ -130,14 +131,24 @@ def test_telegram_init(default_conf, mocker, caplog) -> None:
assert log_has(message_str, caplog) assert log_has(message_str, caplog)
def test_cleanup(default_conf, mocker, ) -> None: async def test_telegram_cleanup(default_conf, mocker, ) -> None:
updater_mock = MagicMock() app_mock = MagicMock()
updater_mock.stop = MagicMock() app_mock.stop = AsyncMock()
mocker.patch('freqtrade.rpc.telegram.Updater', updater_mock) app_mock.initialize = AsyncMock()
telegram, _, _ = get_telegram_testobject(mocker, default_conf, mock=False) updater_mock = MagicMock()
updater_mock.stop = AsyncMock()
app_mock.updater = updater_mock
# mocker.patch('freqtrade.rpc.telegram.Application', app_mock)
telegram, _, _ = get_telegram_testobject(mocker, default_conf)
telegram._app = app_mock
telegram._loop = asyncio.get_running_loop()
telegram._thread = MagicMock()
telegram.cleanup() telegram.cleanup()
assert telegram._app.stop.call_count == 1 await asyncio.sleep(0.1)
assert app_mock.stop.call_count == 1
assert telegram._thread.join.call_count == 1
async def test_authorized_only(default_conf, mocker, caplog, update) -> None: async def test_authorized_only(default_conf, mocker, caplog, update) -> None: