Fix buffering issue on pubsub example test

Signed-off-by: Sergio Herrera <627709+seherv@users.noreply.github.com>

Author: seherv

examples/pubsub-streaming-async/publisher.py

@@ -44,7 +44,7 @@ async def publish_events():
             )
 
             # Print the request
-            print(req_data, flush=True)
+            print(req_data)
 
             await asyncio.sleep(1)
 

examples/pubsub-streaming-async/subscriber.py

@@ -19,7 +19,7 @@ def process_message(message):
     global counter
     counter += 1
     # Process the message here
-    print(f'Processing message: {message.data()} from {message.topic()}...', flush=True)
+    print(f'Processing message: {message.data()} from {message.topic()}...')
     return 'success'
 
 

tests/examples/test_pubsub_streaming_async.py

@@ -6,6 +6,7 @@
     "Processing message: {'id': 3, 'message': 'hello world'} from TOPIC_B1...",
     "Processing message: {'id': 4, 'message': 'hello world'} from TOPIC_B1...",
     "Processing message: {'id': 5, 'message': 'hello world'} from TOPIC_B1...",
+    "Closing subscription...",
 ]
 
 EXPECTED_HANDLER_SUBSCRIBER = [
@@ -14,6 +15,7 @@
     "Processing message: {'id': 3, 'message': 'hello world'} from TOPIC_B2...",
     "Processing message: {'id': 4, 'message': 'hello world'} from TOPIC_B2...",
     "Processing message: {'id': 5, 'message': 'hello world'} from TOPIC_B2...",
+    "Closing subscription...",
 ]
 
 EXPECTED_PUBLISHER = [
@@ -28,12 +30,12 @@
 @pytest.mark.example_dir('pubsub-streaming-async')
 def test_pubsub_streaming_async(dapr):
     dapr.start(
-        '--app-id python-subscriber --app-protocol grpc -- python3 subscriber.py --topic=TOPIC_B1',
+        '--app-id python-subscriber --app-protocol grpc -- python3 -u subscriber.py --topic=TOPIC_B1',
         wait=5,
     )
     publisher_output = dapr.run(
         '--app-id python-publisher --app-protocol grpc --dapr-grpc-port=3500 '
-        '--enable-app-health-check -- python3 publisher.py --topic=TOPIC_B1',
+        '--enable-app-health-check -- python3 -u publisher.py --topic=TOPIC_B1',
         timeout=30,
     )
     for line in EXPECTED_PUBLISHER:
@@ -47,12 +49,12 @@ def test_pubsub_streaming_async(dapr):
 @pytest.mark.example_dir('pubsub-streaming-async')
 def test_pubsub_streaming_async_handler(dapr):
     dapr.start(
-        '--app-id python-subscriber --app-protocol grpc -- python3 subscriber-handler.py --topic=TOPIC_B2',
+        '--app-id python-subscriber --app-protocol grpc -- python3 -u subscriber-handler.py --topic=TOPIC_B2',
         wait=5,
     )
     publisher_output = dapr.run(
         '--app-id python-publisher --app-protocol grpc --dapr-grpc-port=3500 '
-        '--enable-app-health-check -- python3 publisher.py --topic=TOPIC_B2',
+        '--enable-app-health-check -- python3 -u publisher.py --topic=TOPIC_B2',
         timeout=30,
     )
     for line in EXPECTED_PUBLISHER: