[camel] 02/05: CAMEL-15374 - Camel-AWS2-STS: Add more operation to the producer, getSessionToken Spring Test

[acosentino]

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 96aa9b41f28df4b4a56c5fcc683a5f7a9c510e46
Author: Andrea Cosentino <anco...@gmail.com>
AuthorDate: Thu Aug 6 08:38:53 2020 +0200

    CAMEL-15374 - Camel-AWS2-STS: Add more operation to the producer, 
getSessionToken Spring Test
---
 .../component/aws2/sts/STS2ProducerSpringTest.java     | 18 ++++++++++++++++++
 .../aws2/sts/STSComponentSpringTest-context.xml        |  5 +++++
 2 files changed, 23 insertions(+)

diff --git 
a/components/camel-aws2-sts/src/test/java/org/apache/camel/component/aws2/sts/STS2ProducerSpringTest.java
 
b/components/camel-aws2-sts/src/test/java/org/apache/camel/component/aws2/sts/STS2ProducerSpringTest.java
index ab8f161..1cbc874 100644
--- 
a/components/camel-aws2-sts/src/test/java/org/apache/camel/component/aws2/sts/STS2ProducerSpringTest.java
+++ 
b/components/camel-aws2-sts/src/test/java/org/apache/camel/component/aws2/sts/STS2ProducerSpringTest.java
@@ -24,6 +24,7 @@ import 
org.apache.camel.test.spring.junit5.CamelSpringTestSupport;
 import org.junit.jupiter.api.Test;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 import software.amazon.awssdk.services.sts.model.AssumeRoleResponse;
+import software.amazon.awssdk.services.sts.model.GetSessionTokenResponse;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
@@ -50,6 +51,23 @@ public class STS2ProducerSpringTest extends 
CamelSpringTestSupport {
         AssumeRoleResponse resultGet = 
(AssumeRoleResponse)exchange.getIn().getBody();
         assertEquals("arn", resultGet.assumedRoleUser().arn());
     }
+    
+    @Test
+    public void stsGetSessionTokenTest() throws Exception {
+
+        mock.expectedMessageCount(1);
+        Exchange exchange = template.request("direct:getSessionToken", new 
Processor() {
+            @Override
+            public void process(Exchange exchange) throws Exception {
+                exchange.getIn().setHeader(STS2Constants.OPERATION, 
STS2Operations.getSessionToken);
+            }
+        });
+
+        assertMockEndpointsSatisfied();
+
+        GetSessionTokenResponse resultGet = 
(GetSessionTokenResponse)exchange.getIn().getBody();
+        assertEquals("xxx", resultGet.credentials().accessKeyId());
+    }
 
     @Override
     protected ClassPathXmlApplicationContext createApplicationContext() {
diff --git 
a/components/camel-aws2-sts/src/test/resources/org/apache/camel/component/aws2/sts/STSComponentSpringTest-context.xml
 
b/components/camel-aws2-sts/src/test/resources/org/apache/camel/component/aws2/sts/STSComponentSpringTest-context.xml
index e90b25f..5414028 100644
--- 
a/components/camel-aws2-sts/src/test/resources/org/apache/camel/component/aws2/sts/STSComponentSpringTest-context.xml
+++ 
b/components/camel-aws2-sts/src/test/resources/org/apache/camel/component/aws2/sts/STSComponentSpringTest-context.xml
@@ -29,6 +29,11 @@
             <to 
uri="aws2-sts://test?stsClient=#amazonStsClient&amp;operation=assumeRole"/>
             <to uri="mock:result"/>
         </route>
+        <route>
+            <from uri="direct:getSessionToken"/>
+            <to 
uri="aws2-sts://test?stsClient=#amazonStsClient&amp;operation=getSessionToken"/>
+            <to uri="mock:result"/>
+        </route>
     </camelContext>
 
     <bean id="amazonStsClient" 
class="org.apache.camel.component.aws2.sts.AmazonSTSClientMock"/>