update test

Author: GarrettBeatty

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

@@ -21,6 +21,7 @@ public class ApiGatewayEmulatorProcessTests : IAsyncDisposable
     private readonly Mock<IRemainingArguments> _mockRemainingArgs = new Mock<IRemainingArguments>();
     private readonly ITestOutputHelper _testOutputHelper;
     private Process? _lambdaProcess;
+    private readonly List<string> _setEnvironmentVariables = new List<string>();
 
     public ApiGatewayEmulatorProcessTests(ITestOutputHelper testOutputHelper)
     {
@@ -59,7 +60,6 @@ public async Task TestLambdaToUpperV2()
         finally
         {
             await cancellationTokenSource.CancelAsync();
-            await CleanupProcesses();
         }
     }
 
@@ -95,7 +95,6 @@ public async Task TestLambdaToUpperRest()
         finally
         {
             await cancellationTokenSource.CancelAsync();
-            await CleanupProcesses();
         }
     }
 
@@ -131,7 +130,6 @@ public async Task TestLambdaToUpperV1()
         finally
         {
             await cancellationTokenSource.CancelAsync();
-            await CleanupProcesses();
         }
     }
 
@@ -173,7 +171,6 @@ public async Task TestLambdaBinaryResponse()
         finally
         {
             await cancellationTokenSource.CancelAsync();
-            await CleanupProcesses();
         }
     }
 
@@ -209,7 +206,39 @@ public async Task TestLambdaReturnString()
         finally
         {
             await cancellationTokenSource.CancelAsync();
-            await CleanupProcesses();
+        }
+    }
+
+    [Fact]
+    public async Task TestLambdaWithNullEndpoint()
+    {
+        var testProjectDir = Path.GetFullPath("../../../../../testapps");
+        var config = new TestConfig
+        {
+            TestToolPath = Path.GetFullPath(Path.Combine(testProjectDir, "../src/Amazon.Lambda.TestTool")),
+            LambdaPath = Path.GetFullPath(Path.Combine(testProjectDir, "LambdaTestFunctionV2")),
+            FunctionName = "LambdaTestFunctionV2",
+            RouteName = "testfunction",
+            HttpMethod = "Post"
+        };
+
+        var cancellationTokenSource = new CancellationTokenSource();
+
+        try
+        {
+            StartTestToolProcessWithNullEndpoint(ApiGatewayEmulatorMode.HttpV2, Constants.DefaultApiGatewayEmulatorPort, config, cancellationTokenSource);
+            await WaitForGatewayHealthCheck(Constants.DefaultApiGatewayEmulatorPort);
+            await StartLambdaProcess(config, Constants.DefaultLambdaEmulatorPort);
+
+            var response = await TestEndpoint(config, Constants.DefaultApiGatewayEmulatorPort );
+            var responseContent = await response.Content.ReadAsStringAsync();
+
+            Assert.Equal(HttpStatusCode.OK, response.StatusCode);
+            Assert.Equal("HELLO WORLD", responseContent);
+        }
+        finally
+        {
+            await cancellationTokenSource.CancelAsync();
         }
     }
 
@@ -234,10 +263,28 @@ private async Task<HttpResponseMessage> TestEndpoint(TestConfig config, int apiG
         };
     }
 
+    private void StartTestToolProcessWithNullEndpoint(ApiGatewayEmulatorMode apiGatewayMode, int apiGatewayPort, TestConfig config, CancellationTokenSource cancellationTokenSource)
+    {
+        SetEnvironmentVariable("ASPNETCORE_ENVIRONMENT", "Development");
+        SetEnvironmentVariable("APIGATEWAY_EMULATOR_ROUTE_CONFIG", $@"{{
+        ""LambdaResourceName"": ""{config.RouteName}"",
+        ""HttpMethod"": ""{config.HttpMethod}"",
+        ""Path"": ""/{config.RouteName}""
+    }}");
+
+        cancellationTokenSource.CancelAfter(5000);
+        var settings = new RunCommandSettings { NoLaunchWindow = true, ApiGatewayEmulatorMode = apiGatewayMode, ApiGatewayEmulatorPort = apiGatewayPort};
+
+        var command = new RunCommand(_mockInteractiveService.Object);
+        var context = new CommandContext(new List<string>(), _mockRemainingArgs.Object, "run", null);
+
+        _ = command.ExecuteAsync(context, settings, cancellationTokenSource);
+    }
+
     private void StartTestToolProcess(ApiGatewayEmulatorMode apiGatewayMode, TestConfig config, int lambdaPort, int apiGatewayPort, CancellationTokenSource cancellationTokenSource)
     {
-        Environment.SetEnvironmentVariable("ASPNETCORE_ENVIRONMENT", "Development");
-        Environment.SetEnvironmentVariable("APIGATEWAY_EMULATOR_ROUTE_CONFIG", $@"{{
+        SetEnvironmentVariable("ASPNETCORE_ENVIRONMENT", "Development");
+        SetEnvironmentVariable("APIGATEWAY_EMULATOR_ROUTE_CONFIG", $@"{{
             ""LambdaResourceName"": ""{config.RouteName}"",
             ""Endpoint"": ""http://localhost:{lambdaPort}"",
             ""HttpMethod"": ""{config.HttpMethod}"",
@@ -362,6 +409,24 @@ private void LogMessage(string message)
         _testOutputHelper.WriteLine(message);
     }
 
+    private void SetEnvironmentVariable(string name, string value)
+    {
+        Environment.SetEnvironmentVariable(name, value);
+        if (!_setEnvironmentVariables.Contains(name))
+        {
+            _setEnvironmentVariables.Add(name);
+        }
+    }
+
+    private void CleanupEnvironmentVariables()
+    {
+        foreach (var variable in _setEnvironmentVariables)
+        {
+            Environment.SetEnvironmentVariable(variable, null);
+        }
+        _setEnvironmentVariables.Clear();
+    }
+
     private async Task CleanupProcesses()
     {
         var processes = new[] { _lambdaProcess };
@@ -386,5 +451,6 @@ private async Task CleanupProcesses()
     public async ValueTask DisposeAsync()
     {
         await CleanupProcesses();
+        CleanupEnvironmentVariables();
     }
 }