This is an automated email from the ASF dual-hosted git repository. cmueller pushed a commit to branch camel-2.19.x in repository https://gitbox.apache.org/repos/asf/camel.git
The following commit(s) were added to refs/heads/camel-2.19.x by this push: new ac04b60 CAMEL-12292: SnsProducer/SqsProducer setting MessageAttributes with empty values which causes errors ac04b60 is described below commit ac04b60541572f861ef68701779f880f003e2313 Author: Christian Mueller <c...@amazon.com> AuthorDate: Sat Mar 3 17:10:09 2018 +0100 CAMEL-12292: SnsProducer/SqsProducer setting MessageAttributes with empty values which causes errors --- .../camel/component/aws/sns/SnsProducer.java | 4 +- .../camel/component/aws/sqs/SqsProducer.java | 6 +- .../camel/component/aws/sns/SnsProducerTest.java | 67 ++++++++++++++++++++++ .../camel/component/aws/sqs/SqsProducerTest.java | 20 ++++++- 4 files changed, 91 insertions(+), 6 deletions(-) diff --git a/components/camel-aws/src/main/java/org/apache/camel/component/aws/sns/SnsProducer.java b/components/camel-aws/src/main/java/org/apache/camel/component/aws/sns/SnsProducer.java index 36c4a15..d87add7 100644 --- a/components/camel-aws/src/main/java/org/apache/camel/component/aws/sns/SnsProducer.java +++ b/components/camel-aws/src/main/java/org/apache/camel/component/aws/sns/SnsProducer.java @@ -87,14 +87,14 @@ public class SnsProducer extends DefaultProducer { return structure; } - private Map<String, MessageAttributeValue> translateAttributes(Map<String, Object> headers, Exchange exchange) { + Map<String, MessageAttributeValue> translateAttributes(Map<String, Object> headers, Exchange exchange) { Map<String, MessageAttributeValue> result = new HashMap<String, MessageAttributeValue>(); HeaderFilterStrategy headerFilterStrategy = getEndpoint().getHeaderFilterStrategy(); for (Entry<String, Object> entry : headers.entrySet()) { // only put the message header which is not filtered into the message attribute if (!headerFilterStrategy.applyFilterToCamelHeaders(entry.getKey(), entry.getValue(), exchange)) { Object value = entry.getValue(); - if (value instanceof String) { + if (value instanceof String && !((String)value).isEmpty()) { MessageAttributeValue mav = new MessageAttributeValue(); mav.setDataType("String"); mav.withStringValue((String)value); diff --git a/components/camel-aws/src/main/java/org/apache/camel/component/aws/sqs/SqsProducer.java b/components/camel-aws/src/main/java/org/apache/camel/component/aws/sqs/SqsProducer.java index 682d75e..6268ea8 100644 --- a/components/camel-aws/src/main/java/org/apache/camel/component/aws/sqs/SqsProducer.java +++ b/components/camel-aws/src/main/java/org/apache/camel/component/aws/sqs/SqsProducer.java @@ -102,15 +102,15 @@ public class SqsProducer extends DefaultProducer { } return sqsProducerToString; } - - private Map<String, MessageAttributeValue> translateAttributes(Map<String, Object> headers, Exchange exchange) { + + Map<String, MessageAttributeValue> translateAttributes(Map<String, Object> headers, Exchange exchange) { Map<String, MessageAttributeValue> result = new HashMap<String, MessageAttributeValue>(); HeaderFilterStrategy headerFilterStrategy = getEndpoint().getHeaderFilterStrategy(); for (Entry<String, Object> entry : headers.entrySet()) { // only put the message header which is not filtered into the message attribute if (!headerFilterStrategy.applyFilterToCamelHeaders(entry.getKey(), entry.getValue(), exchange)) { Object value = entry.getValue(); - if (value instanceof String) { + if (value instanceof String && !((String)value).isEmpty()) { MessageAttributeValue mav = new MessageAttributeValue(); mav.setDataType("String"); mav.withStringValue((String)value); diff --git a/components/camel-aws/src/test/java/org/apache/camel/component/aws/sns/SnsProducerTest.java b/components/camel-aws/src/test/java/org/apache/camel/component/aws/sns/SnsProducerTest.java new file mode 100644 index 0000000..38fffdc --- /dev/null +++ b/components/camel-aws/src/test/java/org/apache/camel/component/aws/sns/SnsProducerTest.java @@ -0,0 +1,67 @@ +/** + * 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.aws.sns; + +import java.util.HashMap; +import java.util.Map; + +import com.amazonaws.services.dynamodbv2.model.GetRecordsRequest; +import com.amazonaws.services.sns.model.MessageAttributeValue; +import org.apache.camel.Endpoint; +import org.apache.camel.Exchange; +import org.apache.camel.component.aws.firehose.KinesisFirehoseEndpoint; +import org.apache.camel.spi.HeaderFilterStrategy; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.runners.MockitoJUnitRunner; + +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertThat; +import static org.mockito.Mockito.when; + +@RunWith(MockitoJUnitRunner.class) +public class SnsProducerTest { + + @Mock + private Exchange exchange; + @Mock + private SnsEndpoint endpoint; + private SnsProducer producer; + + @Before + public void setUp() { + producer = new SnsProducer(endpoint); + + when(endpoint.getHeaderFilterStrategy()).thenReturn(new SnsHeaderFilterStrategy()); + } + + @Test + public void translateAttributes() { + Map<String, Object> headers = new HashMap<>(); + headers.put("key1", null); + headers.put("key2", ""); + headers.put("key3", "value3"); + + Map<String, MessageAttributeValue> translateAttributes = producer.translateAttributes(headers, exchange); + + assertThat(translateAttributes.size(), is(1)); + assertThat(translateAttributes.get("key3").getDataType(), is("String")); + assertThat(translateAttributes.get("key3").getStringValue(), is("value3")); + } +} \ No newline at end of file diff --git a/components/camel-aws/src/test/java/org/apache/camel/component/aws/sqs/SqsProducerTest.java b/components/camel-aws/src/test/java/org/apache/camel/component/aws/sqs/SqsProducerTest.java index bed350c..b7e0897 100644 --- a/components/camel-aws/src/test/java/org/apache/camel/component/aws/sqs/SqsProducerTest.java +++ b/components/camel-aws/src/test/java/org/apache/camel/component/aws/sqs/SqsProducerTest.java @@ -21,6 +21,7 @@ import java.util.HashMap; import java.util.Map; import com.amazonaws.services.sqs.AmazonSQSClient; +import com.amazonaws.services.sqs.model.MessageAttributeValue; import com.amazonaws.services.sqs.model.SendMessageRequest; import com.amazonaws.services.sqs.model.SendMessageResult; import org.apache.camel.Exchange; @@ -33,10 +34,13 @@ import org.junit.runner.RunWith; import org.mockito.ArgumentCaptor; import org.mockito.Mock; import org.mockito.runners.MockitoJUnitRunner; + +import static org.hamcrest.CoreMatchers.is; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; -import static org.mockito.Matchers.any; +import static org.junit.Assert.assertThat; import static org.mockito.Mockito.RETURNS_DEEP_STUBS; +import static org.mockito.Mockito.any; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; @@ -87,6 +91,20 @@ public class SqsProducerTest { } @Test + public void translateAttributes() { + Map<String, Object> headers = new HashMap<>(); + headers.put("key1", null); + headers.put("key2", ""); + headers.put("key3", "value3"); + + Map<String, MessageAttributeValue> translateAttributes = underTest.translateAttributes(headers, exchange); + + assertThat(translateAttributes.size(), is(1)); + assertThat(translateAttributes.get("key3").getDataType(), is("String")); + assertThat(translateAttributes.get("key3").getStringValue(), is("value3")); + } + + @Test public void itSendsTheBodyFromAnExchange() throws Exception { underTest.process(exchange); -- To stop receiving notification emails like this one, please contact cmuel...@apache.org.