Fix tests

Author: normj

Libraries/test/Amazon.Lambda.RuntimeSupport.Tests/Amazon.Lambda.RuntimeSupport.IntegrationTests/BaseCustomRuntimeTest.cs

@@ -15,6 +15,8 @@ namespace Amazon.Lambda.RuntimeSupport.IntegrationTests
 {
     public class BaseCustomRuntimeTest
     {
+        protected readonly Runtime _providedRuntime = Runtime.ProvidedAl2023;
+
         public const int FUNCTION_MEMORY_MB = 512;
 
         public static readonly RegionEndpoint TestRegion = RegionEndpoint.USWest2;
@@ -110,13 +112,13 @@ await iamClient.DetachRolePolicyAsync(new DetachRolePolicyRequest
         }
 
         public async Task<bool> PrepareTestResources(IAmazonS3 s3Client, IAmazonLambda lambdaClient,
-            AmazonIdentityManagementServiceClient iamClient)
+            AmazonIdentityManagementServiceClient iamClient, Runtime runtime)
         {
             var roleAlreadyExisted = await ValidateAndSetIamRoleArn(iamClient);
 
             var testBucketName = TestBucketRoot + Guid.NewGuid().ToString();
             await CreateBucketWithDeploymentZipAsync(s3Client, testBucketName);
-            await CreateFunctionAsync(lambdaClient, testBucketName);
+            await CreateFunctionAsync(lambdaClient, testBucketName, runtime);
 
             return roleAlreadyExisted;
         }
@@ -273,7 +275,7 @@ private async Task WaitForFunctionToBeReady(IAmazonLambda lambdaClient)
             await Task.Delay(1000);
         }
 
-        protected async Task CreateFunctionAsync(IAmazonLambda lambdaClient, string bucketName)
+        protected async Task CreateFunctionAsync(IAmazonLambda lambdaClient, string bucketName, Runtime runtime)
         {
             await DeleteFunctionIfExistsAsync(lambdaClient);
 
@@ -288,7 +290,7 @@ protected async Task CreateFunctionAsync(IAmazonLambda lambdaClient, string buck
                 Handler = Handler,
                 MemorySize = FUNCTION_MEMORY_MB,
                 Timeout = 30,
-                Runtime = Runtime.ProvidedAl2023,
+                Runtime = runtime,
                 Role = ExecutionRoleArn
             };
 

Libraries/test/Amazon.Lambda.RuntimeSupport.Tests/Amazon.Lambda.RuntimeSupport.IntegrationTests/CustomRuntimeAspNetCoreMinimalApiCustomSerializerTest.cs

@@ -39,7 +39,7 @@ public async Task TestMinimalApi()
 
                 try
                 {
-                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient);
+                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient, _providedRuntime);
                     await InvokeSuccessToWeatherForecastController(lambdaClient);
                 }
                 catch (NoDeploymentPackageFoundException)
@@ -69,7 +69,7 @@ public async Task TestThreadingLogging()
 
                 try
                 {
-                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient);
+                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient, _providedRuntime);
                     await InvokeLoggerTestController(lambdaClient);
                 }
                 catch (NoDeploymentPackageFoundException)

Libraries/test/Amazon.Lambda.RuntimeSupport.Tests/Amazon.Lambda.RuntimeSupport.IntegrationTests/CustomRuntimeAspNetCoreMinimalApiTest.cs

@@ -39,7 +39,7 @@ public async Task TestMinimalApi()
 
                 try
                 {
-                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient);
+                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient, _providedRuntime);
                     await InvokeSuccessToWeatherForecastController(lambdaClient);
                 }
                 catch (NoDeploymentPackageFoundException)
@@ -69,7 +69,7 @@ public async Task TestThreadingLogging()
 
                 try
                 {
-                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient);
+                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient, _providedRuntime);
                     await InvokeLoggerTestController(lambdaClient);
                 }
                 catch (NoDeploymentPackageFoundException)

Libraries/test/Amazon.Lambda.RuntimeSupport.Tests/Amazon.Lambda.RuntimeSupport.IntegrationTests/CustomRuntimeTests.cs

@@ -69,7 +69,7 @@ protected virtual async Task TestAllHandlersAsync()
 
                 try
                 {
-                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient);
+                    roleAlreadyExisted = await PrepareTestResources(s3Client, lambdaClient, iamClient, _providedRuntime);
 
                     await RunMaxHeapMemoryCheck(lambdaClient, "GetTotalAvailableMemoryBytes");
                     await RunWithoutMaxHeapMemoryCheck(lambdaClient, "GetTotalAvailableMemoryBytes");

Libraries/test/Amazon.Lambda.RuntimeSupport.Tests/Amazon.Lambda.RuntimeSupport.IntegrationTests/ResponseStreamingTests.cs

@@ -114,7 +114,7 @@ public async Task EnsureResourcesDeployedAsync(ResponseStreamingTests tests)
                 return;
 
             _tests = tests;
-            _roleAlreadyExisted = await _tests.PrepareTestResources(_s3Client, _lambdaClient, _iamClient);
+            _roleAlreadyExisted = await _tests.PrepareTestResources(_s3Client, _lambdaClient, _iamClient, Runtime.Dotnet10);
 
             _resourcesCreated = true;
         }