chore: add proper testing

Author: Davide Melfi

.github/workflows/aws-lambda-java-profiler.yml

@@ -58,6 +58,10 @@ jobs:
         working-directory: ./experimental/aws-lambda-java-profiler
         run: ./integration_tests/invoke_function.sh
 
+      - name: Invoke Java Custom Options function
+        working-directory: ./experimental/aws-lambda-java-profiler
+        run: ./integration_tests/invoke_function_custom_options.sh
+
       - name: Download from s3
         working-directory: ./experimental/aws-lambda-java-profiler
         run: ./integration_tests/download_from_s3.sh

experimental/aws-lambda-java-profiler/integration_tests/create_function.sh

@@ -2,13 +2,16 @@
 
 # Set variables
 FUNCTION_NAME="aws-lambda-java-profiler-function-${GITHUB_RUN_ID}"
+FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS="aws-lambda-java-profiler-function-custom-profiler-options-${GITHUB_RUN_ID}"
 ROLE_NAME="aws-lambda-java-profiler-role-${GITHUB_RUN_ID}"
 HANDLER="helloworld.Handler::handleRequest"
 RUNTIME="java21"
 LAYER_ARN=$(cat /tmp/layer_arn)
 
 JAVA_TOOL_OPTIONS="-XX:+UnlockDiagnosticVMOptions -XX:+DebugNonSafepoints -javaagent:/opt/profiler-extension.jar"
 AWS_LAMBDA_PROFILER_RESULTS_BUCKET_NAME="aws-lambda-java-profiler-bucket-${GITHUB_RUN_ID}"
+AWS_LAMBDA_PROFILER_START_COMMAND="start,event=wall,interval=1us"
+AWS_LAMBDA_PROFILER_STOP_COMMAND="stop,file=%s,include=*AWSLambda.main,include=start_thread"
 
 # Compile the Hello World project
 cd integration_tests/helloworld
@@ -63,6 +66,19 @@ aws lambda create-function \
     --environment "Variables={JAVA_TOOL_OPTIONS='$JAVA_TOOL_OPTIONS',AWS_LAMBDA_PROFILER_RESULTS_BUCKET_NAME='$AWS_LAMBDA_PROFILER_RESULTS_BUCKET_NAME',AWS_LAMBDA_PROFILER_DEBUG='true'}" \
     --layers "$LAYER_ARN"
 
+
+# Create Lambda function custom profiler options
+aws lambda create-function \
+    --function-name "$FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS" \
+    --runtime "$RUNTIME" \
+    --role "$ROLE_ARN" \
+    --handler "$HANDLER" \
+    --timeout 30 \
+    --memory-size 512 \
+    --zip-file fileb://integration_tests/helloworld/build/distributions/code.zip \
+    --environment "Variables={JAVA_TOOL_OPTIONS='$JAVA_TOOL_OPTIONS',AWS_LAMBDA_PROFILER_RESULTS_BUCKET_NAME='$AWS_LAMBDA_PROFILER_RESULTS_BUCKET_NAME',AWS_LAMBDA_PROFILER_DEBUG='true',AWS_LAMBDA_PROFILER_START_COMMAND='$AWS_LAMBDA_PROFILER_START_COMMAND',AWS_LAMBDA_PROFILER_STOP_COMMAND='$AWS_LAMBDA_PROFILER_STOP_COMMAND'}" \
+    --layers "$LAYER_ARN"
+
 echo "Lambda function '$FUNCTION_NAME' created successfully with Java 21 runtime"
 
 echo "Waiting the function to be ready so we can invoke it..."

experimental/aws-lambda-java-profiler/integration_tests/invoke_function_custom_options.sh

@@ -0,0 +1,88 @@
+#!/bin/bash
+
+# Set variables
+FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS="aws-lambda-java-profiler-function-custom-profiler-options-${GITHUB_RUN_ID}"
+PAYLOAD='{"key": "value"}'
+
+# Expected profiler commands (should match create_function.sh)
+EXPECTED_START_COMMAND="start,event=wall,interval=1us"
+EXPECTED_STOP_COMMAND="stop,file=%s,include=*AWSLambda.main,include=start_thread"
+
+echo "Invoking Lambda function with custom profiler options: $FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS"
+
+# Invoke the Lambda function synchronously and capture the response
+RESPONSE=$(aws lambda invoke \
+    --function-name "$FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS" \
+    --payload "$PAYLOAD" \
+    --cli-binary-format raw-in-base64-out \
+    --log-type Tail \
+    output.json)
+
+# Extract the status code and log result from the response
+STATUS_CODE=$(echo "$RESPONSE" | jq -r '.StatusCode')
+LOG_RESULT=$(echo "$RESPONSE" | jq -r '.LogResult')
+
+echo "Function invocation completed with status code: $STATUS_CODE"
+
+# Decode and display the logs
+if [ -n "$LOG_RESULT" ]; then
+    echo "Function logs:"
+    echo "$LOG_RESULT" | base64 --decode
+else
+    echo "No logs available."
+fi
+
+# Display the function output
+echo "Function output:"
+cat output.json
+
+# Verify profiler started
+echo "$LOG_RESULT" | base64 --decode | grep "starting the profiler for coldstart" || exit 1
+
+# Verify custom start command is being used
+echo "$LOG_RESULT" | base64 --decode | grep "$EXPECTED_START_COMMAND" || exit 1
+
+# Verify no upload on cold start
+echo "$LOG_RESULT" | base64 --decode | grep -v "uploading" || exit 1
+
+# Clean up the output file
+rm output.json
+
+
+# Invoke it a second time for warm start
+echo "Invoking Lambda function (warm start): $FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS"
+
+# Invoke the Lambda function synchronously and capture the response
+RESPONSE=$(aws lambda invoke \
+    --function-name "$FUNCTION_NAME_CUSTOM_PROFILER_OPTIONS" \
+    --payload "$PAYLOAD" \
+    --cli-binary-format raw-in-base64-out \
+    --log-type Tail \
+    output.json)
+
+# Extract the status code and log result from the response
+STATUS_CODE=$(echo "$RESPONSE" | jq -r '.StatusCode')
+LOG_RESULT=$(echo "$RESPONSE" | jq -r '.LogResult')
+
+echo "Function invocation completed with status code: $STATUS_CODE"
+
+# Decode and display the logs
+if [ -n "$LOG_RESULT" ]; then
+    echo "Function logs:"
+    echo "$LOG_RESULT" | base64 --decode
+else
+    echo "No logs available."
+fi
+
+# Display the function output
+echo "Function output:"
+cat output.json
+
+# Verify upload happens on warm start
+echo "$LOG_RESULT" | base64 --decode | grep "uploading" || exit 1
+
+# Verify custom stop command is being used
+echo "$LOG_RESULT" | base64 --decode | grep "$EXPECTED_STOP_COMMAND" || exit 1
+
+# Clean up the output file
+rm output.json