add test for CloudWatch PutMetric via v2 SDK client (#29)

Author: pinzon

pom.xml

@@ -134,6 +134,12 @@
         </dependency>
 
         <!-- AWS SDK version 2 libs -->
+        <dependency>
+            <groupId>software.amazon.awssdk</groupId>
+            <artifactId>cloudwatch</artifactId>
+            <version>${aws.sdkv2.version}</version>
+            <scope>provided</scope>
+        </dependency>
         <dependency>
             <groupId>software.amazon.awssdk</groupId>
             <artifactId>sns</artifactId>

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

@@ -3,6 +3,7 @@
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.utils.*;
 import software.amazon.awssdk.http.*;
+import software.amazon.awssdk.services.cloudwatch.*;
 import software.amazon.awssdk.services.kinesis.*;
 import software.amazon.awssdk.services.sns.*;
 import software.amazon.awssdk.services.sqs.*;
@@ -47,6 +48,10 @@ public static SecretsManagerAsyncClient getClientSecretsManagerAsyncV2() {
     public static S3AsyncClient getClientS3AsyncV2() {
         return wrapApiClientV2(S3AsyncClient.builder(), Localstack.INSTANCE.getEndpointS3()).build();
     }
+    
+    public static CloudWatchAsyncClient getClientCloudWatchAsyncV2() {
+        return wrapApiClientV2(CloudWatchAsyncClient.builder(), Localstack.INSTANCE.getEndpointCloudWatch()).build();
+    }
 
     public static <T extends SdkAsyncClientBuilder> T wrapApiClientV2(T builder, String endpointURL) {
         try {

src/test/java/cloud/localstack/SQSMessagingTest.java

@@ -25,6 +25,10 @@
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.TimeUnit;
 
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+
 /**
  * Test integration of SQS/JMS messaging with LocalStack
  * Based on: https://bitbucket.org/atlassian/localstack/issues/24/not-support-sqs-in-jms
@@ -193,9 +197,10 @@ public void testGetQueueRedrivePolicy() {
         GetQueueAttributesResult dlQueueAttrs = clientSQS.getQueueAttributes(
                 new GetQueueAttributesRequest(dlQueueUrl).withAttributeNames("QueueArn")
         );
-        String dlQueueArn = dlQueueAttrs.getAttributes().get("QueueArn");
 
-        final String redrivePolicy = "{\"maxReceiveCount\": 5, \"deadLetterTargetArn\": \"" + dlQueueArn + "\"}";
+        String dlQueueArn = dlQueueAttrs.getAttributes().get("QueueArn");
+        int maxReceiveCount = 5;
+        final String redrivePolicy = "{\"maxReceiveCount\":"+maxReceiveCount+", \"deadLetterTargetArn\": \"" + dlQueueArn + "\"}";
 
         SetQueueAttributesRequest request = new SetQueueAttributesRequest()
                 .withQueueUrl(srcQueueUrl)
@@ -210,6 +215,17 @@ public void testGetQueueRedrivePolicy() {
 
         result = clientSQS.getQueueAttributes(new GetQueueAttributesRequest(srcQueueUrl).withAttributeNames("RedrivePolicy"));
         Assert.assertEquals(result.getAttributes().size(), 1);
-        Assert.assertEquals(result.getAttributes().get("RedrivePolicy"), redrivePolicy);
+
+        String resultPolicy = result.getAttributes().get("RedrivePolicy");
+        ObjectMapper mapper = new ObjectMapper();
+        HashMap<String, Object> map = new HashMap<String, Object>();
+
+        try {
+            map = mapper.readValue(resultPolicy, new TypeReference<Map<String, Object>>(){});
+            Assert.assertEquals( map.get("maxReceiveCount"), maxReceiveCount);
+            Assert.assertEquals( map.get("deadLetterTargetArn"), dlQueueArn);
+        } catch (Exception e) {
+            throw new RuntimeException("No RedrivePolicy found");
+        }
     }
 }

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

@@ -4,6 +4,8 @@
 import cloud.localstack.LocalstackTestRunner;
 
 import software.amazon.awssdk.core.SdkSystemSetting;
+import software.amazon.awssdk.services.cloudwatch.*;
+import software.amazon.awssdk.services.cloudwatch.model.*;
 import software.amazon.awssdk.services.kinesis.*;
 import software.amazon.awssdk.services.kinesis.model.*;
 import software.amazon.awssdk.services.s3.*;
@@ -25,6 +27,11 @@
 
 import java.util.*;
 import java.nio.ByteBuffer;
+import java.time.ZonedDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.ZoneOffset;
+import java.time.Instant;
+
 import software.amazon.awssdk.core.SdkBytes;
 import java.util.concurrent.CompletableFuture;
 
@@ -139,4 +146,32 @@ public void testGetSecretAsParam() throws Exception {
         Assert.assertNotNull(parameterValue);
         Assert.assertEquals("secretcontent", parameterValue);
     }
+    @Test
+    public void testCWPutMetrics() throws Exception {
+        final CloudWatchAsyncClient clientCW = TestUtils.getClientCloudWatchAsyncV2();
+        Dimension dimension = Dimension.builder()
+                    .name("UNIQUE_PAGES")
+                    .value("URLS")
+                    .build();
+
+            // Set an Instant object
+        String time = ZonedDateTime.now( ZoneOffset.UTC ).format( DateTimeFormatter.ISO_INSTANT );
+        Instant instant = Instant.parse(time);
+
+        double dataPoint = 1.23423;
+
+        MetricDatum datum = MetricDatum.builder()
+            .metricName("PAGES_VISITED")
+            .unit(StandardUnit.NONE)
+            .value(dataPoint)
+            .timestamp(instant)
+            .dimensions(dimension).build();
+
+        PutMetricDataRequest request = PutMetricDataRequest.builder()
+             .namespace("SITE/TRAFFIC")
+             .metricData(datum).build();
+
+        PutMetricDataResponse response = clientCW.putMetricData(request).get();
+        Assert.assertNotNull(response);
+    }
 }