631df64c68
Refactors the existing RPC directory, classes, defs as below. This is to better reflect the nature of the files/classes/defs to support other clients - such as rest The code is functional, ive tested every RPC button in Telegram. 1 unit test is failing ( they all were till find_replace) in test_rpc_telegram. Can I ask for anothers opinion on why this 1 test fails, im at loss to debug - mocker is stretching my working knowledge of python - Ive renamed common methods from RPC_method_name to server_method_name e.g rpc_stop becomes server_stop, the super class is CLIENTS not RPC as it is agnostic to RPC freqtrade/rcp/ - rpc.py ......................- rpc_manager.py ......................- telegram.py to: freqtrade/clients/client_manager.py (was rpc_manager ......................-common/client.py (was rpc.py - but is agnostic to what client calls) ......................-rpc/telegram.py ......................-rpc/*discord.py (example future RPC client ......................-rest/*cmd line.py (example future rest cmdline rest client ......................-rest/*web.py (example future rest web client
143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
"""
|
|
Unit test file for main.py
|
|
"""
|
|
|
|
import logging
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from freqtrade import OperationalException
|
|
from freqtrade.main import main, set_loggers
|
|
from freqtrade.tests.conftest import log_has
|
|
|
|
|
|
def test_parse_args_backtesting(mocker) -> None:
|
|
"""
|
|
Test that main() can start backtesting and also ensure we can pass some specific arguments
|
|
further argument parsing is done in test_arguments.py
|
|
"""
|
|
backtesting_mock = mocker.patch('freqtrade.optimize.backtesting.start', MagicMock())
|
|
main(['backtesting'])
|
|
assert backtesting_mock.call_count == 1
|
|
call_args = backtesting_mock.call_args[0][0]
|
|
assert call_args.config == 'config.json'
|
|
assert call_args.live is False
|
|
assert call_args.loglevel == 20
|
|
assert call_args.subparser == 'backtesting'
|
|
assert call_args.func is not None
|
|
assert call_args.ticker_interval is None
|
|
|
|
|
|
def test_main_start_hyperopt(mocker) -> None:
|
|
"""
|
|
Test that main() can start hyperopt
|
|
"""
|
|
hyperopt_mock = mocker.patch('freqtrade.optimize.hyperopt.start', MagicMock())
|
|
main(['hyperopt'])
|
|
assert hyperopt_mock.call_count == 1
|
|
call_args = hyperopt_mock.call_args[0][0]
|
|
assert call_args.config == 'config.json'
|
|
assert call_args.loglevel == 20
|
|
assert call_args.subparser == 'hyperopt'
|
|
assert call_args.func is not None
|
|
|
|
|
|
def test_set_loggers() -> None:
|
|
"""
|
|
Test set_loggers() update the logger level for third-party libraries
|
|
"""
|
|
previous_value1 = logging.getLogger('requests.packages.urllib3').level
|
|
previous_value2 = logging.getLogger('telegram').level
|
|
|
|
set_loggers()
|
|
|
|
value1 = logging.getLogger('requests.packages.urllib3').level
|
|
assert previous_value1 is not value1
|
|
assert value1 is logging.INFO
|
|
|
|
value2 = logging.getLogger('telegram').level
|
|
assert previous_value2 is not value2
|
|
assert value2 is logging.INFO
|
|
|
|
|
|
def test_main_fatal_exception(mocker, default_conf, caplog) -> None:
|
|
"""
|
|
Test main() function
|
|
In this test we are skipping the while True loop by throwing an exception.
|
|
"""
|
|
mocker.patch.multiple(
|
|
'freqtrade.freqtradebot.FreqtradeBot',
|
|
_init_modules=MagicMock(),
|
|
worker=MagicMock(side_effect=Exception),
|
|
clean=MagicMock(),
|
|
)
|
|
mocker.patch(
|
|
'freqtrade.configuration.Configuration._load_config_file',
|
|
lambda *args, **kwargs: default_conf
|
|
)
|
|
mocker.patch('freqtrade.freqtradebot.CryptoToFiatConverter', MagicMock())
|
|
mocker.patch('freqtrade.freqtradebot.ClientManager', MagicMock())
|
|
|
|
args = ['-c', 'config.json.example']
|
|
|
|
# Test Main + the KeyboardInterrupt exception
|
|
with pytest.raises(SystemExit):
|
|
main(args)
|
|
assert log_has('Using config: config.json.example ...', caplog.record_tuples)
|
|
assert log_has('Fatal exception!', caplog.record_tuples)
|
|
|
|
|
|
def test_main_keyboard_interrupt(mocker, default_conf, caplog) -> None:
|
|
"""
|
|
Test main() function
|
|
In this test we are skipping the while True loop by throwing an exception.
|
|
"""
|
|
mocker.patch.multiple(
|
|
'freqtrade.freqtradebot.FreqtradeBot',
|
|
_init_modules=MagicMock(),
|
|
worker=MagicMock(side_effect=KeyboardInterrupt),
|
|
clean=MagicMock(),
|
|
)
|
|
mocker.patch(
|
|
'freqtrade.configuration.Configuration._load_config_file',
|
|
lambda *args, **kwargs: default_conf
|
|
)
|
|
mocker.patch('freqtrade.freqtradebot.CryptoToFiatConverter', MagicMock())
|
|
mocker.patch('freqtrade.freqtradebot.ClientManager', MagicMock())
|
|
|
|
args = ['-c', 'config.json.example']
|
|
|
|
# Test Main + the KeyboardInterrupt exception
|
|
with pytest.raises(SystemExit):
|
|
main(args)
|
|
assert log_has('Using config: config.json.example ...', caplog.record_tuples)
|
|
assert log_has('SIGINT received, aborting ...', caplog.record_tuples)
|
|
|
|
|
|
def test_main_operational_exception(mocker, default_conf, caplog) -> None:
|
|
"""
|
|
Test main() function
|
|
In this test we are skipping the while True loop by throwing an exception.
|
|
"""
|
|
mocker.patch.multiple(
|
|
'freqtrade.freqtradebot.FreqtradeBot',
|
|
_init_modules=MagicMock(),
|
|
worker=MagicMock(side_effect=OperationalException('Oh snap!')),
|
|
clean=MagicMock(),
|
|
)
|
|
mocker.patch(
|
|
'freqtrade.configuration.Configuration._load_config_file',
|
|
lambda *args, **kwargs: default_conf
|
|
)
|
|
mocker.patch('freqtrade.freqtradebot.CryptoToFiatConverter', MagicMock())
|
|
mocker.patch('freqtrade.freqtradebot.ClientManager', MagicMock())
|
|
|
|
args = ['-c', 'config.json.example']
|
|
|
|
# Test Main + the KeyboardInterrupt exception
|
|
with pytest.raises(SystemExit):
|
|
main(args)
|
|
assert log_has('Using config: config.json.example ...', caplog.record_tuples)
|
|
assert log_has('Oh snap!', caplog.record_tuples)
|