fix: add pattern_event.wait()

airtai · Jan 8, 2025 · 233a127 · 233a127
1 parent 9230fd0
commit 233a127
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/brokers/kafka/test_consume.py b/tests/brokers/kafka/test_consume.py
@@ -40,6 +40,7 @@ async def pattern_handler(msg) -> None:
             await br.start()
             result = await br.publish(1, topic=queue)
             assert isinstance(result, RecordMetadata), result
+        await pattern_event.wait()
         assert pattern_event.is_set()
 
     @pytest.mark.asyncio()