[fix]: input payload too big to fit in initial execution state

Author: zhongkechen

src/aws_durable_execution_sdk_python_testing/execution.py

@@ -91,7 +91,9 @@ def new(input: StartDurableExecutionInput) -> Execution:  # noqa: A002
         # Pattern: arn:(aws[a-zA-Z-]*)?:lambda:[a-z]{2}(-gov)?-[a-z]+-\d{1}:\d{12}:durable-execution:[a-zA-Z0-9-_\.]+:[a-zA-Z0-9-_\.]+:[a-zA-Z0-9-_\.]+
         # Example: arn:aws:lambda:us-east-1:123456789012:durable-execution:myDurableFunction:myDurableExecutionName:ce67da72-3701-4f83-9174-f4189d27b0a5
         return Execution(
-            durable_execution_arn=str(uuid4()), start_input=input, operations=[]
+            durable_execution_arn=str(uuid4()) + "/" + (input.invocation_id or str(uuid4())),
+            start_input=input,
+            operations=[],
         )
 
     def to_json_dict(self) -> dict[str, Any]:

tests/execution_test.py

@@ -33,6 +33,7 @@ def test_execution_init():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = []
 
@@ -61,11 +62,12 @@ def test_execution_new(mock_uuid4):
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id-1234",
     )
 
     execution = Execution.new(start_input)
 
-    assert execution.durable_execution_arn == str(mock_uuid)
+    assert execution.durable_execution_arn == str(mock_uuid) + "/test-invocation-id-1234"
     assert execution.start_input == start_input
     assert execution.operations == []
 
@@ -130,6 +132,7 @@ def test_get_operation_execution_started_not_started():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [])
 
@@ -146,6 +149,7 @@ def test_get_new_checkpoint_token():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="invocation-id",
     )
     execution = Execution("test-arn", start_input, [])
 
@@ -167,6 +171,7 @@ def test_get_navigable_operations():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -194,6 +199,7 @@ def test_get_assertable_operations():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution_op = Operation(
         operation_id="exec-op",
@@ -229,6 +235,7 @@ def test_has_pending_operations_with_pending_step():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -256,6 +263,7 @@ def test_has_pending_operations_with_started_wait():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -283,6 +291,7 @@ def test_has_pending_operations_with_started_callback():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -310,6 +319,7 @@ def test_has_pending_operations_with_started_invoke():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -337,6 +347,7 @@ def test_has_pending_operations_no_pending():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operations = [
         Operation(
@@ -364,6 +375,7 @@ def test_complete_success_with_string_result():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
 
@@ -383,6 +395,7 @@ def test_complete_success_with_none_result():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
 
@@ -402,6 +415,7 @@ def test_complete_fail():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
     error = ErrorObject.from_message("Test error message")
@@ -422,6 +436,7 @@ def test_find_operation_exists():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="test-op-id",
@@ -448,6 +463,7 @@ def test_find_operation_not_exists():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [])
 
@@ -470,6 +486,7 @@ def test_complete_wait_success(mock_datetime):
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="wait-op-id",
@@ -498,6 +515,7 @@ def test_complete_wait_wrong_status():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="wait-op-id",
@@ -524,6 +542,7 @@ def test_complete_wait_wrong_type():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="step-op-id",
@@ -548,6 +567,7 @@ def test_complete_retry_success():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     step_details = StepDetails(
         next_attempt_timestamp=str(datetime.now(timezone.utc)),
@@ -581,6 +601,7 @@ def test_complete_retry_no_step_details():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="step-op-id",
@@ -608,6 +629,7 @@ def test_complete_retry_wrong_status():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="step-op-id",
@@ -634,6 +656,7 @@ def test_complete_retry_wrong_type():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     operation = Operation(
         operation_id="wait-op-id",
@@ -658,6 +681,7 @@ def test_status_running():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [])
 
@@ -673,6 +697,7 @@ def test_status_succeeded():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
     execution.complete_success("success result")
@@ -689,6 +714,7 @@ def test_status_failed():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
     error = ErrorObject.from_message("Test error")
@@ -706,6 +732,7 @@ def test_status_timed_out():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
     error = ErrorObject(
@@ -725,6 +752,7 @@ def test_status_stopped():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="invocation-id",
     )
     execution = Execution("test-arn", start_input, [Mock()])
     error = ErrorObject(
@@ -744,6 +772,7 @@ def test_status_no_result():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="invocation-id",
     )
     execution = Execution("test-arn", start_input, [])
     execution.is_complete = True

tests/invoker_test.py

@@ -71,6 +71,7 @@ def test_in_process_invoker_create_invocation_input():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation-id",
     )
     execution = Execution.new(input_data)
 
@@ -151,6 +152,7 @@ def test_lambda_invoker_create_invocation_input():
         execution_name="test-execution",
         execution_timeout_seconds=300,
         execution_retention_period_days=7,
+        invocation_id="test-invocation",
     )
     execution = Execution.new(input_data)