From 6f7ab97fc363daa90a9c1767498a80192f69edef Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 26 Feb 2023 15:41:42 +0100 Subject: [PATCH] Improve bybit test coverage --- tests/exchange/test_bybit.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tests/exchange/test_bybit.py b/tests/exchange/test_bybit.py index 7c8324bf6..d0d5114a1 100644 --- a/tests/exchange/test_bybit.py +++ b/tests/exchange/test_bybit.py @@ -1,3 +1,4 @@ +from datetime import datetime, timezone from unittest.mock import MagicMock from freqtrade.enums.marginmode import MarginMode @@ -55,3 +56,19 @@ async def test_bybit_fetch_funding_rate(default_conf, mocker): kwargs = api_mock.fetch_funding_rate_history.call_args_list[0][1] assert kwargs['params'] == {'until': since_ms_end} assert kwargs['since'] == since_ms + + +def test_bybit_get_funding_fees(default_conf, mocker): + now = datetime.now(timezone.utc) + exchange = get_patched_exchange(mocker, default_conf, id='bybit') + exchange._fetch_and_calculate_funding_fees = MagicMock() + exchange.get_funding_fees('BTC/USDT:USDT', 1, False, now) + assert exchange._fetch_and_calculate_funding_fees.call_count == 0 + + default_conf['trading_mode'] = 'futures' + default_conf['margin_mode'] = 'isolated' + exchange = get_patched_exchange(mocker, default_conf, id='bybit') + exchange._fetch_and_calculate_funding_fees = MagicMock() + exchange.get_funding_fees('BTC/USDT:USDT', 1, False, now) + + assert exchange._fetch_and_calculate_funding_fees.call_count == 1