mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-09-20 17:41:12 +00:00
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
|
|
|
|
import shutil
|
|
|
|
import pytest
|
|
|
|
from freqtrade.persistence import Trade
|
|
from freqtrade.util.migrations import (migrate_binance_futures_data, migrate_binance_futures_names,
|
|
migrate_data)
|
|
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
|