Add new envs, update function to dump a payload (#6)

chris11-taylor-nttd · web-flow · commit 83edee8e773d · 2025-06-16T13:57:56.000-05:00
* Add new envs, update function to dump a payload

* Fix test, envs

* Update all function names

* Fix handler references
diff --git a/platform/production/us-east-1/000/terraform.tfvars b/platform/production/us-east-1/000/terraform.tfvars
@@ -0,0 +1,10 @@
+create_package = true
+create         = true
+handler        = "app.lambda_handler"
+cors           = { allow_origins = ["*"] }
+source_path    = "../../../../../../../src/function/"
+name           = "platform_sample-useast1-production-000-fn-000"
+environment_variables = {
+  PAYLOAD = "Primary"
+}
+
diff --git a/platform/production/us-east-1/000/terragrunt.hcl b/platform/production/us-east-1/000/terragrunt.hcl
@@ -0,0 +1,7 @@
+include "root" {
+  path = find_in_parent_folders()
+}
+
+terraform {
+  source = "git::https://github.com/launchbynttdata/tf-aws-module_primitive-lambda_function//.?ref=1.0.3"
+}
diff --git a/platform/production/us-east-2/000/terraform.tfvars b/platform/production/us-east-2/000/terraform.tfvars
@@ -3,4 +3,7 @@ create         = true
 handler        = "app.lambda_handler"
 cors           = { allow_origins = ["*"] }
 source_path    = "../../../../../../../src/function/"
-name           = "platform-sample-lambda-function"
+name           = "platform_sample-useast2-production-000-fn-000"
+environment_variables = {
+  PAYLOAD = "Failover"
+}
diff --git a/platform/sandbox/us-east-2/000/terraform.tfvars b/platform/sandbox/us-east-2/000/terraform.tfvars
@@ -3,4 +3,7 @@ create         = true
 handler        = "app.lambda_handler"
 cors           = { allow_origins = ["*"] }
 source_path    = "../../../../../../../src/function/"
-name           = "platform-sample-lambda-function"
+name           = "platform_sample-useast2-sandbox-000-fn-000"
+environment_variables = {
+  PAYLOAD = "Primary-East"
+}
diff --git a/platform/sandbox/us-east-2/001/terraform.tfvars b/platform/sandbox/us-east-2/001/terraform.tfvars
@@ -1,6 +1,9 @@
 create_package = true
 create         = true
-handler        = "app.alternate_lambda_handler"
+handler        = "app.lambda_handler"
 cors           = { allow_origins = ["*"] }
 source_path    = "../../../../../../../src/function/"
-name           = "platform-sample-lambda-function-2"
+name           = "platform_sample-useast2-sandbox-001-fn-000"
+environment_variables = {
+  PAYLOAD = "Failover-East"
+}
diff --git a/platform/sandbox/us-west-2/000/terraform.tfvars b/platform/sandbox/us-west-2/000/terraform.tfvars
@@ -0,0 +1,9 @@
+create_package = true
+create         = true
+handler        = "app.lambda_handler"
+cors           = { allow_origins = ["*"] }
+source_path    = "../../../../../../../src/function/"
+name           = "platform_sample-uswest2-sandbox-000-fn-000"
+environment_variables = {
+  PAYLOAD = "Primary-West"
+}
diff --git a/platform/sandbox/us-west-2/000/terragrunt.hcl b/platform/sandbox/us-west-2/000/terragrunt.hcl
@@ -0,0 +1,7 @@
+include "root" {
+  path = find_in_parent_folders()
+}
+
+terraform {
+  source = "git::https://github.com/launchbynttdata/tf-aws-module_primitive-lambda_function//.?ref=1.0.3"
+}
diff --git a/platform/sandbox/us-west-2/001/terraform.tfvars b/platform/sandbox/us-west-2/001/terraform.tfvars
@@ -0,0 +1,9 @@
+create_package = true
+create         = true
+handler        = "app.lambda_handler"
+cors           = { allow_origins = ["*"] }
+source_path    = "../../../../../../../src/function/"
+name           = "platform_sample-uswest2-sandbox-001-fn-000"
+environment_variables = {
+  PAYLOAD = "Failover-West"
+}
diff --git a/platform/sandbox/us-west-2/001/terragrunt.hcl b/platform/sandbox/us-west-2/001/terragrunt.hcl
@@ -0,0 +1,7 @@
+include "root" {
+  path = find_in_parent_folders()
+}
+
+terraform {
+  source = "git::https://github.com/launchbynttdata/tf-aws-module_primitive-lambda_function//.?ref=1.0.3"
+}
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,6 +10,7 @@ dependencies = []
 dev = [
     "pytest>=8.4.0",
     "pytest-cov>=6.2.1",
+    "pytest-mock>=3.14.1",
     "ruff>=0.11.13",
 ]
 
@@ -52,4 +53,4 @@ exclude = [
     "node_modules",
     "site-packages",
     ".venv",
-]
+]
diff --git a/src/function/app.py b/src/function/app.py
@@ -1,6 +1,10 @@
-def lambda_handler(event, context):
-    return {"statusCode": 200, "body": "Hello from a sample Python Lambda!"}
+import os
+
+PAYLOAD = os.environ.get("PAYLOAD")
 
 
-def alternate_lambda_handler(event, context):
-    return {"statusCode": 200, "body": "Hello from a different sample Python Lambda!"}
+def lambda_handler(event, context):
+    return {
+        "statusCode": 200,
+        "body": f"Hello from a sample Python Lambda! Payload: {PAYLOAD}",
+    }
diff --git a/test/unit/test_app.py b/test/unit/test_app.py
@@ -1,13 +1,18 @@
+from unittest.mock import patch
+
 from function import app
 
 
 def test_primary_handler():
     result = app.lambda_handler(None, None)
     assert result["statusCode"] == 200
-    assert result["body"] == "Hello from a sample Python Lambda!"
+    assert result["body"] == "Hello from a sample Python Lambda! Payload: None"
 
 
-def test_alternate_handler():
-    result = app.alternate_lambda_handler(None, None)
-    assert result["statusCode"] == 200
-    assert result["body"] == "Hello from a different sample Python Lambda!"
+def test_primary_handler_with_payload():
+    with patch.object(app, "PAYLOAD", "test_payload"):
+        result = app.lambda_handler(None, None)
+        assert result["statusCode"] == 200
+        assert (
+            result["body"] == "Hello from a sample Python Lambda! Payload: test_payload"
+        )
diff --git a/uv.lock b/uv.lock
