AWS SDK 2 upgrade: Fix bug with wrong message header conversion - spr…

…ing-atticgh-267
tinexw · Feb 17, 2020 · 430ae5c · 430ae5c
1 parent caee338
commit 430ae5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/...ssaging/src/main/java/org/springframework/cloud/aws/messaging/core/QueueMessageUtils.java b/...ssaging/src/main/java/org/springframework/cloud/aws/messaging/core/QueueMessageUtils.java
@@ -71,7 +71,7 @@ private static Map<String, Object> getAttributesAsMessageHeaders(
 		Map<String, Object> messageHeaders = new HashMap<>();
 		for (Map.Entry<MessageSystemAttributeName, String> attributeKeyValuePair : message
 				.attributes().entrySet()) {
-			messageHeaders.put(attributeKeyValuePair.getKey().name(),
+			messageHeaders.put(attributeKeyValuePair.getKey().toString(),
 					attributeKeyValuePair.getValue());
 		}
 

diff --git a/.../org/springframework/cloud/aws/messaging/listener/SimpleMessageListenerContainerTest.java b/.../org/springframework/cloud/aws/messaging/listener/SimpleMessageListenerContainerTest.java
@@ -451,8 +451,7 @@ protected void executeMessage(
 		container.stop();
 
 		verify(messageHandler).handleMessage(this.stringMessageCaptor.capture());
-		//TODO SDK2 migration: why is this now uppercase? used to be SenderId
-		assertThat(this.stringMessageCaptor.getValue().getHeaders().get("SENDER_ID"))
+		assertThat(this.stringMessageCaptor.getValue().getHeaders().get("SenderId"))
 				.isEqualTo("ID");
 
 	}