Skip to content

Commit 81b4285

Browse files
committed
test: avoid potential odd timing issues in test
1 parent 8225211 commit 81b4285

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

tests/exchange/test_exchange_ws.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -67,12 +67,12 @@ async def test_exchangews_ohlcv(mocker, time_machine):
6767

6868
async def sleeper(*args, **kwargs):
6969
# pass
70-
await asyncio.sleep(1)
70+
await asyncio.sleep(0.12)
7171
return MagicMock()
7272

7373
ccxt_object.watch_ohlcv = AsyncMock(side_effect=sleeper)
7474
ccxt_object.close = AsyncMock()
75-
time_machine.move_to("2024-11-01 01:00:00 +00:00")
75+
time_machine.move_to("2024-11-01 01:00:02 +00:00")
7676

7777
mocker.patch("freqtrade.exchange.exchange_ws.ExchangeWS._start_forever", MagicMock())
7878

@@ -84,7 +84,7 @@ async def sleeper(*args, **kwargs):
8484

8585
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
8686
exchange_ws.schedule_ohlcv("XRP/BTC", "1m", CandleType.SPOT)
87-
await asyncio.sleep(0.5)
87+
await asyncio.sleep(0.2)
8888

8989
assert exchange_ws._klines_watching == {
9090
("ETH/BTC", "1m", CandleType.SPOT),
@@ -95,21 +95,18 @@ async def sleeper(*args, **kwargs):
9595
("XRP/BTC", "1m", CandleType.SPOT),
9696
}
9797
await asyncio.sleep(0.1)
98-
assert ccxt_object.watch_ohlcv.call_count == 2
98+
assert ccxt_object.watch_ohlcv.call_count == 6
9999
ccxt_object.watch_ohlcv.reset_mock()
100100

101101
time_machine.shift(timedelta(minutes=5))
102-
await asyncio.sleep(0.1)
103102
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
104-
await asyncio.sleep(0.1)
103+
await asyncio.sleep(1)
105104
# XRP/BTC should be cleaned up.
106105
assert exchange_ws._klines_watching == {
107106
("ETH/BTC", "1m", CandleType.SPOT),
108107
}
109108
# Cleanup happened.
110-
await asyncio.sleep(0.1)
111109
exchange_ws.schedule_ohlcv("ETH/BTC", "1m", CandleType.SPOT)
112-
await asyncio.sleep(0.1)
113110
assert exchange_ws._klines_watching == {
114111
("ETH/BTC", "1m", CandleType.SPOT),
115112
}

0 commit comments

Comments
 (0)