let parse_args only parse, no continuation
This removes parse_args() from the call stack It pushes down the test-mocking one level [from parse_args() to main()]. Moves parse_args into a more generic 'modules' parsing direction.
This commit is contained in:
parent
e6e57e47cf
commit
984204e380
@ -399,15 +399,18 @@ def cleanup() -> None:
|
|||||||
exit(0)
|
exit(0)
|
||||||
|
|
||||||
|
|
||||||
def main() -> None:
|
def main(sysargv=sys.argv[1:]) -> None:
|
||||||
"""
|
"""
|
||||||
Loads and validates the config and handles the main loop
|
Loads and validates the config and handles the main loop
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
global _CONF
|
global _CONF
|
||||||
args = parse_args(sys.argv[1:],
|
args = parse_args(sysargv,
|
||||||
'Simple High Frequency Trading Bot for crypto currencies')
|
'Simple High Frequency Trading Bot for crypto currencies')
|
||||||
if not args:
|
|
||||||
|
# A subcommand has been issued
|
||||||
|
if hasattr(args, 'func'):
|
||||||
|
args.func(args)
|
||||||
exit(0)
|
exit(0)
|
||||||
|
|
||||||
# Initialize logger
|
# Initialize logger
|
||||||
|
@ -136,14 +136,7 @@ def parse_args(args: List[str], description: str):
|
|||||||
)
|
)
|
||||||
|
|
||||||
build_subcommands(parser)
|
build_subcommands(parser)
|
||||||
parsed_args = parser.parse_args(args)
|
return parser.parse_args(args)
|
||||||
|
|
||||||
# No subcommand as been selected
|
|
||||||
if not hasattr(parsed_args, 'func'):
|
|
||||||
return parsed_args
|
|
||||||
|
|
||||||
parsed_args.func(parsed_args)
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def build_subcommands(parser: argparse.ArgumentParser) -> None:
|
def build_subcommands(parser: argparse.ArgumentParser) -> None:
|
||||||
|
@ -15,6 +15,39 @@ from freqtrade.main import create_trade, handle_trade, init, \
|
|||||||
get_target_bid, _process, execute_sell, check_handle_timedout
|
get_target_bid, _process, execute_sell, check_handle_timedout
|
||||||
from freqtrade.misc import get_state, State
|
from freqtrade.misc import get_state, State
|
||||||
from freqtrade.persistence import Trade
|
from freqtrade.persistence import Trade
|
||||||
|
import freqtrade.main as main
|
||||||
|
|
||||||
|
|
||||||
|
# Test that main() can start backtesting or hyperopt.
|
||||||
|
# and also ensure we can pass some specific arguments
|
||||||
|
# argument parsing is done in test_misc.py
|
||||||
|
|
||||||
|
def test_parse_args_backtesting(mocker):
|
||||||
|
backtesting_mock = mocker.patch(
|
||||||
|
'freqtrade.optimize.backtesting.start', MagicMock())
|
||||||
|
with pytest.raises(SystemExit, match=r'0'):
|
||||||
|
main.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 == 5
|
||||||
|
|
||||||
|
|
||||||
|
def test_main_start_hyperopt(mocker):
|
||||||
|
hyperopt_mock = mocker.patch(
|
||||||
|
'freqtrade.optimize.hyperopt.start', MagicMock())
|
||||||
|
with pytest.raises(SystemExit, match=r'0'):
|
||||||
|
main.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_process_trade_creation(default_conf, ticker, limit_buy_order, health, mocker):
|
def test_process_trade_creation(default_conf, ticker, limit_buy_order, health, mocker):
|
||||||
|
@ -3,7 +3,6 @@ import json
|
|||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
from unittest.mock import MagicMock
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from jsonschema import ValidationError
|
from jsonschema import ValidationError
|
||||||
@ -101,22 +100,6 @@ def test_parse_args_dynamic_whitelist_invalid_values():
|
|||||||
parse_args(['--dynamic-whitelist', 'abc'], '')
|
parse_args(['--dynamic-whitelist', 'abc'], '')
|
||||||
|
|
||||||
|
|
||||||
def test_parse_args_backtesting(mocker):
|
|
||||||
backtesting_mock = mocker.patch(
|
|
||||||
'freqtrade.optimize.backtesting.start', 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():
|
def test_parse_args_backtesting_invalid():
|
||||||
with pytest.raises(SystemExit, match=r'2'):
|
with pytest.raises(SystemExit, match=r'2'):
|
||||||
parse_args(['backtesting --ticker-interval'], '')
|
parse_args(['backtesting --ticker-interval'], '')
|
||||||
@ -125,19 +108,14 @@ def test_parse_args_backtesting_invalid():
|
|||||||
parse_args(['backtesting --ticker-interval', 'abc'], '')
|
parse_args(['backtesting --ticker-interval', 'abc'], '')
|
||||||
|
|
||||||
|
|
||||||
def test_parse_args_backtesting_custom(mocker):
|
def test_parse_args_backtesting_custom():
|
||||||
backtesting_mock = mocker.patch(
|
args = [
|
||||||
'freqtrade.optimize.backtesting.start', MagicMock())
|
|
||||||
args = parse_args([
|
|
||||||
'-c', 'test_conf.json',
|
'-c', 'test_conf.json',
|
||||||
'backtesting',
|
'backtesting',
|
||||||
'--live',
|
'--live',
|
||||||
'--ticker-interval', '1',
|
'--ticker-interval', '1',
|
||||||
'--refresh-pairs-cached'], '')
|
'--refresh-pairs-cached']
|
||||||
assert args is None
|
call_args = parse_args(args, '')
|
||||||
assert backtesting_mock.call_count == 1
|
|
||||||
|
|
||||||
call_args = backtesting_mock.call_args[0][0]
|
|
||||||
assert call_args.config == 'test_conf.json'
|
assert call_args.config == 'test_conf.json'
|
||||||
assert call_args.live is True
|
assert call_args.live is True
|
||||||
assert call_args.loglevel == 20
|
assert call_args.loglevel == 20
|
||||||
@ -147,28 +125,9 @@ def test_parse_args_backtesting_custom(mocker):
|
|||||||
assert call_args.refresh_pairs is True
|
assert call_args.refresh_pairs is True
|
||||||
|
|
||||||
|
|
||||||
def test_parse_args_hyperopt(mocker):
|
|
||||||
hyperopt_mock = mocker.patch(
|
|
||||||
'freqtrade.optimize.hyperopt.start', MagicMock())
|
|
||||||
args = parse_args(['hyperopt'], '')
|
|
||||||
assert args is None
|
|
||||||
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_parse_args_hyperopt_custom(mocker):
|
def test_parse_args_hyperopt_custom(mocker):
|
||||||
hyperopt_mock = mocker.patch(
|
args = ['-c', 'test_conf.json', 'hyperopt', '--epochs', '20']
|
||||||
'freqtrade.optimize.hyperopt.start', MagicMock())
|
call_args = parse_args(args, '')
|
||||||
args = parse_args(['-c', 'test_conf.json', 'hyperopt', '--epochs', '20'], '')
|
|
||||||
assert args is None
|
|
||||||
assert hyperopt_mock.call_count == 1
|
|
||||||
|
|
||||||
call_args = hyperopt_mock.call_args[0][0]
|
|
||||||
assert call_args.config == 'test_conf.json'
|
assert call_args.config == 'test_conf.json'
|
||||||
assert call_args.epochs == 20
|
assert call_args.epochs == 20
|
||||||
assert call_args.loglevel == 20
|
assert call_args.loglevel == 20
|
||||||
|
Loading…
Reference in New Issue
Block a user