add argparse handling tests
This commit is contained in:
parent
b682262486
commit
3f4e4a23a0
@ -1,7 +1,11 @@
|
|||||||
# pragma pylint: disable=missing-docstring
|
# pragma pylint: disable=missing-docstring
|
||||||
import time
|
import time
|
||||||
|
from argparse import Namespace
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
from freqtrade.misc import throttle
|
import pytest
|
||||||
|
|
||||||
|
from freqtrade.misc import throttle, parse_args, start_backtesting
|
||||||
|
|
||||||
|
|
||||||
def test_throttle():
|
def test_throttle():
|
||||||
@ -18,3 +22,99 @@ def test_throttle():
|
|||||||
|
|
||||||
result = throttle(func, -1)
|
result = throttle(func, -1)
|
||||||
assert result == 42
|
assert result == 42
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_defaults():
|
||||||
|
args = parse_args([])
|
||||||
|
assert args is not None
|
||||||
|
assert args.config == 'config.json'
|
||||||
|
assert args.dynamic_whitelist is False
|
||||||
|
assert args.loglevel == 20
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_invalid():
|
||||||
|
with pytest.raises(SystemExit, match=r'2'):
|
||||||
|
parse_args(['-c'])
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_config():
|
||||||
|
args = parse_args(['-c', '/dev/null'])
|
||||||
|
assert args is not None
|
||||||
|
assert args.config == '/dev/null'
|
||||||
|
|
||||||
|
args = parse_args(['--config', '/dev/null'])
|
||||||
|
assert args is not None
|
||||||
|
assert args.config == '/dev/null'
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_verbose():
|
||||||
|
args = parse_args(['-v'])
|
||||||
|
assert args is not None
|
||||||
|
assert args.loglevel == 10
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_dynamic_whitelist():
|
||||||
|
args = parse_args(['--dynamic-whitelist'])
|
||||||
|
assert args is not None
|
||||||
|
assert args.dynamic_whitelist is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_backtesting(mocker):
|
||||||
|
backtesting_mock = mocker.patch('freqtrade.misc.start_backtesting', MagicMock())
|
||||||
|
args = parse_args(['backtesting'])
|
||||||
|
assert args is None
|
||||||
|
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 == 5
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_backtesting_invalid():
|
||||||
|
with pytest.raises(SystemExit, match=r'2'):
|
||||||
|
parse_args(['--ticker-interval'])
|
||||||
|
|
||||||
|
with pytest.raises(SystemExit, match=r'2'):
|
||||||
|
parse_args(['--ticker-interval', 'abc'])
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_args_backtesting_custom(mocker):
|
||||||
|
backtesting_mock = mocker.patch('freqtrade.misc.start_backtesting', MagicMock())
|
||||||
|
args = parse_args(['-c', 'test_conf.json', 'backtesting', '--live', '--ticker-interval', '1'])
|
||||||
|
assert args is None
|
||||||
|
assert backtesting_mock.call_count == 1
|
||||||
|
|
||||||
|
call_args = backtesting_mock.call_args[0][0]
|
||||||
|
assert call_args.config == 'test_conf.json'
|
||||||
|
assert call_args.live is True
|
||||||
|
assert call_args.loglevel == 20
|
||||||
|
assert call_args.subparser == 'backtesting'
|
||||||
|
assert call_args.func is not None
|
||||||
|
assert call_args.ticker_interval == 1
|
||||||
|
|
||||||
|
|
||||||
|
def test_start_backtesting(mocker):
|
||||||
|
pytest_mock = mocker.patch('pytest.main', MagicMock())
|
||||||
|
env_mock = mocker.patch('os.environ', {})
|
||||||
|
args = Namespace(
|
||||||
|
config='config.json',
|
||||||
|
live=True,
|
||||||
|
loglevel=20,
|
||||||
|
ticker_interval=1,
|
||||||
|
)
|
||||||
|
start_backtesting(args)
|
||||||
|
assert env_mock == {
|
||||||
|
'BACKTEST': 'true',
|
||||||
|
'BACKTEST_LIVE': 'true',
|
||||||
|
'BACKTEST_CONFIG': 'config.json',
|
||||||
|
'BACKTEST_TICKER_INTERVAL': '1',
|
||||||
|
}
|
||||||
|
assert pytest_mock.call_count == 1
|
||||||
|
|
||||||
|
main_call_args = pytest_mock.call_args[0][0]
|
||||||
|
assert main_call_args[0] == '-s'
|
||||||
|
assert main_call_args[1].endswith('freqtrade/tests/test_backtesting.py')
|
||||||
|
Loading…
Reference in New Issue
Block a user