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

Fix test_telegram _init test

Browse Source
This commit is contained in:
Matthias 2023-04-16 17:18:28 +02:00
parent cf0b37057c
commit c9e6137ad0
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/rpc/telegram.py
View File

@ -166,6 +166,9 @@ class Telegram(RPCHandler):
logger.info('using custom keyboard from '
f'config.json: {self._keyboard}')
def _init_telegram_app(self):
return Application.builder().token(self._config['telegram']['token']).build()
def _init(self) -> None:
"""
Initializes this module with the given config,
@ -179,7 +182,7 @@ class Telegram(RPCHandler):
self._loop = asyncio.new_event_loop()
asyncio.set_event_loop(self._loop)
self._app = Application.builder().token(self._config['telegram']['token']).build()
self._app = self._init_telegram_app()
# Register command handler and start telegram message polling
handles = [

15
tests/rpc/test_rpc_telegram.py
View File

@ -105,15 +105,18 @@ def test_telegram__init__(default_conf, mocker) -> None:
def test_telegram_init(default_conf, mocker, caplog) -> None:
start_polling = MagicMock()
mocker.patch('freqtrade.rpc.telegram.Updater', MagicMock(return_value=start_polling))
app_mock = MagicMock()
mocker.patch('freqtrade.rpc.telegram.Telegram._start_thread', MagicMock())
mocker.patch('freqtrade.rpc.telegram.Telegram._init_telegram_app', return_value=app_mock)
mocker.patch('freqtrade.rpc.telegram.Telegram._startup_telegram', AsyncMock())
get_telegram_testobject(mocker, default_conf, mock=False)
assert start_polling.call_count == 0
telegram, _, _ = get_telegram_testobject(mocker, default_conf, mock=False)
telegram._init()
assert app_mock.call_count == 0
# number of handles registered
assert start_polling.dispatcher.add_handler.call_count > 0
assert start_polling.start_polling.call_count == 1
assert app_mock.add_handler.call_count > 0
# assert start_polling.start_polling.call_count == 1
message_str = ("rpc.telegram is listening for following commands: [['status'], ['profit'], "
"['balance'], ['start'], ['stop'], "