test setting read flag for shard messages

This commit is contained in:
ameba23
2021-03-22 08:42:25 +01:00
parent 0440c5c7c8
commit 8478097a3c
2 changed files with 9 additions and 2 deletions

View File

@@ -303,7 +303,8 @@ class SocialBackupManagerImpl extends ConversationClientImpl
SHARD.getValue()) { SHARD.getValue()) {
boolean isLocal = meta.getBoolean(MSG_KEY_LOCAL); boolean isLocal = meta.getBoolean(MSG_KEY_LOCAL);
MessageStatus status = db.getMessageStatus(txn, contactId, messageEntry.getKey()); MessageStatus status = db.getMessageStatus(txn, contactId,
messageEntry.getKey());
long timestamp; long timestamp;
if (isLocal) { if (isLocal) {
timestamp = meta.getLong(MSG_KEY_TIMESTAMP); timestamp = meta.getLong(MSG_KEY_TIMESTAMP);
@@ -316,7 +317,8 @@ class SocialBackupManagerImpl extends ConversationClientImpl
new ArrayList<>(); new ArrayList<>();
ShardMessageHeader shardHeader = new ShardMessageHeader( ShardMessageHeader shardHeader = new ShardMessageHeader(
messageEntry.getKey(), contactGroupId, timestamp, messageEntry.getKey(), contactGroupId, timestamp,
isLocal, false, status.isSent(), status.isSeen(), attachmentHeaders); isLocal, false, status.isSent(), status.isSeen(),
attachmentHeaders);
headers.add(shardHeader); headers.add(shardHeader);
} }
} }

View File

@@ -138,6 +138,11 @@ public class SocialBackupIntegrationTest
assertGroupCount(messageTracker0, g2From0.getId(), 1, 0); assertGroupCount(messageTracker0, g2From0.getId(), 1, 0);
assertGroupCount(messageTracker1, g0From1.getId(), 1, 1); assertGroupCount(messageTracker1, g0From1.getId(), 1, 1);
assertGroupCount(messageTracker2, g0From2.getId(), 1, 1); assertGroupCount(messageTracker2, g0From2.getId(), 1, 1);
// mark a message as read
socialBackupManager1.setReadFlag(g0From1.getId(),
messages0At1.iterator().next().getId(), true);
assertGroupCount(messageTracker1, g0From1.getId(), 1, 0);
} }
private Collection<ConversationMessageHeader> getMessages1At0() private Collection<ConversationMessageHeader> getMessages1At0()