CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-05 05:40:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: avoid potential odd timing issues in test

Browse Source
This commit is contained in:
Matthias
2025-01-20 19:12:55 +01:00
parent 822521136f
commit 81b4285d17
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/exchange/test_exchange_ws.py
View File

@@ -67,12 +67,12 @@ async def test_exchangews_ohlcv(mocker, time_machine):
async def sleeper(*args, **kwargs): async def sleeper(*args, **kwargs):
# pass # pass
await asyncio.sleep(1) await asyncio.sleep(0.12)
return MagicMock() return MagicMock()
ccxt_object.watch_ohlcv = AsyncMock(side_effect=sleeper) ccxt_object.watch_ohlcv = AsyncMock(side_effect=sleeper)
ccxt_object.close = AsyncMock() ccxt_object.close = AsyncMock()
time_machine.move_to("2024-11-01 01:00:00 +00:00") time_machine.move_to("2024-11-01 01:00:02 +00:00")
mocker.patch("freqtrade.exchange.exchange_ws.ExchangeWS._start_forever", MagicMock()) mocker.patch("freqtrade.exchange.exchange_ws.ExchangeWS._start_forever", MagicMock())
@@ -84,7 +84,7 @@ async def test_exchangews_ohlcv(mocker, time_machine):
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT) exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
exchange_ws.schedule_ohlcv("XRP/BTC", "1m", CandleType.SPOT) exchange_ws.schedule_ohlcv("XRP/BTC", "1m", CandleType.SPOT)
await asyncio.sleep(0.5) await asyncio.sleep(0.2)
assert exchange_ws._klines_watching == { assert exchange_ws._klines_watching == {
("ETH/BTC", "1m", CandleType.SPOT), ("ETH/BTC", "1m", CandleType.SPOT),
@@ -95,21 +95,18 @@ async def test_exchangews_ohlcv(mocker, time_machine):
("XRP/BTC", "1m", CandleType.SPOT), ("XRP/BTC", "1m", CandleType.SPOT),
} }
await asyncio.sleep(0.1) await asyncio.sleep(0.1)
assert ccxt_object.watch_ohlcv.call_count == 2 assert ccxt_object.watch_ohlcv.call_count == 6
ccxt_object.watch_ohlcv.reset_mock() ccxt_object.watch_ohlcv.reset_mock()
time_machine.shift(timedelta(minutes=5)) time_machine.shift(timedelta(minutes=5))
await asyncio.sleep(0.1)
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT) exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
await asyncio.sleep(0.1) await asyncio.sleep(1)
# XRP/BTC should be cleaned up. # XRP/BTC should be cleaned up.
assert exchange_ws._klines_watching == { assert exchange_ws._klines_watching == {
("ETH/BTC", "1m", CandleType.SPOT), ("ETH/BTC", "1m", CandleType.SPOT),
} }
# Cleanup happened. # Cleanup happened.
await asyncio.sleep(0.1)
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT) exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
await asyncio.sleep(0.1)
assert exchange_ws._klines_watching == { assert exchange_ws._klines_watching == {
("ETH/BTC", "1m", CandleType.SPOT), ("ETH/BTC", "1m", CandleType.SPOT),
} }