chore: makefile and test scenario

Author: darklight3it

Makefile

@@ -14,7 +14,7 @@ HANDLER ?=
 -include .env
 export
 
-.PHONY: help pr-check integration-tests check-event-features fmt build-examples test-rie test-rie-lmi nuke test-dockerized
+.PHONY: help pr-check integration-tests check-event-features fmt build-examples build-test-runner test-rie test-rie-lmi nuke test-dockerized test-dockerized-concurrent
 
 .DEFAULT_GOAL := help
 
@@ -129,23 +129,39 @@ build-examples:
 nuke:
 	docker kill $$(docker ps -q)
 
-test-dockerized: build-examples
-	@echo "Running dockerized tests locally..."
-	
+build-test-runner: build-examples
 	@echo "Building base Docker image with RIE and custom entrypoint..."
 	docker build \
 	-t local/test-base \
 	-f Dockerfile.test \
 	.
-	
+
 	@echo "Setting up containerized test runner..."
 	@if [ ! -d ".test-runner" ]; then \
 		echo "Cloning containerized-test-runner-for-aws-lambda..."; \
 		git clone --quiet https://github.com/aws/containerized-test-runner-for-aws-lambda.git .test-runner; \
 	fi
 	@echo "Building test runner Docker image..."
 	@docker build -t test-runner:local -f .test-runner/Dockerfile .test-runner
-	
+
+test-dockerized-concurrent: build-test-runner
+	@echo "Running concurrent scenarios in Docker..."
+	@docker network create concurrent-test-net 2>/dev/null || true
+	@docker run --rm \
+		--network concurrent-test-net \
+		-e INPUT_SUITE_FILE_ARRAY='[]' \
+		-e INPUT_SCENARIO_DIR=/workspace/test/dockerized/scenarios \
+		-e DOCKER_IMAGE_NAME=local/test-base \
+		-e TASK_FOLDER=./test/dockerized/tasks \
+		-e GITHUB_WORKSPACE=/workspace \
+		-e DOCKER_SHARED_NETWORK=concurrent-test-net \
+		-v /var/run/docker.sock:/var/run/docker.sock \
+		-v "$(CURDIR):/workspace" \
+		-w /workspace \
+		test-runner:local
+	@docker network rm --force concurrent-test-net 2>/dev/null || true
+
+test-dockerized: build-test-runner
 	@echo "Running tests in Docker..."
 	@docker run --rm \
 		-e INPUT_SUITE_FILE_ARRAY='["./test/dockerized/suites/*.json"]' \
@@ -179,6 +195,7 @@ help: ## Show this help message
 	@echo '                        Usage: HANDLERS_TO_BUILD="basic-lambda" HANDLER="basic-lambda" make test-rie'
 	@echo '  test-rie-lmi          Test RIE in Lambda Managed Instance mode'
 	@echo '                        Usage: RIE_MAX_CONCURRENCY=4 HANDLERS_TO_BUILD="basic-lambda-concurrent" make test-rie-lmi'
+	@echo '  test-dockerized-concurrent Run concurrent LMI test scenarios'
 	@echo '  test-dockerized       Run dockerized test harness'
 	@echo '  nuke                  Kill all running Docker containers'
 	@echo ''

test/dockerized/scenarios/lmi_scenarios.py

@@ -0,0 +1,64 @@
+"""
+Multi-concurrency test scenarios for basic-lambda-concurrent.
+
+The handler expects: { "command": "<string>" }
+and responds with:   { "req_id": "<id>", "msg": "Command <command> executed." }
+"""
+
+import os
+from containerized_test_runner.models import Request, ConcurrentTest
+
+HANDLER = "basic-lambda-concurrent"
+IMAGE = os.environ.get("TEST_IMAGE", "local/test-base")
+DEFAULT_CONCURRENCY = 1
+
+
+def _make_env(concurrency: int = DEFAULT_CONCURRENCY) -> dict:
+    return {
+        "_HANDLER": HANDLER,
+        "AWS_LAMBDA_MAX_CONCURRENCY": str(concurrency),
+        "AWS_LAMBDA_LOG_FORMAT": "JSON",
+    }
+
+
+def get_lmi_scenarios():
+    scenarios = []
+
+    # Happy path: DEFAULT_CONCURRENCY unique commands all succeed concurrently
+    batch = [
+        Request(
+            payload={"command": f"task-{i}"},
+            assertions=[{"transform": "{msg: .msg}", "response": {"msg": f"Command task-{i} executed."}}],
+        )
+        for i in range(DEFAULT_CONCURRENCY)
+    ]
+    scenarios.append(ConcurrentTest(
+        name="lmi_concurrent_happy_path",
+        handler=HANDLER,
+        environment_variables=_make_env(),
+        request_batches=[batch],
+        image=IMAGE,
+    ))
+
+    # Error isolation: 9 invalid payloads + 1 valid — the valid one must still succeed
+    mixed_batch = [
+        Request(
+            payload={"not_a_command": "oops"},
+            assertions=[{"transform": ".errorType", "error": "Runtime.UnmarshalError"}],
+        )
+        for _ in range(DEFAULT_CONCURRENCY - 1)
+    ] + [
+        Request(
+            payload={"command": "survivor"},
+            assertions=[{"transform": "{msg: .msg}", "response": {"msg": "Command survivor executed."}}],
+        )
+    ]
+    scenarios.append(ConcurrentTest(
+        name="lmi_error_isolation",
+        handler=HANDLER,
+        environment_variables=_make_env(),
+        request_batches=[mixed_batch],
+        image=IMAGE,
+    ))
+
+    return scenarios