@Test public void testRemove() { Message<Integer> message1 = MessageBuilder.withPayload(1) .setHeader("foo", "bar").build(); Message<Integer> message2 = MessageBuilder.fromMessage(message1) .removeHeader("foo") .build(); assertFalse(message2.getHeaders().containsKey("foo")); }
@SuppressWarnings("unchecked") private <T> Message<O> wrapOutputToMessage(T value, Message<I> originalMessage) { Message<O> returnMessage = (Message<O>) MessageBuilder.withPayload(value).copyHeaders(originalMessage.getHeaders()) .removeHeader(MessageHeaders.CONTENT_TYPE).build(); return returnMessage; }