Add test to check consumer_logger

airtai · May 24, 2024 · 2bb6092 · 2bb6092
1 parent b9c36a0
commit 2bb6092
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/brokers/confluent/test_logger.py b/tests/brokers/confluent/test_logger.py
@@ -37,9 +37,9 @@ def subscriber(m):
         async with self.patch_broker(consume_broker) as br:
             await br.start()
 
-            await br.publish_batch(1, "hi", topic=queue)
-
-            # ToDo: Fetch consumer logger and assert it
+            for sub in br._subscribers.values():
+                consumer_logger = sub.consumer.logger
+                assert consumer_logger == test_logger
 
             producer_logger = br._producer._producer.logger
             assert producer_logger == test_logger