fix cloud tests

Author: zhongkechen

examples/examples-catalog.json

@@ -221,6 +221,17 @@
       },
       "path": "./src/parallel/parallel.py"
     },
+    {
+      "name": "Parallel Operations",
+      "description": "Executing multiple durable operations in parallel",
+      "handler": "parallel.handler",
+      "integration": true,
+      "durableConfig": {
+        "RetentionPeriodInDays": 7,
+        "ExecutionTimeout": 300
+      },
+      "path": "./src/parallel/parallel.py"
+    },
     {
       "name": "Map Operations",
       "description": "Processing collections using map-like durable operations",
@@ -232,6 +243,17 @@
       },
       "path": "./src/map/map_operations.py"
     },
+    {
+      "name": "Map Operations Flat",
+      "description": "Processing collections using map-like durable operations in FLAT mode",
+      "handler": "map_operations_flat.handler",
+      "integration": true,
+      "durableConfig": {
+        "RetentionPeriodInDays": 7,
+        "ExecutionTimeout": 300
+      },
+      "path": "./src/map/map_operations_flat.py"
+    },
     {
       "name": "Map Large Scale",
       "description": "Processing collections using map-like durable operations in large scale",

examples/template.yaml

@@ -420,6 +420,24 @@
         }
       }
     },
+    "MapOperationsFlat": {
+      "Type": "AWS::Serverless::Function",
+      "Properties": {
+        "CodeUri": "build/",
+        "Handler": "map_operations_flat.handler",
+        "Description": "Processing collections using map-like durable operations in FLAT mode",
+        "Role": {
+          "Fn::GetAtt": [
+            "DurableFunctionRole",
+            "Arn"
+          ]
+        },
+        "DurableConfig": {
+          "RetentionPeriodInDays": 7,
+          "ExecutionTimeout": 300
+        }
+      }
+    },
     "MapWithLargeScale": {
       "Type": "AWS::Serverless::Function",
       "Properties": {

examples/test/map/test_map_operations_flat.py

@@ -11,7 +11,7 @@
 @pytest.mark.example
 @pytest.mark.durable_execution(
     handler=map_operations_flat.handler,
-    lambda_function_name="map operations",
+    lambda_function_name="Map Operations Flat",
 )
 def test_map_operations_flat(durable_runner):
     """Test map_operations example using context.map()."""

examples/test/parallel/test_parallel.py

@@ -2,7 +2,7 @@
 
 import pytest
 from aws_durable_execution_sdk_python.execution import InvocationStatus
-from aws_durable_execution_sdk_python.lambda_service import OperationStatus
+from aws_durable_execution_sdk_python.lambda_service import OperationStatus, OperationType
 
 from src.parallel import parallel
 from test.conftest import deserialize_operation_payload
@@ -35,4 +35,5 @@ def test_parallel(durable_runner):
 
     # Verify all children succeeded
     for child in parallel_op.child_operations:
+        assert child.operation_type == OperationType.CONTEXT
         assert child.status is OperationStatus.SUCCEEDED

examples/test/parallel/test_parallel_flat.py

@@ -2,7 +2,7 @@
 
 import pytest
 from aws_durable_execution_sdk_python.execution import InvocationStatus
-from aws_durable_execution_sdk_python.lambda_service import OperationStatus
+from aws_durable_execution_sdk_python.lambda_service import OperationStatus, OperationType
 
 from src.parallel import parallel_flat
 from test.conftest import deserialize_operation_payload
@@ -11,7 +11,7 @@
 @pytest.mark.example
 @pytest.mark.durable_execution(
     handler=parallel_flat.handler,
-    lambda_function_name="Parallel Operations",
+    lambda_function_name="Parallel Operations Flat",
 )
 def test_parallel_flat(durable_runner):
     """Test parallel example using context.parallel()."""
@@ -35,4 +35,5 @@ def test_parallel_flat(durable_runner):
 
     # Verify all children succeeded
     for child in parallel_op.child_operations:
+        assert child.operation_type != OperationType.CONTEXT
         assert child.status is OperationStatus.SUCCEEDED