PR comments

Author: GarrettBeatty

Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Components/Pages/Home.razor.cs

@@ -352,11 +352,14 @@ private async Task<bool> InvokeLambdaFunction(string payload)
         }
         catch (AmazonLambdaException e)
         {
+            Logger.LogInformation(e.Message);
+
             // lambda client automatically adds some extra verbiage: "The service returned an error with Error Code xxxx and HTTP Body: <bodyhere>".
             // removing the extra verbiage to make the error message smaller and look better on the ui.
             _errorMessage = e.Message.Contains("HTTP Body: ")
                 ? e.Message.Split("HTTP Body: ")[1]
-                : e.Message;        }
+                : e.Message;
+        }
         return false;
     }
 }

Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Extensions/InvokeResponseExtensions.cs

@@ -87,13 +87,19 @@ public static APIGatewayProxyResponse ToApiGatewayErrorResponse(ApiGatewayEmulat
 
     /// <summary>
     /// Creates a standard API Gateway response for a "Request Entity Too Large" (413) error.
+    /// Not compatible with HTTP V2 API Gateway mode.
     /// </summary>
-    /// <param name="emulatorMode">The API Gateway emulator mode (Rest or Http).</param>
+    /// <param name="emulatorMode">The API Gateway emulator mode (Rest or HttpV1 only).</param>
     /// <returns>An APIGatewayProxyResponse configured with:
     /// - Status code 413
-    /// - JSON error message
+    /// - JSON error message ("Request Too Long" for REST, "Request Entity Too Large" for HTTP V1)
     /// - Content-Type header set to application/json
     /// </returns>
+    /// <exception cref="InvalidOperationException">Thrown when emulatorMode is HttpV2 or invalid value</exception>
+    /// <remarks>
+    /// This method only supports REST and HTTP V1 API Gateway modes. For HTTP V2,
+    /// use <seealso cref="ToHttpApiV2RequestTooLargeResponse"/>.
+    /// </remarks>
     public static APIGatewayProxyResponse ToHttpApiRequestTooLargeResponse(ApiGatewayEmulatorMode emulatorMode)
     {
         if (emulatorMode == ApiGatewayEmulatorMode.Rest)
@@ -122,7 +128,7 @@ public static APIGatewayProxyResponse ToHttpApiRequestTooLargeResponse(ApiGatewa
                 IsBase64Encoded = false
             };
         }
-        throw new InvalidOperationException();
+        throw new ArgumentException($"Unsupported API Gateway emulator mode: {emulatorMode}. Only Rest and HttpV1 modes are supported.");
     }
 
     /// <summary>

Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Models/LambdaOptions.cs

@@ -3,7 +3,16 @@
 
 namespace Amazon.Lambda.TestTool.Models;
 
+/// <summary>
+/// Configuration options for invoking lambda functions.
+/// </summary>
 public class LambdaOptions
 {
+    /// <summary>
+    /// Gets or sets the endpoint URL for Lambda function invocations.
+    /// </summary>
+    /// <value>
+    /// A string containing the endpoint URL. Defaults to an empty string.
+    /// </value>
     public string Endpoint { get; set; } = string.Empty;
 }

Tools/LambdaTestTool-v2/tests/Amazon.Lambda.TestTool.IntegrationTests/ApiGatewayEmulatorProcessTests.cs

@@ -301,17 +301,15 @@ public async Task TestLambdaWithLargeRequestPayload_RestAndV1(ApiGatewayEmulator
 
             var handler = (APIGatewayProxyRequest request, ILambdaContext context) =>
             {
-                var env = Environment.GetEnvironmentVariable("AWS_LAMBDA_RUNTIME_API");
-                testOutputHelper.WriteLine($"TestLambdaWithLargeRequestPayload {mode}: {env}");
                 return new APIGatewayProxyResponse
                 {
                     StatusCode = 200,
                     Body = request.Body.Length.ToString()
                 };
             };
 
-            Environment.SetEnvironmentVariable("AWS_LAMBDA_RUNTIME_API", $"localhost:{lambdaPort}/largerequestfunction");
             _ = LambdaBootstrapBuilder.Create(handler, new DefaultLambdaJsonSerializer())
+                .ConfigureOptions(x => x.RuntimeApiEndpoint = $"localhost:{lambdaPort}/largerequestfunction")
                 .Build()
                 .RunAsync(_cancellationTokenSource.Token);
 
@@ -346,17 +344,15 @@ public async Task TestLambdaWithLargeRequestPayload_HttpV2()
 
             var handler = (APIGatewayHttpApiV2ProxyRequest request, ILambdaContext context) =>
             {
-                var env = Environment.GetEnvironmentVariable("AWS_LAMBDA_RUNTIME_API");
-                testOutputHelper.WriteLine($"TestLambdaWithLargeRequestPayload HttpV2: {env}");
                 return new APIGatewayHttpApiV2ProxyResponse
                 {
                     StatusCode = 200,
                     Body = request.Body.Length.ToString()
                 };
             };
 
-            Environment.SetEnvironmentVariable("AWS_LAMBDA_RUNTIME_API", $"localhost:{lambdaPort}/largerequestfunction");
             _ = LambdaBootstrapBuilder.Create(handler, new DefaultLambdaJsonSerializer())
+                .ConfigureOptions(x => x.RuntimeApiEndpoint = $"localhost:{lambdaPort}/largerequestfunction")
                 .Build()
                 .RunAsync(_cancellationTokenSource.Token);