from copy import deepcopy
from unittest.mock import MagicMock

import pytest

from freqtrade.data.history.history_utils import get_timerange
from freqtrade.optimize.backtesting import Backtesting
from freqtrade.persistence import Trade, disable_database_use, enable_database_use
from freqtrade.persistence.custom_data import CustomDataWrapper
from tests.conftest import (
    EXMS,
    create_mock_trades_usdt,
    generate_test_data,
    get_patched_freqtradebot,
    patch_exchange,
)


@pytest.mark.usefixtures("init_persistence")
@pytest.mark.parametrize("use_db", [True, False])
def test_trade_custom_data(fee, use_db):
    if not use_db:
        disable_database_use('5m')
    Trade.reset_trades()
    CustomDataWrapper.reset_custom_data()

    create_mock_trades_usdt(fee, use_db=use_db)

    trade1 = Trade.get_trades_proxy()[0]
    if not use_db:
        trade1.id = 1

    assert trade1.get_all_custom_data() == []
    trade1.set_custom_data('test_str', 'test_value')
    trade1.set_custom_data('test_int', 1)
    trade1.set_custom_data('test_float', 1.55)
    trade1.set_custom_data('test_bool', True)
    trade1.set_custom_data('test_dict', {'test': 'dict'})

    assert len(trade1.get_all_custom_data()) == 5
    assert trade1.get_custom_data('test_str') == 'test_value'
    trade1.set_custom_data('test_str', 'test_value_updated')
    assert trade1.get_custom_data('test_str') == 'test_value_updated'

    assert trade1.get_custom_data('test_int') == 1
    assert isinstance(trade1.get_custom_data('test_int'), int)

    assert trade1.get_custom_data('test_float') == 1.55
    assert isinstance(trade1.get_custom_data('test_float'), float)

    assert trade1.get_custom_data('test_bool') is True
    assert isinstance(trade1.get_custom_data('test_bool'), bool)

    assert trade1.get_custom_data('test_dict') == {'test': 'dict'}
    assert isinstance(trade1.get_custom_data('test_dict'), dict)
    if not use_db:
        enable_database_use()


def test_trade_custom_data_strategy_compat(mocker, default_conf_usdt, fee):

    mocker.patch(f'{EXMS}.get_rate', return_value=0.50)
    mocker.patch('freqtrade.freqtradebot.FreqtradeBot.get_real_amount', return_value=None)
    default_conf_usdt["minimal_roi"] = {"0":  100}

    freqtrade = get_patched_freqtradebot(mocker, default_conf_usdt)
    create_mock_trades_usdt(fee)

    trade1 = Trade.get_trades_proxy(pair='ADA/USDT')[0]
    trade1.set_custom_data('test_str', 'test_value')
    trade1.set_custom_data('test_int', 1)

    def custom_exit(pair, trade, **kwargs):

        if pair == 'ADA/USDT':
            custom_val = trade.get_custom_data('test_str')
            custom_val_i = trade.get_custom_data('test_int')

            return f"{custom_val}_{custom_val_i}"

    freqtrade.strategy.custom_exit = custom_exit
    ff_spy = mocker.spy(freqtrade.strategy, 'custom_exit')
    trades = Trade.get_open_trades()
    freqtrade.exit_positions(trades)
    Trade.commit()

    trade_after = Trade.get_trades_proxy(pair='ADA/USDT')[0]
    assert trade_after.get_custom_data('test_str') == 'test_value'
    assert trade_after.get_custom_data('test_int') == 1
    # 2 open pairs eligible for exit
    assert ff_spy.call_count == 2

    assert trade_after.exit_reason == 'test_value_1'


def test_trade_custom_data_strategy_backtest_compat(mocker, default_conf_usdt, fee):

    mocker.patch(f'{EXMS}.get_fee', fee)
    mocker.patch(f"{EXMS}.get_min_pair_stake_amount", return_value=10)
    mocker.patch(f"{EXMS}.get_max_pair_stake_amount", return_value=float('inf'))
    mocker.patch(f"{EXMS}.get_max_leverage", return_value=10)
    mocker.patch(f"{EXMS}.get_maintenance_ratio_and_amt", return_value=(0.1, 0.1))
    mocker.patch('freqtrade.optimize.backtesting.Backtesting._run_funding_fees')

    patch_exchange(mocker)
    default_conf_usdt.update({
        "stake_amount": 100.0,
        "max_open_trades": 2,
        "dry_run_wallet": 1000.0,
        "strategy": "StrategyTestV3",
        "trading_mode": "futures",
        "margin_mode": "isolated",
        "stoploss": -2,
        "minimal_roi": {"0":  100},
    })
    default_conf_usdt['pairlists'] = [{'method': 'StaticPairList', 'allow_inactive': True}]
    backtesting = Backtesting(default_conf_usdt)

    df = generate_test_data(default_conf_usdt['timeframe'], 100, '2022-01-01 00:00:00+00:00')

    pair_exp = 'XRP/USDT:USDT'

    def custom_exit(pair, trade, **kwargs):
        custom_val = trade.get_custom_data('test_str')
        custom_val_i = trade.get_custom_data('test_int', 0)

        if pair == pair_exp:
            trade.set_custom_data('test_str', 'test_value')
            trade.set_custom_data('test_int', custom_val_i + 1)

        if custom_val_i >= 2:
            return f"{custom_val}_{custom_val_i}"

    backtesting._set_strategy(backtesting.strategylist[0])
    processed = backtesting.strategy.advise_all_indicators({
        pair_exp: df,
        'BTC/USDT:USDT': df,
    })

    def fun(dataframe, *args, **kwargs):
        dataframe.loc[dataframe.index == 50, 'enter_long'] = 1
        return dataframe

    backtesting.strategy.advise_entry = fun
    backtesting.strategy.leverage = MagicMock(return_value=1)
    backtesting.strategy.custom_exit = custom_exit
    ff_spy = mocker.spy(backtesting.strategy, 'custom_exit')

    min_date, max_date = get_timerange(processed)

    result = backtesting.backtest(
        processed=deepcopy(processed),
        start_date=min_date,
        end_date=max_date,
    )
    results = result['results']
    assert not results.empty
    assert len(results) == 2
    assert results['pair'][0] == pair_exp
    assert results['pair'][1] == 'BTC/USDT:USDT'
    assert results['exit_reason'][0] == 'test_value_2'
    assert results['exit_reason'][1] == 'exit_signal'

    assert ff_spy.call_count == 7
    Backtesting.cleanup()