freqtrade_origin/tests/optimize/test_edge_cli.py

# 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