increasing timeout error in tests

.github/workflows/test.yaml

@@ -19,7 +19,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     services:
       rabbitmq-streaming:
-        image: rabbitmq:3.13-rc-management
+        image: rabbitmq:3.13.1-management
         env:
           RABBITMQ_SERVER_ADDITIONAL_ERL_ARGS: "-rabbitmq_stream advertised_host localhost"
         ports:

tests/test_consumer.py

@@ -127,7 +127,7 @@ async def test_offset_type_last(stream: str, consumer: Consumer, producer: Produ
         subscriber_name="test-subscriber",
     )
 
-    await wait_for(lambda: len(captured) > 0 and captured[-1] == b"4999")
+    await wait_for(lambda: len(captured) > 0 and captured[-1] == b"4999", 2)
     assert len(captured) < len(messages)
 
 
@@ -195,7 +195,7 @@ async def test_offset_type_timestamp(stream: str, consumer: Consumer, producer:
         callback=lambda message, message_context: captured.append(bytes(message)),
         offset_specification=ConsumerOffsetSpecification(offset_type=OffsetType.TIMESTAMP, offset=now),
     )
-    await wait_for(lambda: len(captured) > 0 and captured[0] >= b"5000")
+    await wait_for(lambda: len(captured) > 0 and captured[0] >= b"5000", 2)
 
 
 async def test_offset_type_next(stream: str, consumer: Consumer, producer: Producer) -> None:

tests/test_producer.py

@@ -73,7 +73,7 @@ async def test_publishing_several_messages(stream: str, producer: Producer, cons
     for i in range(0, 100000):
         await producer.send(stream, b"one")
 
-    await wait_for(lambda: len(captured) == 100000)
+    await wait_for(lambda: len(captured) == 100000, 2)
 
 
 async def test_publishing_several_messages_different_streams(
@@ -93,8 +93,8 @@ async def test_publishing_several_messages_different_streams(
     for i in range(0, 100000):
         await producer.send(stream2, b"one")
 
-    await wait_for(lambda: len(captured_stream_1) == 100000)
-    await wait_for(lambda: len(captured_stream_2) == 100000)
+    await wait_for(lambda: len(captured_stream_1) == 100000, 2)
+    await wait_for(lambda: len(captured_stream_2) == 100000, 2)
 
 
 async def test_publishing_sequence_subbatching_nocompression(

tests/util.py

@@ -138,7 +138,7 @@ async def task_to_delete_connection(connection_name: str) -> None:
     for connection in connections:
         if connection["client_properties"]["connection_name"] == connection_name:
             delete_connection(connection["name"])
-            await wait_for(lambda: get_connection(connection["name"]) is False)
+            await wait_for(lambda: get_connection(connection["name"]) is False, 2)
 
 
 async def task_to_delete_stream_producer(producer: Producer, stream: str) -> None: