[camel] 10/10: CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end user to pass an AWS Request pojo as body, aws2-kms added spring test too

[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 248b998e09326f0e99c98246db0cd3fdfe03e51a
Author: Andrea Cosentino <an...@gmail.com>
AuthorDate: Tue Apr 14 11:31:45 2020 +0200

    CAMEL-14868 - Camel-AWS2-*: Where possible, give the possiblity to the end user to pass an AWS Request pojo as body, aws2-kms added spring test too
---
 .../component/aws2/kms/KMSProducerSpringTest.java    | 20 ++++++++++++++++++++
 .../aws2/kms/KMSComponentSpringTest-context.xml      |  5 +++++
 2 files changed, 25 insertions(+)

diff --git a/components/camel-aws2-kms/src/test/java/org/apache/camel/component/aws2/kms/KMSProducerSpringTest.java b/components/camel-aws2-kms/src/test/java/org/apache/camel/component/aws2/kms/KMSProducerSpringTest.java
index 241d107..d2c895d 100644
--- a/components/camel-aws2-kms/src/test/java/org/apache/camel/component/aws2/kms/KMSProducerSpringTest.java
+++ b/components/camel-aws2-kms/src/test/java/org/apache/camel/component/aws2/kms/KMSProducerSpringTest.java
@@ -25,6 +25,7 @@ import org.junit.jupiter.api.Test;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 import software.amazon.awssdk.services.kms.model.CreateKeyResponse;
 import software.amazon.awssdk.services.kms.model.DescribeKeyResponse;
+import software.amazon.awssdk.services.kms.model.ListKeysRequest;
 import software.amazon.awssdk.services.kms.model.ListKeysResponse;
 import software.amazon.awssdk.services.kms.model.ScheduleKeyDeletionResponse;
 
@@ -141,6 +142,25 @@ public class KMSProducerSpringTest extends CamelSpringTestSupport {
         assertEquals("MyCamelKey", resultGet.keyMetadata().description());
         assertFalse(resultGet.keyMetadata().enabled());
     }
+    
+    @Test
+    public void kmsListKeysPojoTest() throws Exception {
+
+        mock.expectedMessageCount(1);
+        Exchange exchange = template.request("direct:listKeysPojo", new Processor() {
+            @Override
+            public void process(Exchange exchange) throws Exception {
+                exchange.getIn().setHeader(KMS2Constants.OPERATION, KMS2Operations.listKeys);
+                exchange.getIn().setBody(ListKeysRequest.builder().limit(10).build());
+            }
+        });
+
+        assertMockEndpointsSatisfied();
+
+        ListKeysResponse resultGet = (ListKeysResponse)exchange.getIn().getBody();
+        assertEquals(1, resultGet.keys().size());
+        assertEquals("keyId", resultGet.keys().get(0).keyId());
+    }
 
     @Override
     protected ClassPathXmlApplicationContext createApplicationContext() {
diff --git a/components/camel-aws2-kms/src/test/resources/org/apache/camel/component/aws2/kms/KMSComponentSpringTest-context.xml b/components/camel-aws2-kms/src/test/resources/org/apache/camel/component/aws2/kms/KMSComponentSpringTest-context.xml
index 4a8a0f7..1b65cb2 100644
--- a/components/camel-aws2-kms/src/test/resources/org/apache/camel/component/aws2/kms/KMSComponentSpringTest-context.xml
+++ b/components/camel-aws2-kms/src/test/resources/org/apache/camel/component/aws2/kms/KMSComponentSpringTest-context.xml
@@ -30,6 +30,11 @@
             <to uri="mock:result"/>
         </route>
         <route>
+            <from uri="direct:listKeysPojo"/>
+            <to uri="aws2-kms://Test?kmsClient=#amazonKmsClient&amp;operation=listKeys&amp;pojoRequest=true"/>
+            <to uri="mock:result"/>
+        </route>
+        <route>
             <from uri="direct:createKey"/>
             <to uri="aws2-kms://Test?kmsClient=#amazonKmsClient&amp;operation=createKey"/>
             <to uri="mock:result"/>