From 6cce455835d94948fad9726274032865ccea1cf0 Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 4 Jan 2024 17:08:57 +0100 Subject: [PATCH] Test migrate_data wrapper --- tests/test_binance_mig.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/test_binance_mig.py b/tests/test_binance_mig.py index 9b3e927d3..39a853a27 100644 --- a/tests/test_binance_mig.py +++ b/tests/test_binance_mig.py @@ -5,7 +5,8 @@ import shutil import pytest from freqtrade.persistence import Trade -from freqtrade.util.migrations import migrate_binance_futures_data, migrate_binance_futures_names +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 @@ -55,3 +56,13 @@ def test_binance_mig_db_conversion(default_conf_usdt, fee, caplog): 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