SpringCloud/spring-cloud-stream
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add test for EmbeddedHeaderUtils.headersToEmbed

Browse Source
This commit is contained in:
Ilayaperumal Gopinathan
2017-03-30 13:59:34 +05:30
parent 27184a4cfc
commit c812fa0374
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
spring-cloud-stream/src/test/java/org/springframework/cloud/stream/binder/MessageConverterTests.java
View File

@@ -29,7 +29,6 @@ import static org.assertj.core.api.Assertions.assertThat;
* @author Gary Russell
* @author Ilayaperumal Gopinathan
* @since 1.0
*
*/
public class MessageConverterTests {
@@ -48,6 +47,28 @@ public class MessageConverterTests {
assertThat(extracted.get("baz")).isEqualTo("quxx");
}
@Test
public void testConfigurableHeaders() throws Exception {
Message<byte[]> message = MessageBuilder.withPayload("Hello".getBytes()).setHeader("foo", "bar")
.setHeader("baz", "quxx").setHeader("contentType", "text/plain").build();
String[] headers = new String[]{"foo"};
byte[] embedded = EmbeddedHeaderUtils.embedHeaders(new MessageValues(message), EmbeddedHeaderUtils.headersToEmbed(headers));
assertThat(embedded[0] & 0xff).isEqualTo(0xff);
assertThat(new String(embedded).substring(1)).isEqualTo(
"\u0002\u000BcontentType\u0000\u0000\u0000\u000C\"text/plain\"\u0003foo\u0000\u0000\u0000\u0005\"bar\"Hello");
MessageValues extracted = EmbeddedHeaderUtils.extractHeaders(MessageBuilder.withPayload(embedded).build(), false);
assertThat(new String((byte[]) extracted.getPayload())).isEqualTo("Hello");
assertThat(extracted.get("foo")).isEqualTo("bar");
assertThat(extracted.get("baz")).isNull();
assertThat(extracted.get("contentType")).isEqualTo("text/plain");
assertThat(extracted.get("timestamp")).isNull();
MessageValues extractedWithRequestHeaders = EmbeddedHeaderUtils.extractHeaders(MessageBuilder.withPayload(embedded).build(), true);
assertThat(extractedWithRequestHeaders.get("foo")).isEqualTo("bar");
assertThat(extractedWithRequestHeaders.get("baz")).isNull();
assertThat(extractedWithRequestHeaders.get("contentType")).isEqualTo("text/plain");
assertThat(extractedWithRequestHeaders.get("timestamp")).isNotNull();
}
@Test
public void testHeaderExtractionWithDirectPayload() throws Exception {
Message<byte[]> message = MessageBuilder.withPayload("Hello".getBytes()).setHeader("foo", "bar")