camel-aws2-sns: added SNS test with subscription to SQS

components/camel-aws/camel-aws2-sns/src/test/java/org/apache/camel/component/aws2/sns/integration/SnsTopicProducerWithSubscriptionIT.java

@@ -0,0 +1,134 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.camel.component.aws2.sns.integration;
+
+import java.util.List;
+
+import org.apache.camel.Exchange;
+import org.apache.camel.ExchangePattern;
+import org.apache.camel.Processor;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.aws2.sns.Sns2Constants;
+import org.apache.camel.test.infra.aws2.clients.AWSSDKClientUtils;
+import org.apache.camel.test.infra.common.SharedNameGenerator;
+import org.apache.camel.test.infra.common.TestEntityNameGenerator;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.RegisterExtension;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import software.amazon.awssdk.services.sqs.SqsClient;
+import software.amazon.awssdk.services.sqs.model.CreateQueueRequest;
+import software.amazon.awssdk.services.sqs.model.GetQueueUrlRequest;
+import software.amazon.awssdk.services.sqs.model.GetQueueUrlResponse;
+import software.amazon.awssdk.services.sqs.model.Message;
+import software.amazon.awssdk.services.sqs.model.QueueDoesNotExistException;
+import software.amazon.awssdk.services.sqs.model.ReceiveMessageRequest;
+import software.amazon.awssdk.services.sqs.model.ReceiveMessageResponse;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+public class SnsTopicProducerWithSubscriptionIT extends Aws2SNSBase {
+    @RegisterExtension
+    public static SharedNameGenerator sharedNameGenerator = new TestEntityNameGenerator();
+
+    private static final Logger LOG = LoggerFactory.getLogger(SnsTopicProducerWithSubscriptionIT.class);
+
+    private String sqsQueueUrl;
+    private SqsClient client;
+
+    public List<Message> receive() {
+        LOG.debug("Consuming messages from {}", sqsQueueUrl);
+        final int maxNumberOfMessages = 1;
+
+        int maxWaitTime = 10;
+        final ReceiveMessageRequest request = ReceiveMessageRequest.builder()
+                .queueUrl(sqsQueueUrl)
+                .waitTimeSeconds(maxWaitTime)
+                .maxNumberOfMessages(maxNumberOfMessages)
+                .build();
+
+        while (true) {
+            ReceiveMessageResponse response = client.receiveMessage(request);
+
+            if (!response.sdkHttpResponse().isSuccessful()) {
+                LOG.warn("Did not receive a success response from SQS: status code {}",
+                        response.sdkHttpResponse().statusCode());
+            }
+
+            List<Message> messages = response.messages();
+            for (Message message : messages) {
+                LOG.info("Received message: {}", message.body());
+            }
+
+            if (!messages.isEmpty()) {
+                return messages;
+            }
+        }
+    }
+
+    public void setUpQueue() {
+        client = AWSSDKClientUtils.newSQSClient();
+        String queue = sharedNameGenerator.getName() + "sqs";
+
+        GetQueueUrlRequest getQueueUrlRequest = GetQueueUrlRequest.builder()
+                .queueName(queue)
+                .build();
+
+        try {
+            GetQueueUrlResponse getQueueUrlResult = client.getQueueUrl(getQueueUrlRequest);
+
+            sqsQueueUrl = getQueueUrlResult.queueUrl();
+        } catch (QueueDoesNotExistException e) {
+            CreateQueueRequest createQueueRequest = CreateQueueRequest.builder()
+                    .queueName(queue)
+                    .build();
+
+            sqsQueueUrl = client.createQueue(createQueueRequest).queueUrl();
+        }
+    }
+
+    @Test
+    public void sendInOnly() throws Exception {
+        Exchange exchange = template.send("direct:start", ExchangePattern.InOnly, new Processor() {
+            public void process(Exchange exchange) throws Exception {
+                exchange.getIn().setHeader(Sns2Constants.SUBJECT, "This is my subject");
+                exchange.getIn().setBody("This is my message text.");
+            }
+        });
+
+        assertNotNull(exchange.getIn().getHeader(Sns2Constants.MESSAGE_ID));
+
+        List<Message> messages = receive();
+
+        assertEquals(1, messages.size(), "Did not receive as many messages as expected");
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        setUpQueue();
+
+        return new RouteBuilder() {
+            @Override
+            public void configure() throws Exception {
+                from("direct:start")
+                        .toF("aws2-sns://%s?subject=The+subject+message&autoCreateTopic=true&subscribeSNStoSQS=true&queueUrl=%s",
+                                sharedNameGenerator.getName(), sqsQueueUrl);
+            }
+        };
+    }
+}