tests: use timeout (#1637)

airtai · Aug 4, 2024 · 9cd4664 · 9cd4664
1 parent d705076
commit 9cd4664
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/brokers/base/testclient.py b/tests/brokers/base/testclient.py
@@ -126,10 +126,10 @@ async def m(msg):
         async with self.test_class(test_broker, with_real=True) as br:
             await br.publish("hello", queue)
 
-            await m.wait_call(3.0)
+            await m.wait_call(self.timeout)
 
             m.mock.assert_called_once_with("hello")
-            with anyio.fail_after(3.0):
+            with anyio.fail_after(self.timeout):
                 while not publisher.mock.called:
                     await asyncio.sleep(0.1)
                 publisher.mock.assert_called_once_with("response: hello")