fix(integration-tests): address review feedback

- Remove unused payload parsing in invoke.ts
- Implement SLEEP_MS support in default handlers for trace isolation testing
- Remove unused RETURN_REQUEST_ID env var from lmi.ts
- Fix missing await in forceColdStart causing race condition
- Fix grammar typo in datadog.ts

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: jchrostek-dd

integration-tests/lambda/default-dotnet/Function.cs

@@ -1,6 +1,8 @@
 using Amazon.Lambda.Core;
+using System;
 using System.Collections.Generic;
 using System.Text.Json;
+using System.Threading;
 
 [assembly: LambdaSerializer(typeof(Amazon.Lambda.Serialization.SystemTextJson.DefaultLambdaJsonSerializer))]
 
@@ -12,6 +14,13 @@ public Dictionary<string, object> FunctionHandler(JsonElement input, ILambdaCont
         {
             context.Logger.LogLine("Hello world!");
 
+            var sleepMsStr = Environment.GetEnvironmentVariable("SLEEP_MS");
+            if (!string.IsNullOrEmpty(sleepMsStr) && int.TryParse(sleepMsStr, out int sleepMs) && sleepMs > 0)
+            {
+                context.Logger.LogLine($"Sleeping for {sleepMs}ms");
+                Thread.Sleep(sleepMs);
+            }
+
             var body = new Dictionary<string, object>
             {
                 { "message", "Success" },

integration-tests/lambda/default-java/src/main/java/example/Handler.java

@@ -14,6 +14,21 @@ public class Handler implements RequestHandler<Map<String, Object>, Map<String,
     public Map<String, Object> handleRequest(Map<String, Object> event, Context context) {
         context.getLogger().log("Hello world!");
 
+        String sleepMsStr = System.getenv("SLEEP_MS");
+        if (sleepMsStr != null && !sleepMsStr.isEmpty()) {
+            try {
+                int sleepMs = Integer.parseInt(sleepMsStr);
+                if (sleepMs > 0) {
+                    context.getLogger().log("Sleeping for " + sleepMs + "ms");
+                    Thread.sleep(sleepMs);
+                }
+            } catch (InterruptedException e) {
+                Thread.currentThread().interrupt();
+            } catch (NumberFormatException e) {
+                context.getLogger().log("Invalid SLEEP_MS value: " + sleepMsStr);
+            }
+        }
+
         Map<String, Object> body = new HashMap<>();
         body.put("message", "Success");
         body.put("requestId", context.getAwsRequestId());

integration-tests/lambda/default-node/index.js

@@ -1,6 +1,12 @@
 exports.handler = async (event, context) => {
   console.log('Hello world!');
 
+  const sleepMs = parseInt(process.env.SLEEP_MS || '0', 10);
+  if (sleepMs > 0) {
+    console.log(`Sleeping for ${sleepMs}ms`);
+    await new Promise(resolve => setTimeout(resolve, sleepMs));
+  }
+
   return {
     statusCode: 200,
     body: JSON.stringify({

integration-tests/lambda/default-python/lambda_function.py

@@ -1,5 +1,7 @@
 import json
 import logging
+import os
+import time
 
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
@@ -8,6 +10,11 @@
 def handler(event, context):
     logger.info('Hello world!')
 
+    sleep_ms = int(os.environ.get('SLEEP_MS', '0'))
+    if sleep_ms > 0:
+        logger.info(f'Sleeping for {sleep_ms}ms')
+        time.sleep(sleep_ms / 1000)
+
     return {
         'statusCode': 200,
         'body': json.dumps({

integration-tests/lib/stacks/lmi.ts

@@ -41,7 +41,6 @@ export class LambdaManagedInstancesStack extends cdk.Stack {
         DD_SERVICE: nodeFunctionName,
         DD_TRACE_ENABLED: 'true',
         DD_LAMBDA_HANDLER: 'index.handler',
-        RETURN_REQUEST_ID: 'true',
       },
       logGroup: createLogGroup(this, nodeFunctionName)
     });
@@ -69,7 +68,6 @@ export class LambdaManagedInstancesStack extends cdk.Stack {
         DD_TRACE_ENABLED: 'true',
         DD_LAMBDA_HANDLER: 'lambda_function.handler',
         DD_TRACE_AGENT_URL: 'http://127.0.0.1:8126',
-        RETURN_REQUEST_ID: 'true',
       },
       logGroup: createLogGroup(this, pythonFunctionName)
     });
@@ -96,7 +94,6 @@ export class LambdaManagedInstancesStack extends cdk.Stack {
         AWS_LAMBDA_EXEC_WRAPPER: '/opt/datadog_wrapper',
         DD_SERVICE: javaFunctionName,
         DD_TRACE_ENABLED: 'true',
-        RETURN_REQUEST_ID: 'true',
       },
       logGroup: createLogGroup(this, javaFunctionName)
     });
@@ -123,7 +120,6 @@ export class LambdaManagedInstancesStack extends cdk.Stack {
         AWS_LAMBDA_EXEC_WRAPPER: '/opt/datadog_wrapper',
         DD_SERVICE: dotnetFunctionName,
         DD_TRACE_ENABLED: 'true',
-        RETURN_REQUEST_ID: 'true',
       },
       logGroup: createLogGroup(this, dotnetFunctionName)
     });

integration-tests/tests/utils/datadog.ts

@@ -70,7 +70,7 @@ export async function getTraces(
   requestId: string,
 ): Promise<DatadogTrace[]> {
   const now = Date.now();
-  const fromTime = now - (1 * 60 * 60 * 1000); // 1 hours ago
+  const fromTime = now - (1 * 60 * 60 * 1000); // 1 hour ago
   const toTime = now;
   try {
     // Convert service name to lowercase as Datadog stores it that way

integration-tests/tests/utils/invoke.ts

@@ -25,16 +25,6 @@ export async function invokeLambda(
     throw error;
   }
 
-  let responsePayload;
-  try {
-    responsePayload = JSON.parse(new TextDecoder().decode(response.Payload));
-    console.log(`Response payload: ${JSON.stringify(responsePayload)}`);
-  } catch (error: any) {
-    console.error('Failed to parse response payload:', error.message);
-    console.log('Raw payload:', new TextDecoder().decode(response.Payload));
-    throw error;
-  }
-
   const requestId: string = response.$metadata.requestId || '';
 
   return {

integration-tests/tests/utils/lambda.ts

@@ -3,7 +3,7 @@ import { LambdaClient, UpdateFunctionConfigurationCommand, GetFunctionConfigurat
 const lambdaClient = new LambdaClient({ region: 'us-east-1' });
 
 export async function forceColdStart(functionName: string): Promise<void> {
-  setTimestampEnvVar(functionName)
+  await setTimestampEnvVar(functionName);
   await new Promise(resolve => setTimeout(resolve, 10000));
 }