Test workflow and activity payload converters have context

Author: dandavison

tests/test_serialization_context.py

@@ -155,6 +155,7 @@ def __init__(self):
 
 @activity.defn
 async def passthrough_activity(input: TraceData) -> TraceData:
+    activity.payload_converter().to_payload(input)
     activity.heartbeat(input)
     # Wait for the heartbeat to be processed so that it modifies the data before the activity returns
     await asyncio.sleep(0.2)
@@ -172,6 +173,7 @@ async def run(self, data: TraceData) -> TraceData:
 class PayloadConversionWorkflow:
     @workflow.run
     async def run(self, data: TraceData) -> TraceData:
+        workflow.payload_converter().to_payload(data)
         data = await workflow.execute_activity(
             passthrough_activity,
             data,
@@ -242,6 +244,10 @@ async def test_payload_conversion_calls_follow_expected_sequence_and_contexts(
                 method="from_payload",
                 context=workflow_context,  # Inbound workflow input
             ),
+            TraceItem(
+                method="to_payload",
+                context=workflow_context,  # workflow payload converter
+            ),
             TraceItem(
                 method="to_payload",
                 context=activity_context,  # Outbound activity input
@@ -250,6 +256,10 @@ async def test_payload_conversion_calls_follow_expected_sequence_and_contexts(
                 method="from_payload",
                 context=activity_context,  # Inbound activity input
             ),
+            TraceItem(
+                method="to_payload",
+                context=activity_context,  # activity payload converter
+            ),
             TraceItem(
                 method="to_payload",
                 context=activity_context,  # Outbound heartbeat