chore(sdk-testing): re-factor testing examples

- Refactor testing examples
- Remove InvokeAccount for testing cases

Author: Alex Wang

.github/workflows/deploy-examples.yml

@@ -75,7 +75,6 @@ jobs:
       env:
         AWS_ACCOUNT_ID: ${{ secrets.AWS_ACCOUNT_ID }}
         LAMBDA_ENDPOINT: ${{ secrets.LAMBDA_ENDPOINT_BETA }}
-        INVOKE_ACCOUNT_ID: ${{ secrets.INVOKE_ACCOUNT_ID_BETA }}
         KMS_KEY_ARN: ${{ secrets.KMS_KEY_ARN }}
       run: |
         # Build function name

.gitignore

@@ -30,5 +30,7 @@ dist/
 .idea
 .env
 
+.durable_executions
+
 examples/build/*
 examples/*.zip

examples/cli.py

@@ -66,7 +66,9 @@ def build_examples():
 
     # Copy example functions
     logger.info("Copying examples from %s", src_dir)
-    shutil.copytree(src_dir, build_dir / "src")
+    for file_path in src_dir.rglob("*"):
+        if file_path.is_file():
+            shutil.copy2(file_path, build_dir / file_path.name)
 
     logger.info("Build completed successfully")
     return True
@@ -296,12 +298,11 @@ def get_aws_config():
         "region": os.getenv("AWS_REGION", "us-west-2"),
         "lambda_endpoint": os.getenv("LAMBDA_ENDPOINT"),
         "account_id": os.getenv("AWS_ACCOUNT_ID"),
-        "invoke_account_id": os.getenv("INVOKE_ACCOUNT_ID"),
         "kms_key_arn": os.getenv("KMS_KEY_ARN"),
     }
 
     if not all(
-        [config["account_id"], config["lambda_endpoint"], config["invoke_account_id"]]
+        [config["account_id"], config["lambda_endpoint"]]
     ):
         msg = "Missing required environment variables"
         raise ValueError(msg)
@@ -406,26 +407,6 @@ def deploy_function(example_name: str, function_name: str | None = None):
     except lambda_client.exceptions.ResourceNotFoundException:
         lambda_client.create_function(**function_config, Code={"ZipFile": zip_content})
 
-    # Update invoke permission for worker account using put_resource_policy
-    function_arn = f"arn:aws:lambda:{config['region']}:{config['account_id']}:function:{function_name}"
-
-    policy_document = {
-        "Version": "2012-10-17",
-        "Statement": [
-            {
-                "Sid": "dex-invoke-permission",
-                "Effect": "Allow",
-                "Principal": {"AWS": config["invoke_account_id"]},
-                "Action": "lambda:InvokeFunction",
-                "Resource": f"{function_arn}:*",
-            }
-        ],
-    }
-
-    lambda_client.put_resource_policy(
-        ResourceArn=function_arn, Policy=json.dumps(policy_document)
-    )
-
     logger.info("Function deployed successfully! %s", function_name)
     return True
 

examples/examples-catalog.json

@@ -21,7 +21,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/step.py"
+      "path": "./src/step/step.py"
     },
     {
       "name": "Step with Name",
@@ -32,7 +32,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/step_with_name.py"
+      "path": "./src/step/step_with_name.py"
     },
     {
       "name": "Step with Retry",
@@ -43,7 +43,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/step_with_retry.py"
+      "path": "./src/step/step_with_retry.py"
     },
     {
       "name": "Wait State",
@@ -54,7 +54,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/wait.py"
+      "path": "./src/wait/wait.py"
     },
     {
       "name": "Callback",
@@ -65,7 +65,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/callback.py"
+      "path": "./src/callback/callback.py"
     },
     {
       "name": "Wait for Callback",
@@ -76,7 +76,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/wait_for_callback.py"
+      "path": "./src/wait_for_callback/wait_for_callback.py"
     },
     {
       "name": "Run in Child Context",
@@ -87,7 +87,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/run_in_child_context.py"
+      "path": "./src/run_in_child_context/run_in_child_context.py"
     },
     {
       "name": "Parallel Operations",
@@ -98,7 +98,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/parallel.py"
+      "path": "./src/parallel/parallel.py"
     },
     {
       "name": "Map Operations",
@@ -109,7 +109,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/map_operations.py"
+      "path": "./src/map/map_operations.py"
     },
     {
       "name": "Block Example",
@@ -120,7 +120,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/block_example.py"
+      "path": "./src/block_example/block_example.py"
     },
     {
       "name": "Logger Example",
@@ -131,7 +131,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/logger_example.py"
+      "path": "./src/logger_example/logger_example.py"
     },
     {
       "name": "Steps with Retry",
@@ -142,7 +142,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/steps_with_retry.py"
+      "path": "./src/step/steps_with_retry.py"
     },
     {
       "name": "Wait for Condition",
@@ -153,7 +153,7 @@
         "RetentionPeriodInDays": 7,
         "ExecutionTimeout": 300
       },
-      "path": "./src/wait_for_condition.py"
+      "path": "./src/wait_for_condition/wait_for_condition.py"
     }
   ]
 }