MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a (failing!) check for destination UUIDs

This commit is contained in:
Jon Chambers 2021-11-30 17:12:36 -05:00 committed by Jon Chambers
parent 674bf1b0e0
commit 394f58f6cc
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
service/src/test/java/org/whispersystems/textsecuregcm/tests/storage/MessagesDynamoDbTest.java
View File

@ -38,6 +38,7 @@ class MessagesDynamoDbTest {
builder.setContent(ByteString.copyFrom(new byte[]{(byte) 0xDE, (byte) 0xAD, (byte) 0xBE, (byte) 0xEF})); builder.setContent(ByteString.copyFrom(new byte[]{(byte) 0xDE, (byte) 0xAD, (byte) 0xBE, (byte) 0xEF}));
builder.setServerGuid(UUID.randomUUID().toString()); builder.setServerGuid(UUID.randomUUID().toString());
builder.setServerTimestamp(serverTimestamp); builder.setServerTimestamp(serverTimestamp);
builder.setDestinationUuid(UUID.randomUUID().toString());
MESSAGE1 = builder.build(); MESSAGE1 = builder.build();
@ -48,6 +49,7 @@ class MessagesDynamoDbTest {
builder.setContent(ByteString.copyFromUtf8("MOO")); builder.setContent(ByteString.copyFromUtf8("MOO"));
builder.setServerGuid(UUID.randomUUID().toString()); builder.setServerGuid(UUID.randomUUID().toString());
builder.setServerTimestamp(serverTimestamp + 1); builder.setServerTimestamp(serverTimestamp + 1);
builder.setDestinationUuid(UUID.randomUUID().toString());
MESSAGE2 = builder.build(); MESSAGE2 = builder.build();
@ -58,6 +60,7 @@ class MessagesDynamoDbTest {
builder.setContent(ByteString.copyFromUtf8("COW")); builder.setContent(ByteString.copyFromUtf8("COW"));
builder.setServerGuid(UUID.randomUUID().toString()); builder.setServerGuid(UUID.randomUUID().toString());
builder.setServerTimestamp(serverTimestamp); // Test same millisecond arrival for two different messages builder.setServerTimestamp(serverTimestamp); // Test same millisecond arrival for two different messages
builder.setDestinationUuid(UUID.randomUUID().toString());
MESSAGE3 = builder.build(); MESSAGE3 = builder.build();
} }
@ -175,6 +178,7 @@ class MessagesDynamoDbTest {
assertThat(retrieved.getMessage()).isEqualTo(inserted.hasLegacyMessage() ? inserted.getLegacyMessage().toByteArray() : null); assertThat(retrieved.getMessage()).isEqualTo(inserted.hasLegacyMessage() ? inserted.getLegacyMessage().toByteArray() : null);
assertThat(retrieved.getServerTimestamp()).isEqualTo(inserted.getServerTimestamp()); assertThat(retrieved.getServerTimestamp()).isEqualTo(inserted.getServerTimestamp());
assertThat(retrieved.getGuid()).isEqualTo(UUID.fromString(inserted.getServerGuid())); assertThat(retrieved.getGuid()).isEqualTo(UUID.fromString(inserted.getServerGuid()));
assertThat(retrieved.getDestinationUuid()).isEqualTo(UUID.fromString(inserted.getDestinationUuid()));
} }
private static VerifyMessage verify(MessageProtos.Envelope expected) { private static VerifyMessage verify(MessageProtos.Envelope expected) {