mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-14 12:13:57 +00:00
134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
# pragma pylint: disable=missing-docstring, C0103, C0330
|
|
# pragma pylint: disable=protected-access, too-many-lines, invalid-name, too-many-arguments
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from freqtrade.commands.optimize_commands import setup_optimize_configuration, start_edge
|
|
from freqtrade.enums import RunMode
|
|
from freqtrade.optimize.edge_cli import EdgeCli
|
|
from tests.conftest import (
|
|
CURRENT_TEST_STRATEGY,
|
|
EXMS,
|
|
get_args,
|
|
log_has,
|
|
patch_exchange,
|
|
patched_configuration_load_config_file,
|
|
)
|
|
|
|
|
|
def test_setup_optimize_configuration_without_arguments(mocker, default_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
|
|
args = [
|
|
"edge",
|
|
"--config",
|
|
"config.json",
|
|
"--strategy",
|
|
CURRENT_TEST_STRATEGY,
|
|
]
|
|
|
|
config = setup_optimize_configuration(get_args(args), RunMode.EDGE)
|
|
assert config["runmode"] == RunMode.EDGE
|
|
|
|
assert "max_open_trades" in config
|
|
assert "stake_currency" in config
|
|
assert "stake_amount" in config
|
|
assert "exchange" in config
|
|
assert "pair_whitelist" in config["exchange"]
|
|
assert "datadir" in config
|
|
assert log_has("Using data directory: {} ...".format(config["datadir"]), caplog)
|
|
assert "timeframe" in config
|
|
|
|
assert "timerange" not in config
|
|
assert "stoploss_range" not in config
|
|
|
|
|
|
def test_setup_edge_configuration_with_arguments(mocker, edge_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, edge_conf)
|
|
mocker.patch("freqtrade.configuration.configuration.create_datadir", lambda c, x: x)
|
|
|
|
args = [
|
|
"edge",
|
|
"--config",
|
|
"config.json",
|
|
"--strategy",
|
|
CURRENT_TEST_STRATEGY,
|
|
"--datadir",
|
|
"/foo/bar",
|
|
"--timeframe",
|
|
"1m",
|
|
"--timerange",
|
|
":100",
|
|
"--stoplosses=-0.01,-0.10,-0.001",
|
|
]
|
|
|
|
config = setup_optimize_configuration(get_args(args), RunMode.EDGE)
|
|
assert "max_open_trades" in config
|
|
assert "stake_currency" in config
|
|
assert "stake_amount" in config
|
|
assert "exchange" in config
|
|
assert "pair_whitelist" in config["exchange"]
|
|
assert "datadir" in config
|
|
assert config["runmode"] == RunMode.EDGE
|
|
assert log_has("Using data directory: {} ...".format(config["datadir"]), caplog)
|
|
assert "timeframe" in config
|
|
assert log_has("Parameter -i/--timeframe detected ... Using timeframe: 1m ...", caplog)
|
|
|
|
assert "timerange" in config
|
|
assert log_has("Parameter --timerange detected: {} ...".format(config["timerange"]), caplog)
|
|
|
|
|
|
def test_start(mocker, fee, edge_conf, caplog) -> None:
|
|
start_mock = MagicMock()
|
|
mocker.patch(f"{EXMS}.get_fee", fee)
|
|
patch_exchange(mocker)
|
|
mocker.patch("freqtrade.optimize.edge_cli.EdgeCli.start", start_mock)
|
|
patched_configuration_load_config_file(mocker, edge_conf)
|
|
|
|
args = [
|
|
"edge",
|
|
"--config",
|
|
"config.json",
|
|
"--strategy",
|
|
CURRENT_TEST_STRATEGY,
|
|
]
|
|
pargs = get_args(args)
|
|
start_edge(pargs)
|
|
assert log_has("Starting freqtrade in Edge mode", caplog)
|
|
assert start_mock.call_count == 1
|
|
|
|
|
|
def test_edge_init(mocker, edge_conf) -> None:
|
|
patch_exchange(mocker)
|
|
edge_conf["stake_amount"] = 20
|
|
edge_cli = EdgeCli(edge_conf)
|
|
assert edge_cli.config == edge_conf
|
|
assert edge_cli.config["stake_amount"] == "unlimited"
|
|
assert callable(edge_cli.edge.calculate)
|
|
assert edge_cli.strategy.bot_started is True
|
|
|
|
|
|
def test_edge_init_fee(mocker, edge_conf) -> None:
|
|
patch_exchange(mocker)
|
|
edge_conf["fee"] = 0.01234
|
|
edge_conf["stake_amount"] = 20
|
|
fee_mock = mocker.patch(f"{EXMS}.get_fee", return_value=0.5)
|
|
edge_cli = EdgeCli(edge_conf)
|
|
assert edge_cli.edge.fee == 0.01234
|
|
assert fee_mock.call_count == 0
|
|
|
|
|
|
def test_edge_start(mocker, edge_conf) -> None:
|
|
mock_calculate = mocker.patch(
|
|
"freqtrade.edge.edge_positioning.Edge.calculate", return_value=True
|
|
)
|
|
table_mock = mocker.patch("freqtrade.optimize.edge_cli.generate_edge_table")
|
|
|
|
patch_exchange(mocker)
|
|
edge_conf["stake_amount"] = 20
|
|
|
|
edge_cli = EdgeCli(edge_conf)
|
|
edge_cli.start()
|
|
assert mock_calculate.call_count == 1
|
|
assert table_mock.call_count == 1
|