signal-desktop/ts/test-electron/sql/storyDistribution_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

235 lines
7.3 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 { generateStoryDistributionId } from '../../types/StoryDistributionId';
import type { StoryDistributionWithMembersType } from '../../sql/Interface';
const {
_deleteAllStoryDistributions,
_getAllStoryDistributionMembers,
_getAllStoryDistributions,
createNewStoryDistribution,
deleteStoryDistribution,
getAllStoryDistributionsWithMembers,
modifyStoryDistribution,
modifyStoryDistributionMembers,
modifyStoryDistributionWithMembers,
} = dataInterface;
describe('sql/storyDistribution', () => {
beforeEach(async () => {
await _deleteAllStoryDistributions();
});
it('roundtrips with create/fetch/delete', async () => {
const list: StoryDistributionWithMembersType = {
id: generateStoryDistributionId(),
name: 'My Story',
allowsReplies: true,
isBlockList: false,
members: [generateAci(), generateAci()],
senderKeyInfo: {
createdAtDate: Date.now(),
distributionId: generateUuid(),
memberDevices: [],
},
storageID: generateUuid(),
storageVersion: 1,
storageNeedsSync: false,
storageUnknownFields: undefined,
deletedAtTimestamp: undefined,
};
await createNewStoryDistribution(list);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
const allHydratedLists = await getAllStoryDistributionsWithMembers();
assert.lengthOf(allHydratedLists, 1);
assert.deepEqual(allHydratedLists[0], list);
await deleteStoryDistribution(list.id);
assert.lengthOf(await _getAllStoryDistributions(), 0);
assert.lengthOf(await _getAllStoryDistributionMembers(), 0);
assert.lengthOf(await getAllStoryDistributionsWithMembers(), 0);
});
it('updates core fields with modifyStoryDistribution', async () => {
const SERVICE_ID_1 = generateAci();
const SERVICE_ID_2 = generateAci();
const list: StoryDistributionWithMembersType = {
id: generateStoryDistributionId(),
name: 'My Story',
allowsReplies: true,
isBlockList: false,
members: [SERVICE_ID_1, SERVICE_ID_2],
senderKeyInfo: {
createdAtDate: Date.now(),
distributionId: generateUuid(),
memberDevices: [],
},
storageID: generateUuid(),
storageVersion: 1,
storageNeedsSync: false,
storageUnknownFields: undefined,
deletedAtTimestamp: undefined,
};
await createNewStoryDistribution(list);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
const updated = {
...list,
name: 'Updated story',
senderKeyInfo: {
createdAtDate: Date.now() + 10,
distributionId: generateUuid(),
memberDevices: [
{
id: 1,
identifier: SERVICE_ID_1,
registrationId: 232,
},
],
},
};
await modifyStoryDistribution(updated);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
const allHydratedLists = await getAllStoryDistributionsWithMembers();
assert.lengthOf(allHydratedLists, 1);
assert.deepEqual(allHydratedLists[0], updated);
});
it('adds and removes with modifyStoryDistributionMembers', async () => {
const SERVICE_ID_1 = generateAci();
const SERVICE_ID_2 = generateAci();
const SERVICE_ID_3 = generateAci();
const SERVICE_ID_4 = generateAci();
const list: StoryDistributionWithMembersType = {
id: generateStoryDistributionId(),
name: 'My Story',
allowsReplies: true,
isBlockList: false,
members: [SERVICE_ID_1, SERVICE_ID_2],
senderKeyInfo: {
createdAtDate: Date.now(),
distributionId: generateUuid(),
memberDevices: [],
},
storageID: generateUuid(),
storageVersion: 1,
storageNeedsSync: false,
storageUnknownFields: undefined,
deletedAtTimestamp: undefined,
};
await createNewStoryDistribution(list);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
await modifyStoryDistributionMembers(list.id, {
toAdd: [SERVICE_ID_3, SERVICE_ID_4],
toRemove: [SERVICE_ID_1],
});
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 3);
const allHydratedLists = await getAllStoryDistributionsWithMembers();
assert.lengthOf(allHydratedLists, 1);
assert.deepEqual(allHydratedLists[0], {
...list,
members: [SERVICE_ID_2, SERVICE_ID_3, SERVICE_ID_4],
});
});
it('adds and removes with modifyStoryDistributionWithMembers', async () => {
const SERVICE_ID_1 = generateAci();
const SERVICE_ID_2 = generateAci();
const SERVICE_ID_3 = generateAci();
const SERVICE_ID_4 = generateAci();
const list: StoryDistributionWithMembersType = {
id: generateStoryDistributionId(),
name: 'My Story',
allowsReplies: true,
isBlockList: false,
members: [SERVICE_ID_1, SERVICE_ID_2],
senderKeyInfo: {
createdAtDate: Date.now(),
distributionId: generateUuid(),
memberDevices: [],
},
storageID: generateUuid(),
storageVersion: 1,
storageNeedsSync: false,
storageUnknownFields: undefined,
deletedAtTimestamp: undefined,
};
await createNewStoryDistribution(list);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
await modifyStoryDistributionWithMembers(list, {
toAdd: [SERVICE_ID_3, SERVICE_ID_4],
toRemove: [SERVICE_ID_1],
});
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 3);
const allHydratedLists = await getAllStoryDistributionsWithMembers();
assert.lengthOf(allHydratedLists, 1);
assert.deepEqual(allHydratedLists[0], {
...list,
members: [SERVICE_ID_2, SERVICE_ID_3, SERVICE_ID_4],
});
});
it('eliminates duplicates without complaint in createNewStoryDistribution', async () => {
const SERVICE_ID_1 = generateAci();
const SERVICE_ID_2 = generateAci();
const list: StoryDistributionWithMembersType = {
id: generateStoryDistributionId(),
name: 'My Story',
allowsReplies: true,
isBlockList: false,
members: [SERVICE_ID_1, SERVICE_ID_1, SERVICE_ID_2],
senderKeyInfo: {
createdAtDate: Date.now(),
distributionId: generateUuid(),
memberDevices: [],
},
storageID: generateUuid(),
storageVersion: 1,
storageNeedsSync: false,
storageUnknownFields: undefined,
deletedAtTimestamp: undefined,
};
await createNewStoryDistribution(list);
assert.lengthOf(await _getAllStoryDistributions(), 1);
assert.lengthOf(await _getAllStoryDistributionMembers(), 2);
const allHydratedLists = await getAllStoryDistributionsWithMembers();
assert.lengthOf(allHydratedLists, 1);
assert.deepEqual(allHydratedLists[0].members, [SERVICE_ID_1, SERVICE_ID_2]);
});
});