mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions 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
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/rpc/telegram.py
+4 -1
View File
@@ -166,6 +166,9 @@ class Telegram(RPCHandler):
logger.info('using custom keyboard from ' logger.info('using custom keyboard from '
f'config.json: {self._keyboard}') f'config.json: {self._keyboard}')
def _init_telegram_app(self):
return Application.builder().token(self._config['telegram']['token']).build()
def _init(self) -> None: def _init(self) -> None:
""" """
Initializes this module with the given config, Initializes this module with the given config,
@@ -179,7 +182,7 @@ class Telegram(RPCHandler):
self._loop = asyncio.new_event_loop() self._loop = asyncio.new_event_loop()
asyncio.set_event_loop(self._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 # Register command handler and start telegram message polling
handles = [ handles = [
tests/rpc/test_rpc_telegram.py
+9 -6
View File
@@ -105,15 +105,18 @@ def test_telegram__init__(default_conf, mocker) -> None:
def test_telegram_init(default_conf, mocker, caplog) -> None: def test_telegram_init(default_conf, mocker, caplog) -> None:
start_polling = MagicMock() app_mock = MagicMock()
mocker.patch('freqtrade.rpc.telegram.Updater', MagicMock(return_value=start_polling)) 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) telegram, _, _ = get_telegram_testobject(mocker, default_conf, mock=False)
assert start_polling.call_count == 0 telegram._init()
assert app_mock.call_count == 0
# number of handles registered # number of handles registered
assert start_polling.dispatcher.add_handler.call_count > 0 assert app_mock.add_handler.call_count > 0
assert start_polling.start_polling.call_count == 1 # assert start_polling.start_polling.call_count == 1
message_str = ("rpc.telegram is listening for following commands: [['status'], ['profit'], " message_str = ("rpc.telegram is listening for following commands: [['status'], ['profit'], "
"['balance'], ['start'], ['stop'], " "['balance'], ['start'], ['stop'], "