ci: default durable examples to JSON logs

Author: zhongkechen

packages/aws-durable-execution-sdk-python-examples/scripts/generate_sam_template.py

@@ -7,6 +7,7 @@
 
 
 DEFAULT_FUNCTION_NAME_PREFIX = "DurablePythonExample-"
+DEFAULT_DURABLE_LOGGING_CONFIG: dict[str, str] = {"LogFormat": "JSON"}
 
 
 def load_catalog() -> dict[str, Any]:
@@ -77,9 +78,13 @@ def build_template(examples: list[dict[str, Any]]) -> dict[str, Any]:
 
         if "durableConfig" in example:
             properties["DurableConfig"] = example["durableConfig"]
+            properties["LoggingConfig"] = DEFAULT_DURABLE_LOGGING_CONFIG
 
         if "loggingConfig" in example:
-            properties["LoggingConfig"] = example["loggingConfig"]
+            properties["LoggingConfig"] = {
+                **DEFAULT_DURABLE_LOGGING_CONFIG,
+                **example["loggingConfig"],
+            }
 
         if "layers" in example:
             properties["Layers"] = example["layers"]

packages/aws-durable-execution-sdk-python-examples/test/test_generate_sam_template.py

@@ -0,0 +1,52 @@
+"""Tests for SAM template generation."""
+
+import importlib.util
+from pathlib import Path
+
+
+SCRIPT_PATH = (
+    Path(__file__).resolve().parents[1] / "scripts" / "generate_sam_template.py"
+)
+SPEC = importlib.util.spec_from_file_location("generate_sam_template", SCRIPT_PATH)
+assert SPEC is not None
+assert SPEC.loader is not None
+generate_sam_template = importlib.util.module_from_spec(SPEC)
+SPEC.loader.exec_module(generate_sam_template)
+
+
+def test_build_template_adds_default_json_logging_to_durable_functions():
+    template = generate_sam_template.build_template(
+        [
+            {
+                "handler": "hello_world.handler",
+                "description": "A durable example",
+                "durableConfig": {
+                    "RetentionPeriodInDays": 7,
+                    "ExecutionTimeout": 300,
+                },
+            },
+            {
+                "handler": "logger_example.handler",
+                "description": "A durable example with custom logging",
+                "durableConfig": {
+                    "RetentionPeriodInDays": 7,
+                    "ExecutionTimeout": 300,
+                },
+                "loggingConfig": {"ApplicationLogLevel": "INFO"},
+            },
+            {
+                "handler": "plain_lambda.handler",
+                "description": "A non-durable example",
+            },
+        ]
+    )
+
+    resources = template["Resources"]
+    assert resources["HelloWorld"]["Properties"]["LoggingConfig"] == {
+        "LogFormat": "JSON"
+    }
+    assert resources["LoggerExample"]["Properties"]["LoggingConfig"] == {
+        "LogFormat": "JSON",
+        "ApplicationLogLevel": "INFO",
+    }
+    assert "LoggingConfig" not in resources["PlainLambda"]["Properties"]