[camel] 03/06: CAMEL-15374 - Added Spring test for getFederationToken

[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 5bcd48ceef5ceb8c83253054fff400932121d108
Author: Andrea Cosentino <anco...@gmail.com>
AuthorDate: Thu Aug 27 08:41:17 2020 +0200

    CAMEL-15374 - Added Spring test for getFederationToken
---
 .../component/aws2/sts/STS2ProducerSpringTest.java      | 17 +++++++++++++++++
 .../aws2/sts/STSComponentSpringTest-context.xml         |  5 +++++
 2 files changed, 22 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 53d303a..c5b688e 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.GetFederationTokenResponse;
 import software.amazon.awssdk.services.sts.model.GetSessionTokenResponse;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -68,6 +69,22 @@ public class STS2ProducerSpringTest extends 
CamelSpringTestSupport {
         GetSessionTokenResponse resultGet = (GetSessionTokenResponse) 
exchange.getIn().getBody();
         assertEquals("xxx", resultGet.credentials().accessKeyId());
     }
+    
+    public void stsGetFederationTokenTest() throws Exception {
+
+        mock.expectedMessageCount(1);
+        Exchange exchange = template.request("direct:getFederationToken", new 
Processor() {
+            @Override
+            public void process(Exchange exchange) throws Exception {
+                exchange.getIn().setHeader(STS2Constants.OPERATION, 
STS2Operations.getFederationToken);
+            }
+        });
+
+        assertMockEndpointsSatisfied();
+
+        GetFederationTokenResponse resultGet = (GetFederationTokenResponse) 
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 5414028..9b0e2d9 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
@@ -34,6 +34,11 @@
             <to 
uri="aws2-sts://test?stsClient=#amazonStsClient&amp;operation=getSessionToken"/>
             <to uri="mock:result"/>
         </route>
+        <route>
+            <from uri="direct:getFederationToken"/>
+            <to 
uri="aws2-sts://test?stsClient=#amazonStsClient&amp;operation=getFederationToken"/>
+            <to uri="mock:result"/>
+        </route>
     </camelContext>
 
     <bean id="amazonStsClient" 
class="org.apache.camel.component.aws2.sts.AmazonSTSClientMock"/>