feat: add three more conformance suites

Adds callback (4-1..4-19), invoke (5-1..5-16), and wait_for_callback
(7-1..7-15) suites; template-driven discovery picks them up automatically.
Inject script now preserves CloudFormation short-form intrinsic tags, and
the workflow uses a stack-safe suite slug plus runner env passthrough.

Author: Alex Wang

.github/workflows/conformance-tests.yml

@@ -104,13 +104,26 @@ jobs:
           --template template_${{ matrix.suite }}.yaml \
           --role-arn "$ROLE_ARN"
 
+    - name: Compute stack-safe suite slug
+      run: |
+        # CloudFormation stack names allow only [a-zA-Z][-a-zA-Z0-9]*;
+        # suite names like wait_for_callback contain underscores.
+        echo "SUITE_SLUG=$(echo '${{ matrix.suite }}' | tr '_' '-')" >> "$GITHUB_ENV"
+
     - name: Run conformance suite
+      env:
+        # Same env the SAM-based integration deploy passes. LAMBDA_ENDPOINT is
+        # the durable-execution endpoint override; the role/account are used by
+        # the SDK/tooling when present.
+        LAMBDA_ENDPOINT: ${{ secrets.LAMBDA_ENDPOINT }}
+        TEST_ACCOUNT_ID: ${{ secrets.TEST_ACCOUNT_ID }}
+        TEST_LAMBDA_EXECUTION_ROLE_ARN: ${{ secrets.TEST_LAMBDA_EXECUTION_ROLE_ARN }}
       run: |
         python -m aws_durable_execution_conformance_tests.app \
           --template template_${{ matrix.suite }}.yaml \
           --language python \
           --suite ${{ matrix.suite }} \
-          --name conf-py-${{ matrix.suite }}-${{ github.run_id }} \
+          --name conf-py-${SUITE_SLUG}-${{ github.run_id }} \
           --region ${{ env.AWS_REGION }} \
           --history-dir history-${{ matrix.suite }} \
           --report junit \

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/__init__.py

@@ -0,0 +1,11 @@
+# 4-1: Create callback basic (success via external callback)
+from typing import Any
+
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(name=event)
+    return callback.result()

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_basic.py

@@ -0,0 +1,12 @@
+# 4-6: Callback failure (external system reports failure)
+from typing import Any
+
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(name=event)
+    # Do not catch — let the exception propagate so the execution fails.
+    return callback.result()

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_failure.py

@@ -0,0 +1,15 @@
+# 4-5: Create callback with heartbeat then success
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import CallbackConfig, Duration
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(
+        name=event,
+        config=CallbackConfig(heartbeat_timeout=Duration.from_seconds(10)),
+    )
+    return callback.result()

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_heartbeat_success.py

@@ -0,0 +1,15 @@
+# 4-4: Create callback heartbeat timeout (no heartbeat sent)
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import CallbackConfig, Duration
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(
+        name=event,
+        config=CallbackConfig(heartbeat_timeout=Duration.from_seconds(5)),
+    )
+    return callback.result()

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_heartbeat_timeout.py

@@ -0,0 +1,22 @@
+# 4-13: Callback failure caught → Wait → return
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import Duration
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.exceptions import CallbackError
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(name=event)
+
+    try:
+        outcome = callback.result()
+    except CallbackError as e:
+        outcome = f"caught_failure:{e}"
+    except Exception as e:  # pragma: no cover - safety net
+        outcome = f"caught_other:{type(e).__name__}:{e}"
+
+    context.wait(Duration.from_seconds(2), name="after-cb")
+    return outcome

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_replay_failure.py

@@ -0,0 +1,25 @@
+# 4-14: Callback timeout caught → Wait → return
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import CallbackConfig, Duration
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.exceptions import CallbackError
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(
+        name=event,
+        config=CallbackConfig(timeout=Duration.from_seconds(3)),
+    )
+
+    try:
+        outcome = callback.result()
+    except CallbackError as e:
+        outcome = f"caught_timeout:{e}"
+    except Exception as e:  # pragma: no cover - safety net
+        outcome = f"caught_other:{type(e).__name__}:{e}"
+
+    context.wait(Duration.from_seconds(2), name="after-cb")
+    return outcome

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_replay_timeout.py

@@ -0,0 +1,14 @@
+# 4-12: Callback success → Wait → verify replay
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import Duration
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> str:
+    callback = context.create_callback(name=event)
+    cb_result = callback.result()
+    context.wait(Duration.from_seconds(2), name="after-cb")
+    return cb_result

packages/aws-durable-execution-sdk-python-conformance-tests/handlers/callback/callback_replay_with_wait.py

@@ -0,0 +1,59 @@
+# 4-15: Callback with custom serdes (happy path - Date roundtrip)
+import json
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Any
+
+from aws_durable_execution_sdk_python.config import CallbackConfig
+from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.execution import durable_execution
+from aws_durable_execution_sdk_python.serdes import SerDes, SerDesContext
+
+
+@dataclass
+class CustomData:
+    id: int
+    message: str
+    timestamp: datetime
+
+
+class CustomDataSerDes(SerDes[CustomData]):
+    def serialize(self, value: CustomData | None, _: SerDesContext) -> str | None:
+        if value is None:
+            return None
+        return json.dumps(
+            {
+                "id": value.id,
+                "message": value.message,
+                "timestamp": value.timestamp.isoformat(),
+            }
+        )
+
+    def deserialize(self, payload: str | None, _: SerDesContext) -> CustomData | None:
+        if payload is None:
+            return None
+        data = json.loads(payload)
+        ts_str = data["timestamp"]
+        if ts_str.endswith("Z"):
+            ts_str = ts_str[:-1] + "+00:00"
+        return CustomData(
+            id=data["id"],
+            message=data["message"],
+            timestamp=datetime.fromisoformat(ts_str),
+        )
+
+
+@durable_execution
+def handler(event: Any, context: DurableContext) -> dict[str, Any]:
+    callback = context.create_callback(
+        name=event,
+        config=CallbackConfig(serdes=CustomDataSerDes()),
+    )
+    result: CustomData = callback.result()
+    return {
+        "received": {
+            "id": result.id,
+            "message": result.message,
+            "timestamp": int(result.timestamp.timestamp()),
+        },
+    }