Multi-select forwarding and deleting

ts/test-electron/sql/getMessagesBetween_test.ts

@@ -0,0 +1,123 @@
+// Copyright 2023 Signal Messenger, LLC
+// SPDX-License-Identifier: AGPL-3.0-only
+
+import { assert } from 'chai';
+
+import dataInterface from '../../sql/Client';
+import { UUID } from '../../types/UUID';
+import type { UUIDStringType } from '../../types/UUID';
+
+import type { MessageAttributesType } from '../../model-types';
+
+const {
+  saveMessages,
+  _getAllMessages,
+  _removeAllMessages,
+  getMessagesBetween,
+} = dataInterface;
+
+function getUuid(): UUIDStringType {
+  return UUID.generate().toString();
+}
+
+describe('sql/getMessagesBetween', () => {
+  beforeEach(async () => {
+    await _removeAllMessages();
+  });
+
+  it('finds all messages between two in-order messages', async () => {
+    assert.lengthOf(await _getAllMessages(), 0);
+
+    const now = Date.now();
+    const conversationId = getUuid();
+    const ourUuid = getUuid();
+
+    function getMessage(body: string, offset: number): MessageAttributesType {
+      return {
+        id: getUuid(),
+        body,
+        type: 'outgoing',
+        conversationId,
+        sent_at: now + offset,
+        received_at: now + offset,
+        timestamp: now + offset,
+      };
+    }
+
+    const message1 = getMessage('message 1', -50);
+    const message2 = getMessage('message 2', -40); // after
+    const message3 = getMessage('message 3', -30);
+    const message4 = getMessage('message 4', -20); // before
+    const message5 = getMessage('message 5', -10);
+
+    await saveMessages([message1, message2, message3, message4, message5], {
+      forceSave: true,
+      ourUuid,
+    });
+
+    assert.lengthOf(await _getAllMessages(), 5);
+
+    const ids = await getMessagesBetween(conversationId, {
+      after: {
+        received_at: message2.received_at,
+        sent_at: message2.sent_at,
+      },
+      before: {
+        received_at: message4.received_at,
+        sent_at: message4.sent_at,
+      },
+      includeStoryReplies: false,
+    });
+
+    assert.lengthOf(ids, 1);
+    assert.deepEqual(ids, [message3.id]);
+  });
+
+  it('returns based on timestamps even if one message doesnt exist', async () => {
+    assert.lengthOf(await _getAllMessages(), 0);
+
+    const now = Date.now();
+    const conversationId = getUuid();
+    const ourUuid = getUuid();
+
+    function getMessage(body: string, offset: number): MessageAttributesType {
+      return {
+        id: getUuid(),
+        body,
+        type: 'outgoing',
+        conversationId,
+        sent_at: now + offset,
+        received_at: now + offset,
+        timestamp: now + offset,
+      };
+    }
+
+    const message1 = getMessage('message 1', -50);
+    const message2 = getMessage('message 2', -40); // after
+    const message3 = getMessage('message 3', -30);
+    const message4 = getMessage('message 4', -20); // before, doesn't exist
+    const message5 = getMessage('message 5', -10);
+
+    await saveMessages([message1, message2, message3, message5], {
+      forceSave: true,
+      ourUuid,
+    });
+
+    assert.lengthOf(await _getAllMessages(), 4);
+
+    const ids = await getMessagesBetween(conversationId, {
+      after: {
+        received_at: message2.received_at,
+        sent_at: message2.sent_at,
+      },
+      before: {
+        received_at: message4.received_at,
+        sent_at: message4.sent_at,
+      },
+      includeStoryReplies: false,
+    });
+
+    assert.lengthOf(ids, 1);
+    assert.deepEqual(ids, [message3.id]);
+  });
+});