Skip to content

Commit 14810ed

Browse files
Update receive public trades integration test
The integration test for receiving public trades has been updated to use the new method `receive_public_trades`. Signed-off-by: Daniel Zullo <[email protected]>
1 parent 10e88c8 commit 14810ed

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

integration_tests/test_api.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -390,13 +390,14 @@ async def test_gridpool_orders_stream(set_up: dict[str, Any]) -> None:
390390
pytest.fail("Streaming timed out, no order received in 15 seconds")
391391

392392

393-
async def test_stream_public_trades(set_up: dict[str, Any]) -> None:
394-
"""Test stream public trades."""
395-
stream = await set_up["client"].stream_public_trades()
393+
async def test_receive_public_trades(set_up: dict[str, Any]) -> None:
394+
"""Test receive public trades."""
395+
stream = set_up["client"].receive_public_trades()
396+
receiver = stream.new_receiver()
396397

397398
try:
398399
# Stream trades with a 15-second timeout to avoid indefinite hanging
399-
streamed_trade = await asyncio.wait_for(anext(stream), timeout=15)
400+
streamed_trade = await asyncio.wait_for(anext(receiver), timeout=15)
400401
assert streamed_trade is not None, "Failed to receive streamed trade"
401402
except asyncio.TimeoutError:
402403
pytest.fail("Streaming timed out, no trade received in 15 seconds")

0 commit comments

Comments
 (0)