Print history

Author: dandavison

pyproject.toml

@@ -81,7 +81,7 @@ lint-types = [
   { cmd = "uv run mypy --namespace-packages --check-untyped-defs ."},
 ]
 run-bench = "uv run python scripts/run_bench.py"
-test = "uv run pytest -s 'tests/nexus/test_workflow_caller_cancellation_types.py::test_cancellation_type[WAIT_COMPLETED]' 'tests/nexus/test_workflow_caller_cancellation_types_when_cancel_handler_fails.py::test_cancellation_type[WAIT_COMPLETED]'"
+test = "uv run pytest -s 'tests/nexus/test_workflow_caller_cancellation_types.py::test_print_history'"
 
 
 [tool.pytest.ini_options]

tests/helpers/__init__.py

@@ -297,7 +297,7 @@ async def print_history(handle: WorkflowHandle):
     i = 1
     async for evt in handle.fetch_history_events():
         event = EventType.Name(evt.event_type).removeprefix("EVENT_TYPE_")
-        print(f"{i:2}: {event}")
+        print(f"{i:2}: {evt.event_time.ToDatetime()} {event}")
         i += 1
 
 

tests/nexus/test_workflow_caller_cancellation_types.py

@@ -21,7 +21,7 @@
 from temporalio.common import WorkflowIDConflictPolicy
 from temporalio.testing import WorkflowEnvironment
 from temporalio.worker import Worker
-from tests.helpers import print_interleaved_histories
+from tests.helpers import print_history, print_interleaved_histories
 from tests.helpers.nexus import create_nexus_endpoint, make_nexus_endpoint_name
 
 
@@ -513,3 +513,31 @@ def _event_time(
             )
         previous_expected_event_type_name = expected_event_type_name
         previous_expected_handle = expected_handle
+
+
+@workflow.defn(sandboxed=False)
+class PrintHistoryWorkflow:
+    @workflow.run
+    async def run(self) -> None:
+        await workflow.sleep(1)
+
+
+async def test_print_history(env: WorkflowEnvironment):
+    if env.supports_time_skipping:
+        pytest.skip("Nexus tests don't work with time-skipping server")
+
+    client = env.client
+
+    async with Worker(
+        client,
+        task_queue=str(uuid.uuid4()),
+        workflows=[PrintHistoryWorkflow],
+    ) as worker:
+        print_history_wf = await client.start_workflow(
+            PrintHistoryWorkflow.run,
+            id="print-history-wf-1",
+            task_queue=worker.task_queue,
+        )
+        await print_history_wf.result()
+        print()
+        await print_history(print_history_wf)