Added removeHeader() method to MessageBuilder.

This commit is contained in:
Mark Fisher
2008-09-02 22:24:37 +00:00
parent 6f2f1640bd
commit 614ffaf825
2 changed files with 31 additions and 0 deletions

View File

@@ -17,6 +17,7 @@
package org.springframework.integration.message;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import java.util.Date;
@@ -116,4 +117,24 @@ public class MessageBuilderTests {
assertEquals(past, expiredMessage.getHeaders().getExpirationDate());
}
@Test
public void testRemove() {
Message<Integer> message1 = MessageBuilder.fromPayload(1)
.setHeader("foo", "bar").build();
Message<Integer> message2 = MessageBuilder.fromMessage(message1)
.removeHeader("foo")
.build();
assertFalse(message2.getHeaders().containsKey("foo"));
}
@Test
public void testSettingToNullRemoves() {
Message<Integer> message1 = MessageBuilder.fromPayload(1)
.setHeader("foo", "bar").build();
Message<Integer> message2 = MessageBuilder.fromMessage(message1)
.setHeader("foo", null)
.build();
assertFalse(message2.getHeaders().containsKey("foo"));
}
}