Use default host and port if endpoint not specified

Author: GarrettBeatty

Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Processes/ApiGatewayEmulatorProcess.cs

@@ -96,7 +96,7 @@ public static ApiGatewayEmulatorProcess Startup(RunCommandSettings settings, Can
                 PayloadStream = lambdaRequestStream
             };
 
-            using var lambdaClient = CreateLambdaServiceClient(routeConfig);
+            using var lambdaClient = CreateLambdaServiceClient(routeConfig, settings);
             var response =  await lambdaClient.InvokeAsync(invokeRequest);
 
             if (response.FunctionError != null)
@@ -111,13 +111,11 @@ public static ApiGatewayEmulatorProcess Startup(RunCommandSettings settings, Can
             {
                 var lambdaResponse = response.ToApiGatewayHttpApiV2ProxyResponse();
                 await lambdaResponse.ToHttpResponseAsync(context);
-                return;
             }
             else
             {
                 var lambdaResponse = response.ToApiGatewayProxyResponse(settings.ApiGatewayEmulatorMode.Value);
                 await lambdaResponse.ToHttpResponseAsync(context, settings.ApiGatewayEmulatorMode.Value);
-                return;
             }
         });
 
@@ -131,12 +129,27 @@ public static ApiGatewayEmulatorProcess Startup(RunCommandSettings settings, Can
         };
     }
 
-    private static IAmazonLambda CreateLambdaServiceClient(ApiGatewayRouteConfig routeConfig)
+    /// <summary>
+    /// Creates an Amazon Lambda service client with the specified configuration.
+    /// </summary>
+    /// <param name="routeConfig">The API Gateway route configuration containing the endpoint information.
+    /// If the endpoint is specified in routeConfig, it will be used as the service URL.</param>
+    /// <param name="settings">The run command settings containing host and port information.
+    /// If routeConfig endpoint is null, the service URL will be constructed using settings.Host and settings.Port.</param>
+    /// <returns>An instance of IAmazonLambda configured with the specified endpoint and credentials.</returns>
+    /// <remarks>
+    /// The function uses hard-coded AWS credentials ("accessKey", "secretKey") for authentication since they are not actually being used.
+    /// The service URL is determined by either:
+    /// - Using routeConfig.Endpoint if it's not null
+    /// - Combining settings.Host and settings.Port if routeConfig.Endpoint is null
+    /// </remarks>
+    private static IAmazonLambda CreateLambdaServiceClient(ApiGatewayRouteConfig routeConfig, RunCommandSettings settings)
     {
-        // TODO: Handle routeConfig.Endpoint to null and use the settings versions of runtime.
+        var endpoint = routeConfig.Endpoint ?? $"http://{settings.Host}:{settings.Port}";
+
         var lambdaConfig = new AmazonLambdaConfig
         {
-            ServiceURL = routeConfig.Endpoint
+            ServiceURL = endpoint
         };
 
         return new AmazonLambdaClient(new Amazon.Runtime.BasicAWSCredentials("accessKey", "secretKey"), lambdaConfig);

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

@@ -38,7 +38,7 @@ public async Task TestLambdaToUpperV2()
 
         try
         {
-            await StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
+            StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
             await WaitForGatewayHealthCheck();
             await StartLambdaProcess(config);
 
@@ -69,7 +69,7 @@ public async Task TestLambdaToUpperRest()
 
         try
         {
-            await StartTestToolProcess(ApiGatewayEmulatorMode.Rest, config);
+            StartTestToolProcess(ApiGatewayEmulatorMode.Rest, config);
             await WaitForGatewayHealthCheck();
             await StartLambdaProcess(config);
 
@@ -100,7 +100,7 @@ public async Task TestLambdaToUpperV1()
 
         try
         {
-            await StartTestToolProcess(ApiGatewayEmulatorMode.HttpV1, config);
+            StartTestToolProcess(ApiGatewayEmulatorMode.HttpV1, config);
             await WaitForGatewayHealthCheck();
             await StartLambdaProcess(config);
 
@@ -131,7 +131,7 @@ public async Task TestLambdaBinaryResponse()
 
         try
         {
-            await StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
+            StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
             await WaitForGatewayHealthCheck();
             await StartLambdaProcess(config);
 
@@ -168,7 +168,38 @@ public async Task TestLambdaReturnString()
 
         try
         {
-            await StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
+            StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config);
+            await WaitForGatewayHealthCheck();
+            await StartLambdaProcess(config);
+
+            var response = await TestEndpoint(config);
+            var responseContent = await response.Content.ReadAsStringAsync();
+
+            Assert.Equal(HttpStatusCode.OK, response.StatusCode);
+            Assert.Equal("HELLO WORLD", responseContent);
+        }
+        finally
+        {
+            await CleanupProcesses();
+        }
+    }
+
+    [Fact]
+    public async Task TestLambdaWithNullEndpoint()
+    {
+        var testProjectDir = Path.GetFullPath("../../../../");
+        var config = new TestConfig
+        {
+            TestToolPath = Path.GetFullPath(Path.Combine(testProjectDir, "../src/Amazon.Lambda.TestTool")),
+            LambdaPath = Path.GetFullPath(Path.Combine(testProjectDir, "LambdaTestFunctionV2/src/LambdaTestFunctionV2")),
+            FunctionName = "LambdaTestFunctionV2",
+            RouteName = "testfunction",
+            HttpMethod = "Post"
+        };
+
+        try
+        {
+            StartTestToolProcess(ApiGatewayEmulatorMode.HttpV2, config, useNullEndpoint: true);
             await WaitForGatewayHealthCheck();
             await StartLambdaProcess(config);
 
@@ -205,7 +236,7 @@ private async Task<HttpResponseMessage> TestEndpoint(TestConfig config, HttpCont
         };
     }
 
-    private async Task StartTestToolProcess(ApiGatewayEmulatorMode apiGatewayMode, TestConfig config)
+    private void StartTestToolProcess(ApiGatewayEmulatorMode apiGatewayMode, TestConfig config, bool useNullEndpoint = false)
     {
         var startInfo = new ProcessStartInfo
         {
@@ -217,14 +248,21 @@ private async Task StartTestToolProcess(ApiGatewayEmulatorMode apiGatewayMode, T
             CreateNoWindow = true
         };
 
-        startInfo.EnvironmentVariables["ASPNETCORE_ENVIRONMENT"] = "Development";
-        startInfo.EnvironmentVariables["APIGATEWAY_EMULATOR_ROUTE_CONFIG"] = $@"{{
+        var routeConfig = useNullEndpoint
+            ? $@"{{
+            ""LambdaResourceName"": ""{config.RouteName}"",
+            ""HttpMethod"": ""{config.HttpMethod}"",
+            ""Path"": ""/{config.RouteName}""
+        }}"
+            : $@"{{
             ""LambdaResourceName"": ""{config.RouteName}"",
             ""Endpoint"": ""http://localhost:{LambdaPort}"",
             ""HttpMethod"": ""{config.HttpMethod}"",
             ""Path"": ""/{config.RouteName}""
         }}";
 
+        startInfo.EnvironmentVariables["APIGATEWAY_EMULATOR_ROUTE_CONFIG"] = routeConfig;
+
         _mainProcess = Process.Start(startInfo) ?? throw new Exception("Failed to start test tool process");
         ConfigureProcessLogging(_mainProcess, "TestTool");
     }