added logging test

Author: GSmithApps

hello/hello_change_log_level.py

@@ -33,11 +33,14 @@ def filter(self, record: logging.LogRecord) -> bool:
 # --- End logging set‑up ----------------------------------------------------------
 
 
+LOG_MESSAGE = "This error is an experiment to check the log level"
+
+
 @workflow.defn
 class GreetingWorkflow:
     @workflow.run
     async def run(self):
-        raise RuntimeError("This error is an experiment to check the log level")
+        raise RuntimeError(LOG_MESSAGE)
 
 
 async def main():

tests/hello/hello_change_log_level_test.py

@@ -1,14 +1,23 @@
-import uuid
 import asyncio
+import io
 import logging
+import uuid
 
 from temporalio.client import Client
 from temporalio.worker import Worker
 
-from hello.hello_change_log_level import GreetingWorkflow
+from hello.hello_change_log_level import LOG_MESSAGE, GreetingWorkflow
+
 
+async def test_workflow_with_log_capture(client: Client):
 
-async def test_workflow_with_changed_log_level(client: Client, caplog):
+    log_stream = io.StringIO()
+    handler = logging.StreamHandler(log_stream)
+    handler.setLevel(logging.DEBUG)
+
+    logger = logging.getLogger()
+    logger.addHandler(handler)
+    logger.setLevel(logging.DEBUG)
 
     task_queue = f"tq-{uuid.uuid4()}"
 
@@ -17,14 +26,18 @@ async def test_workflow_with_changed_log_level(client: Client, caplog):
         task_queue=task_queue,
         workflows=[GreetingWorkflow],
     ):
-        with caplog.at_level(logging.ERROR):
-            handle = await client.start_workflow(
-                GreetingWorkflow.run,
-                id=f"wf-{uuid.uuid4()}",
-                task_queue=task_queue,
-            )
-            await asyncio.sleep(.1)
-            handle.terminate()
-
-    assert any("log level" in m for m in caplog.messages)
-    assert True
+        handle = await client.start_workflow(
+            GreetingWorkflow.run,
+            id=f"wf-{uuid.uuid4()}",
+            task_queue=task_queue,
+        )
+        await asyncio.sleep(
+            0.2
+        )  # arbitrary wait to ensure the workflow has started and logged
+        await handle.terminate()
+
+    logger.removeHandler(handler)
+    handler.flush()
+
+    logs = log_stream.getvalue()
+    assert LOG_MESSAGE in logs