Update tests for leverage_tier_loading
This commit is contained in:
parent
cce8d1aa4d
commit
ea537b32c7
@ -78,9 +78,21 @@ def get_args(args):
|
|||||||
|
|
||||||
|
|
||||||
# Source: https://stackoverflow.com/questions/29881236/how-to-mock-asyncio-coroutines
|
# Source: https://stackoverflow.com/questions/29881236/how-to-mock-asyncio-coroutines
|
||||||
def get_mock_coro(return_value):
|
# TODO: This should be replaced with AsyncMock once support for python 3.7 is dropped.
|
||||||
|
def get_mock_coro(return_value=None, side_effect=None):
|
||||||
async def mock_coro(*args, **kwargs):
|
async def mock_coro(*args, **kwargs):
|
||||||
return return_value
|
if side_effect:
|
||||||
|
if isinstance(side_effect, list):
|
||||||
|
effect = side_effect.pop(0)
|
||||||
|
else:
|
||||||
|
effect = side_effect
|
||||||
|
if isinstance(effect, Exception):
|
||||||
|
raise effect
|
||||||
|
if callable(effect):
|
||||||
|
return effect(*args, **kwargs)
|
||||||
|
return effect
|
||||||
|
else:
|
||||||
|
return return_value
|
||||||
|
|
||||||
return Mock(wraps=mock_coro)
|
return Mock(wraps=mock_coro)
|
||||||
|
|
||||||
|
@ -6,7 +6,7 @@ import pytest
|
|||||||
from freqtrade.enums import MarginMode, TradingMode
|
from freqtrade.enums import MarginMode, TradingMode
|
||||||
from freqtrade.enums.candletype import CandleType
|
from freqtrade.enums.candletype import CandleType
|
||||||
from freqtrade.exchange.exchange import timeframe_to_minutes
|
from freqtrade.exchange.exchange import timeframe_to_minutes
|
||||||
from tests.conftest import get_patched_exchange
|
from tests.conftest import get_mock_coro, get_patched_exchange
|
||||||
from tests.exchange.test_exchange import ccxt_exceptionhandlers
|
from tests.exchange.test_exchange import ccxt_exceptionhandlers
|
||||||
|
|
||||||
|
|
||||||
@ -273,7 +273,7 @@ def test_load_leverage_tiers_okx(default_conf, mocker, markets):
|
|||||||
'fetchLeverageTiers': False,
|
'fetchLeverageTiers': False,
|
||||||
'fetchMarketLeverageTiers': True,
|
'fetchMarketLeverageTiers': True,
|
||||||
})
|
})
|
||||||
api_mock.fetch_market_leverage_tiers = MagicMock(side_effect=[
|
api_mock.fetch_market_leverage_tiers = get_mock_coro(side_effect=[
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
'tier': 1,
|
'tier': 1,
|
||||||
|
Loading…
Reference in New Issue
Block a user