signal-desktop/ts/test-electron/sql/getRecentStoryReplies_test.ts
Fedor Indutny 366b875fd2 Introduce Service Id Types
Co-authored-by: Scott Nonnenberg <scott@signal.org>
2023-08-21 09:30:32 -07:00

114 lines
3.2 KiB
TypeScript

// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import { assert } from 'chai';
import { v4 as generateUuid } from 'uuid';
import dataInterface from '../../sql/Client';
import { generateAci } from '../../types/ServiceId';
import type { MessageAttributesType } from '../../model-types.d';
const { _getAllMessages, getRecentStoryReplies, removeAll, saveMessages } =
dataInterface;
describe('sql/getRecentStoryReplies', () => {
beforeEach(async () => {
await removeAll();
});
it('returns message matching storyId in all converssations ', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId1 = generateUuid();
const conversationId2 = generateUuid();
const conversationId3 = generateUuid();
const ourAci = generateAci();
const storyId = generateUuid();
const message1: MessageAttributesType = {
id: generateUuid(),
body: 'message 1 - reply #1',
type: 'incoming',
conversationId: conversationId1,
sent_at: now - 20,
received_at: now - 20,
timestamp: now - 20,
storyId,
};
const message2: MessageAttributesType = {
id: generateUuid(),
body: 'message 2 - reply #2',
type: 'incoming',
conversationId: conversationId2,
sent_at: now - 10,
received_at: now - 10,
timestamp: now - 10,
storyId,
};
const message3: MessageAttributesType = {
id: generateUuid(),
body: 'message 3 - reply #3',
type: 'incoming',
conversationId: conversationId3,
sent_at: now,
received_at: now,
timestamp: now,
storyId,
};
const message4: MessageAttributesType = {
id: generateUuid(),
body: 'message 4 - the story itself',
type: 'story',
conversationId: conversationId3,
sent_at: now,
received_at: now,
timestamp: now,
storyId,
};
const message5: MessageAttributesType = {
id: generateUuid(),
body: 'message 5 - different story reply',
type: 'incoming',
conversationId: conversationId1,
sent_at: now,
received_at: now,
timestamp: now,
storyId: generateUuid(),
};
const message6: MessageAttributesType = {
id: generateUuid(),
body: 'message 6 - no story fields',
type: 'incoming',
conversationId: conversationId1,
sent_at: now,
received_at: now,
timestamp: now,
};
await saveMessages(
[message1, message2, message3, message4, message5, message6],
{
forceSave: true,
ourAci,
}
);
assert.lengthOf(await _getAllMessages(), 6);
const searchResultsPage1 = await getRecentStoryReplies(storyId, {
limit: 2,
});
assert.lengthOf(searchResultsPage1, 2, 'page 1');
assert.strictEqual(searchResultsPage1[0].body, message3.body);
assert.strictEqual(searchResultsPage1[1].body, message2.body);
const searchResultsPage2 = await getRecentStoryReplies(storyId, {
messageId: message2.id,
receivedAt: message2.received_at,
limit: 2,
});
assert.lengthOf(searchResultsPage2, 1, 'page 2');
assert.strictEqual(searchResultsPage2[0].body, message1.body);
});
});