add lambdaAsync utils for awssdkv2 (localstack#45)

Author: vitorfec

pom.xml

@@ -200,6 +200,12 @@
             <version>${aws.sdkv2.version}</version>
             <scope>provided</scope>
         </dependency>
+        <dependency>
+            <groupId>software.amazon.awssdk</groupId>
+            <artifactId>lambda</artifactId>
+            <version>${aws.sdkv2.version}</version>
+            <scope>provided</scope>
+        </dependency>
 
 
         <!-- test dependencies -->

src/main/java/cloud/localstack/awssdkv2/TestUtils.java

@@ -2,6 +2,7 @@
 
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.dynamodb.DynamoDbAsyncClient;
+import software.amazon.awssdk.services.lambda.LambdaAsyncClient;
 import software.amazon.awssdk.utils.*;
 import software.amazon.awssdk.http.*;
 import software.amazon.awssdk.services.cloudwatch.*;
@@ -58,6 +59,10 @@ public static CloudWatchAsyncClient getClientCloudWatchAsyncV2() {
         return wrapApiClientV2(CloudWatchAsyncClient.builder(), Localstack.INSTANCE.getEndpointCloudWatch()).build();
     }
 
+    public static LambdaAsyncClient getClientLambdaAsyncV2() {
+        return wrapApiClientV2(LambdaAsyncClient.builder(), Localstack.INSTANCE.getEndpointCloudWatch()).build();
+    }
+
     public static <T extends SdkAsyncClientBuilder> T wrapApiClientV2(T builder, String endpointURL) {
         try {
             return (T) ((AwsClientBuilder)builder

src/test/java/cloud/localstack/awssdkv2/BasicFeaturesSDKV2Test.java

@@ -2,16 +2,20 @@
 
 import cloud.localstack.Constants;
 import cloud.localstack.LocalstackTestRunner;
+import cloud.localstack.sample.LambdaHandler;
+import cloud.localstack.utils.LocalTestUtil;
+
+import lombok.val;
 
-import com.amazonaws.services.dynamodbv2.model.ListTablesResult;
-import org.assertj.core.api.Assertions;
 import software.amazon.awssdk.core.SdkSystemSetting;
 import software.amazon.awssdk.services.cloudwatch.*;
 import software.amazon.awssdk.services.cloudwatch.model.*;
 import software.amazon.awssdk.services.dynamodb.DynamoDbAsyncClient;
 import software.amazon.awssdk.services.dynamodb.model.*;
 import software.amazon.awssdk.services.kinesis.*;
 import software.amazon.awssdk.services.kinesis.model.*;
+import software.amazon.awssdk.services.lambda.model.CreateFunctionRequest;
+import software.amazon.awssdk.services.lambda.model.Runtime;
 import software.amazon.awssdk.services.s3.*;
 import software.amazon.awssdk.services.s3.model.*;
 import software.amazon.awssdk.services.secretsmanager.SecretsManagerAsyncClient;
@@ -214,7 +218,7 @@ public void testCWMultipleDimentionsAndMetrics() throws Exception {
                     .name("UNIQUE_PAGES"+i)
                     .value("URLS"+i)
                     .build());
-        };
+        }
 
         // Set an Instant object
         String time = ZonedDateTime.now( ZoneOffset.UTC ).format( DateTimeFormatter.ISO_INSTANT );
@@ -229,7 +233,7 @@ public void testCWMultipleDimentionsAndMetrics() throws Exception {
             .value(dataPoint)
             .timestamp(instant)
             .dimensions(awsDimensionList).build());
-        };
+        }
 
         PutMetricDataRequest request = PutMetricDataRequest.builder()
              .namespace("SITE/TRAFFIC")
@@ -238,4 +242,20 @@ public void testCWMultipleDimentionsAndMetrics() throws Exception {
         PutMetricDataResponse response = clientCW.putMetricData(request).get();
         Assert.assertNotNull(response);
     }
+
+    @Test
+    public void testLambdaCreateListFunctions() throws Exception {
+        val functionName = "test-f-"+UUID.randomUUID().toString();
+        val lambdaClient = TestUtils.getClientLambdaAsyncV2();
+        val createFunctionRequest = CreateFunctionRequest.builder().functionName(functionName)
+                .runtime(Runtime.JAVA8)
+                .role("r1")
+                .code(LocalTestUtil.createFunctionCodeSDKV2(LambdaHandler.class))
+                .handler(LambdaHandler.class.getName()).build();
+        val response = lambdaClient.createFunction(createFunctionRequest).get();
+        Assert.assertNotNull(response);
+        val functions = lambdaClient.listFunctions().get();
+        val function = functions.functions().stream().filter(f -> f.functionName().equals(functionName)).findFirst().get();
+        Assert.assertNotNull(function);
+    }
 }

src/test/java/cloud/localstack/utils/LocalTestUtil.java

@@ -10,10 +10,12 @@
 import java.util.zip.ZipOutputStream;
 
 import com.amazonaws.services.lambda.runtime.events.SQSEvent;
+import lombok.val;
 import org.apache.commons.io.IOUtils;
 
 import com.amazonaws.services.kinesis.model.Record;
 import com.amazonaws.services.lambda.model.FunctionCode;
+import software.amazon.awssdk.core.SdkBytes;
 
 /**
  * Utility methods used for the LocalStack unit and integration tests.
@@ -23,7 +25,18 @@
 public class LocalTestUtil {
 
 	public static FunctionCode createFunctionCode(Class<?> clazz) throws Exception {
-		FunctionCode code = new FunctionCode();
+		val code = new FunctionCode();
+		code.setZipFile(createFunctionByteBuffer(clazz));
+		return code;
+	}
+
+	public static software.amazon.awssdk.services.lambda.model.FunctionCode createFunctionCodeSDKV2(Class<?> clazz) throws Exception{
+		val codeBuilder = software.amazon.awssdk.services.lambda.model.FunctionCode.builder();
+		codeBuilder.zipFile(SdkBytes.fromByteBuffer(createFunctionByteBuffer(clazz)));
+		return codeBuilder.build();
+	}
+
+	private static ByteBuffer createFunctionByteBuffer(Class<?> clazz) throws Exception{
 		ByteArrayOutputStream zipOut = new ByteArrayOutputStream();
 		ByteArrayOutputStream jarOut = new ByteArrayOutputStream();
 		// create zip file
@@ -48,9 +61,7 @@ public static FunctionCode createFunctionCode(Class<?> clazz) throws Exception {
 		zipStream.closeEntry();
 
 		zipStream.close();
-		code.setZipFile(ByteBuffer.wrap(zipOut.toByteArray()));
-
-		return code;
+		return ByteBuffer.wrap(zipOut.toByteArray());
 	}
 
 	private static void addClassToJar(Class<?> clazz, JarOutputStream jarStream) throws IOException {