mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-16 05:03:55 +00:00
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
import shutil
|
|
|
|
import pytest
|
|
|
|
from freqtrade.persistence import Trade
|
|
from freqtrade.util.migrations import migrate_binance_futures_data, migrate_data
|
|
from freqtrade.util.migrations.binance_mig import migrate_binance_futures_names
|
|
from tests.conftest import create_mock_trades_usdt, log_has
|
|
|
|
|
|
def test_binance_mig_data_conversion(default_conf_usdt, tmp_path, testdatadir):
|
|
# call doing nothing (spot mode)
|
|
migrate_binance_futures_data(default_conf_usdt)
|
|
default_conf_usdt["trading_mode"] = "futures"
|
|
pair_old = "XRP_USDT"
|
|
pair_unified = "XRP_USDT_USDT"
|
|
futures_src = testdatadir / "futures"
|
|
futures_dst = tmp_path / "futures"
|
|
futures_dst.mkdir()
|
|
files = [
|
|
"-1h-mark.feather",
|
|
"-1h-futures.feather",
|
|
"-8h-funding_rate.feather",
|
|
"-8h-mark.feather",
|
|
]
|
|
|
|
# Copy files to tmpdir and rename to old naming
|
|
for file in files:
|
|
fn_after = futures_dst / f"{pair_old}{file}"
|
|
shutil.copy(futures_src / f"{pair_unified}{file}", fn_after)
|
|
|
|
default_conf_usdt["datadir"] = tmp_path
|
|
# Migrate files to unified namings
|
|
migrate_binance_futures_data(default_conf_usdt)
|
|
|
|
for file in files:
|
|
fn_after = futures_dst / f"{pair_unified}{file}"
|
|
assert fn_after.exists()
|
|
|
|
|
|
@pytest.mark.usefixtures("init_persistence")
|
|
def test_binance_mig_db_conversion(default_conf_usdt, fee, caplog):
|
|
# Does nothing in spot mode
|
|
migrate_binance_futures_names(default_conf_usdt)
|
|
|
|
create_mock_trades_usdt(fee, None)
|
|
|
|
for t in Trade.get_trades():
|
|
t.trading_mode = "FUTURES"
|
|
t.exchange = "binance"
|
|
Trade.commit()
|
|
|
|
default_conf_usdt["trading_mode"] = "futures"
|
|
migrate_binance_futures_names(default_conf_usdt)
|
|
assert log_has("Migrating binance futures pairs in database.", caplog)
|
|
|
|
|
|
def test_migration_wrapper(default_conf_usdt, mocker):
|
|
default_conf_usdt["trading_mode"] = "futures"
|
|
binmock = mocker.patch("freqtrade.util.migrations.migrate_binance_futures_data")
|
|
funding_mock = mocker.patch("freqtrade.util.migrations.migrate_funding_fee_timeframe")
|
|
migrate_data(default_conf_usdt)
|
|
|
|
assert binmock.call_count == 1
|
|
assert funding_mock.call_count == 1
|