|
18 | 18 | import static org.assertj.core.api.Assertions.assertThat; |
19 | 19 |
|
20 | 20 | import io.awspring.cloud.sqs.listener.SqsHeaders; |
| 21 | +import java.util.Collection; |
| 22 | +import java.util.List; |
21 | 23 | import org.junit.jupiter.api.Test; |
22 | 24 | import org.springframework.messaging.Message; |
23 | 25 | import org.springframework.messaging.support.MessageBuilder; |
24 | 26 |
|
25 | | -import java.util.Collection; |
26 | | -import java.util.List; |
27 | | - |
28 | 27 | /** |
29 | 28 | * Tests for {@link MessageHeaderUtils}. |
30 | 29 | * |
@@ -100,52 +99,49 @@ void shouldPreserveOtherHeaders() { |
100 | 99 | } |
101 | 100 |
|
102 | 101 | @Test |
103 | | - void shouldReturnAwsMessageIdWhenHeaderPresent() { |
| 102 | + void shouldReturnRawMessageIdWhenHeaderPresent() { |
104 | 103 | // given |
105 | | - String awsMessageId = "92898073-7bd6a160-5797b060-54a7e539"; |
| 104 | + String rawMessageId = "92898073-7bd6a160-5797b060-54a7e539"; |
106 | 105 | Message<String> message = MessageBuilder.withPayload("test-payload") |
107 | | - .setHeader(SqsHeaders.SQS_AWS_MESSAGE_ID_HEADER, awsMessageId) |
108 | | - .build(); |
| 106 | + .setHeader(SqsHeaders.SQS_RAW_MESSAGE_ID_HEADER, rawMessageId).build(); |
109 | 107 |
|
110 | 108 | // when |
111 | | - String result = MessageHeaderUtils.getAwsMessageId(message); |
| 109 | + String result = MessageHeaderUtils.getRawMessageId(message); |
112 | 110 |
|
113 | 111 | // then |
114 | | - assertThat(result).isEqualTo(awsMessageId); |
| 112 | + assertThat(result).isEqualTo(rawMessageId); |
115 | 113 | } |
116 | 114 |
|
117 | 115 | @Test |
118 | | - void shouldFallbackToSpringMessageIdWhenAwsHeaderNotPresent() { |
| 116 | + void shouldFallbackToSpringMessageIdWhenRawHeaderNotPresent() { |
119 | 117 | // given |
120 | 118 | Message<String> message = MessageBuilder.withPayload("test-payload").build(); |
121 | 119 | String expectedId = message.getHeaders().getId().toString(); |
122 | 120 |
|
123 | 121 | // when |
124 | | - String result = MessageHeaderUtils.getAwsMessageId(message); |
| 122 | + String result = MessageHeaderUtils.getRawMessageId(message); |
125 | 123 |
|
126 | 124 | // then |
127 | 125 | assertThat(result).isEqualTo(expectedId); |
128 | 126 | } |
129 | 127 |
|
130 | 128 | @Test |
131 | | - void shouldConcatenateAwsMessageIdsFromCollection() { |
| 129 | + void shouldConcatenateRawMessageIdsFromCollection() { |
132 | 130 | // given |
133 | | - String awsMessageId1 = "aws-id-1"; |
134 | | - String awsMessageId2 = "aws-id-2"; |
| 131 | + String rawMessageId1 = "raw-id-1"; |
| 132 | + String rawMessageId2 = "raw-id-2"; |
135 | 133 |
|
136 | 134 | Message<String> message1 = MessageBuilder.withPayload("payload1") |
137 | | - .setHeader(SqsHeaders.SQS_AWS_MESSAGE_ID_HEADER, awsMessageId1) |
138 | | - .build(); |
| 135 | + .setHeader(SqsHeaders.SQS_RAW_MESSAGE_ID_HEADER, rawMessageId1).build(); |
139 | 136 | Message<String> message2 = MessageBuilder.withPayload("payload2") |
140 | | - .setHeader(SqsHeaders.SQS_AWS_MESSAGE_ID_HEADER, awsMessageId2) |
141 | | - .build(); |
| 137 | + .setHeader(SqsHeaders.SQS_RAW_MESSAGE_ID_HEADER, rawMessageId2).build(); |
142 | 138 |
|
143 | 139 | Collection<Message<String>> messages = List.of(message1, message2); |
144 | 140 |
|
145 | 141 | // when |
146 | | - String result = MessageHeaderUtils.getAwsMessageId(messages); |
| 142 | + String result = MessageHeaderUtils.getRawMessageId(messages); |
147 | 143 |
|
148 | 144 | // then |
149 | | - assertThat(result).isEqualTo("aws-id-1; aws-id-2"); |
| 145 | + assertThat(result).isEqualTo("raw-id-1; raw-id-2"); |
150 | 146 | } |
151 | 147 | } |
0 commit comments