Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use strategy for customizing ID/TIMESTAMP generation

Browse Source 
Issue: SPR-11468
This commit is contained in:
Rossen Stoyanchev
2014-04-14 21:03:13 -04:00
parent 95e25d4e6f
commit 4e1781ae8c
12 changed files with 374 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
spring-messaging/src/test/java/org/springframework/messaging/support/MessageHeaderAccessorTests.java
View File

@@ -201,15 +201,6 @@ public class MessageHeaderAccessorTests {
assertNull(accessor.getMessageHeaders().getTimestamp());
}
@Test
public void timestampBehaviorCopyFromExistingMessage() {
MessageHeaderAccessor accessor = new MessageHeaderAccessor();
accessor.setEnableTimestamp(true);
Message<?> message = MessageBuilder.createMessage("payload", accessor.getMessageHeaders());
MessageHeaderAccessor secondAccessor = new MessageHeaderAccessor(message);
assertNotNull(secondAccessor.getMessageHeaders().getTimestamp());
}
@Test
public void idGeneratorCustom() {
final UUID id = new UUID(0L, 23L);
@@ -229,19 +220,35 @@ public class MessageHeaderAccessorTests {
assertNotNull(accessor.getMessageHeaders().getId());
}
@Test
public void idGeneratorCopyFromExistingMessage() {
final UUID id = new UUID(0L, 23L);
public void idTimestampWithMutableHeaders() {
MessageHeaderAccessor accessor = new MessageHeaderAccessor();
accessor.setIdGenerator(new IdGenerator() {
@Override
public UUID generateId() {
return MessageHeaders.ID_VALUE_NONE;
}
});
accessor.setEnableTimestamp(false);
accessor.setLeaveMutable(true);
MessageHeaders headers = accessor.getMessageHeaders();
assertNull(headers.getId());
assertNull(headers.getTimestamp());
final UUID id = new UUID(0L, 23L);
accessor.setIdGenerator(new IdGenerator() {
@Override
public UUID generateId() {
return id;
}
});
Message<?> message = MessageBuilder.createMessage("payload", accessor.getMessageHeaders());
MessageHeaderAccessor secondAccessor = new MessageHeaderAccessor(message);
assertSame(id, secondAccessor.getMessageHeaders().getId());
accessor.setEnableTimestamp(true);
accessor.setImmutable();
assertSame(id, accessor.getMessageHeaders().getId());
assertNotNull(headers.getTimestamp());
}