[camel] 08/09: CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end user to pass an AWS Request pojo as body, aws2-msk add test

[ac...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

acosentino pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/camel.git

commit 59e75748774d7e176318dc1129c832c9a0833f18
Author: Andrea Cosentino <an...@gmail.com>
AuthorDate: Thu Apr 16 09:40:20 2020 +0200

    CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end user to pass an AWS Request pojo as body, aws2-msk add test
---
 .../camel/component/aws2/msk/MSKProducerTest.java    | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/components/camel-aws2-msk/src/test/java/org/apache/camel/component/aws2/msk/MSKProducerTest.java b/components/camel-aws2-msk/src/test/java/org/apache/camel/component/aws2/msk/MSKProducerTest.java
index 138bf18..cb6581c 100644
--- a/components/camel-aws2-msk/src/test/java/org/apache/camel/component/aws2/msk/MSKProducerTest.java
+++ b/components/camel-aws2-msk/src/test/java/org/apache/camel/component/aws2/msk/MSKProducerTest.java
@@ -29,6 +29,7 @@ import software.amazon.awssdk.services.kafka.model.ClusterState;
 import software.amazon.awssdk.services.kafka.model.CreateClusterResponse;
 import software.amazon.awssdk.services.kafka.model.DeleteClusterResponse;
 import software.amazon.awssdk.services.kafka.model.DescribeClusterResponse;
+import software.amazon.awssdk.services.kafka.model.ListClustersRequest;
 import software.amazon.awssdk.services.kafka.model.ListClustersResponse;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -57,6 +58,24 @@ public class MSKProducerTest extends CamelTestSupport {
         assertEquals(1, resultGet.clusterInfoList().size());
         assertEquals("test-kafka", resultGet.clusterInfoList().get(0).clusterName());
     }
+    
+    @Test
+    public void mskListKeysPojoTest() throws Exception {
+
+        mock.expectedMessageCount(1);
+        Exchange exchange = template.request("direct:listClustersPojo", new Processor() {
+            @Override
+            public void process(Exchange exchange) throws Exception {
+                exchange.getIn().setBody(ListClustersRequest.builder().maxResults(10).build());
+            }
+        });
+
+        assertMockEndpointsSatisfied();
+
+        ListClustersResponse resultGet = (ListClustersResponse)exchange.getIn().getBody();
+        assertEquals(1, resultGet.clusterInfoList().size());
+        assertEquals("test-kafka", resultGet.clusterInfoList().get(0).clusterName());
+    }
 
     @Test
     public void mskCreateClusterTest() throws Exception {
@@ -122,6 +141,7 @@ public class MSKProducerTest extends CamelTestSupport {
             @Override
             public void configure() throws Exception {
                 from("direct:listClusters").to("aws2-msk://test?mskClient=#amazonMskClient&operation=listClusters").to("mock:result");
+                from("direct:listClustersPojo").to("aws2-msk://test?mskClient=#amazonMskClient&operation=listClusters&pojoRequest=true").to("mock:result");
                 from("direct:createCluster").to("aws2-msk://test?mskClient=#amazonMskClient&operation=createCluster").to("mock:result");
                 from("direct:deleteCluster").to("aws2-msk://test?mskClient=#amazonMskClient&operation=deleteCluster").to("mock:result");
                 from("direct:describeCluster").to("aws2-msk://test?mskClient=#amazonMskClient&operation=describeCluster").to("mock:result");