Migrate schema to service ids
This commit is contained in:
Fedor Indutny
Jamie Kyle
parent
71958f8a01
commit
8b0da36caa
258 changed files with 4795 additions and 2613 deletions
559
ts/test-node/sql/migration_88_test.ts
Normal file
559
ts/test-node/sql/migration_88_test.ts
Normal file
|
|
@ -0,0 +1,559 @@
|
|||
// Copyright 2023 Signal Messenger, LLC
|
||||
// SPDX-License-Identifier: AGPL-3.0-only
|
||||
|
||||
import { assert } from 'chai';
|
||||
import type { Database } from '@signalapp/better-sqlite3';
|
||||
import SQL from '@signalapp/better-sqlite3';
|
||||
import { v4 as generateGuid } from 'uuid';
|
||||
|
||||
import { updateToVersion, insertData, getTableData } from './helpers';
|
||||
|
||||
const CONVO_ID = generateGuid();
|
||||
const GROUP_ID = generateGuid();
|
||||
const UUID = generateGuid();
|
||||
const PNI = generateGuid();
|
||||
const OUR_UUID = generateGuid();
|
||||
const OUR_PNI = generateGuid();
|
||||
const THEIR_UUID = generateGuid();
|
||||
|
||||
describe('SQL/updateToSchemaVersion88', () => {
|
||||
let db: Database;
|
||||
|
||||
beforeEach(() => {
|
||||
db = new SQL(':memory:');
|
||||
updateToVersion(db, 87);
|
||||
|
||||
insertData(db, 'items', [
|
||||
{
|
||||
id: 'uuid_id',
|
||||
json: {
|
||||
id: 'uuid_id',
|
||||
value: `${OUR_UUID}.1`,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: 'pni',
|
||||
json: {
|
||||
id: 'pni',
|
||||
value: OUR_PNI,
|
||||
},
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
db.close();
|
||||
});
|
||||
|
||||
it('should migrate conversations', () => {
|
||||
insertData(db, 'conversations', [
|
||||
{
|
||||
id: CONVO_ID,
|
||||
type: 'direct',
|
||||
uuid: UUID,
|
||||
json: {
|
||||
id: CONVO_ID,
|
||||
uuid: UUID,
|
||||
pni: PNI,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: GROUP_ID,
|
||||
type: 'group',
|
||||
json: {
|
||||
id: GROUP_ID,
|
||||
bannedMembersV2: [
|
||||
{
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
lastMessageBodyRanges: [
|
||||
{
|
||||
mentionUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
membersV2: [
|
||||
{
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
pendingAdminApprovalV2: [
|
||||
{
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
pendingMembersV2: [
|
||||
{
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
senderKeyInfo: {
|
||||
memberDevices: [{ identifier: CONVO_ID }],
|
||||
},
|
||||
},
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, 'conversations'), [
|
||||
{
|
||||
id: CONVO_ID,
|
||||
type: 'direct',
|
||||
serviceId: UUID,
|
||||
json: {
|
||||
id: CONVO_ID,
|
||||
serviceId: UUID,
|
||||
pni: `PNI:${PNI}`,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: GROUP_ID,
|
||||
type: 'group',
|
||||
json: {
|
||||
id: GROUP_ID,
|
||||
bannedMembersV2: [{ serviceId: THEIR_UUID }],
|
||||
lastMessageBodyRanges: [{ mentionAci: THEIR_UUID }],
|
||||
membersV2: [{ aci: THEIR_UUID }],
|
||||
pendingAdminApprovalV2: [{ aci: THEIR_UUID }],
|
||||
pendingMembersV2: [{ serviceId: THEIR_UUID }],
|
||||
senderKeyInfo: {
|
||||
memberDevices: [{ serviceId: UUID }],
|
||||
},
|
||||
},
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
it('should migrate items', () => {
|
||||
insertData(db, 'items', [
|
||||
{
|
||||
id: 'registrationIdMap',
|
||||
json: {
|
||||
id: 'registrationIdMap',
|
||||
value: {
|
||||
[OUR_UUID]: 123,
|
||||
[OUR_PNI]: 456,
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
id: 'identityKeyMap',
|
||||
json: {
|
||||
id: 'identityKeyMap',
|
||||
value: {
|
||||
[OUR_UUID]: {},
|
||||
[OUR_PNI]: {},
|
||||
},
|
||||
},
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, 'items'), [
|
||||
{
|
||||
id: 'uuid_id',
|
||||
json: {
|
||||
id: 'uuid_id',
|
||||
value: `${OUR_UUID}.1`,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: 'pni',
|
||||
json: {
|
||||
id: 'pni',
|
||||
value: `PNI:${OUR_PNI}`,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: 'registrationIdMap',
|
||||
json: {
|
||||
id: 'registrationIdMap',
|
||||
value: {
|
||||
[OUR_UUID]: 123,
|
||||
[`PNI:${OUR_PNI}`]: 456,
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
id: 'identityKeyMap',
|
||||
json: {
|
||||
id: 'identityKeyMap',
|
||||
value: {
|
||||
[OUR_UUID]: {},
|
||||
[`PNI:${OUR_PNI}`]: {},
|
||||
},
|
||||
},
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
it('should migrate sessions', () => {
|
||||
insertData(db, 'sessions', [
|
||||
{
|
||||
id: `${OUR_UUID}:${THEIR_UUID}.1`,
|
||||
ourUuid: OUR_UUID,
|
||||
uuid: THEIR_UUID,
|
||||
json: {
|
||||
id: `${OUR_UUID}:${THEIR_UUID}.1`,
|
||||
ourUuid: OUR_UUID,
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: `${OUR_PNI}:${THEIR_UUID}.1`,
|
||||
ourUuid: OUR_PNI,
|
||||
uuid: THEIR_UUID,
|
||||
json: {
|
||||
id: `${OUR_PNI}:${THEIR_UUID}.1`,
|
||||
ourUuid: OUR_PNI,
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, 'sessions'), [
|
||||
{
|
||||
id: `${OUR_UUID}:${THEIR_UUID}.1`,
|
||||
ourServiceId: OUR_UUID,
|
||||
serviceId: THEIR_UUID,
|
||||
json: {
|
||||
id: `${OUR_UUID}:${THEIR_UUID}.1`,
|
||||
ourServiceId: OUR_UUID,
|
||||
serviceId: THEIR_UUID,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: `PNI:${OUR_PNI}:${THEIR_UUID}.1`,
|
||||
ourServiceId: `PNI:${OUR_PNI}`,
|
||||
serviceId: THEIR_UUID,
|
||||
json: {
|
||||
id: `PNI:${OUR_PNI}:${THEIR_UUID}.1`,
|
||||
ourServiceId: `PNI:${OUR_PNI}`,
|
||||
serviceId: THEIR_UUID,
|
||||
},
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
it('should migrate messages', () => {
|
||||
insertData(db, 'messages', [
|
||||
{
|
||||
id: 'message-id',
|
||||
json: {
|
||||
id: 'message-id',
|
||||
bodyRanges: [{ mentionUuid: THEIR_UUID }],
|
||||
quote: {
|
||||
bodyRanges: [{ mentionUuid: THEIR_UUID }],
|
||||
},
|
||||
sourceUuid: THEIR_UUID,
|
||||
expirationTimerUpdate: {
|
||||
sourceUuid: THEIR_UUID,
|
||||
},
|
||||
reactions: [{ authorUuid: THEIR_UUID }],
|
||||
storyReaction: { authorUuid: THEIR_UUID },
|
||||
storyReplyContext: {
|
||||
authorUuid: THEIR_UUID,
|
||||
},
|
||||
editHistory: [
|
||||
{
|
||||
bodyRanges: [{ mentionUuid: THEIR_UUID }],
|
||||
quote: {
|
||||
bodyRanges: [{ mentionUuid: THEIR_UUID }],
|
||||
},
|
||||
},
|
||||
],
|
||||
groupV2Change: {
|
||||
from: 'abc',
|
||||
details: [
|
||||
{
|
||||
type: 'member-add',
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
{
|
||||
type: 'pending-add-one',
|
||||
uuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
},
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, 'messages'), [
|
||||
{
|
||||
id: 'message-id',
|
||||
json: {
|
||||
id: 'message-id',
|
||||
bodyRanges: [{ mentionAci: THEIR_UUID }],
|
||||
quote: {
|
||||
bodyRanges: [{ mentionAci: THEIR_UUID }],
|
||||
},
|
||||
sourceServiceId: THEIR_UUID,
|
||||
expirationTimerUpdate: {
|
||||
sourceServiceId: THEIR_UUID,
|
||||
},
|
||||
reactions: [{ authorAci: THEIR_UUID }],
|
||||
storyReaction: { authorAci: THEIR_UUID },
|
||||
storyReplyContext: {
|
||||
authorAci: THEIR_UUID,
|
||||
},
|
||||
editHistory: [
|
||||
{
|
||||
bodyRanges: [{ mentionAci: THEIR_UUID }],
|
||||
quote: {
|
||||
bodyRanges: [{ mentionAci: THEIR_UUID }],
|
||||
},
|
||||
},
|
||||
],
|
||||
groupV2Change: {
|
||||
from: 'abc',
|
||||
details: [
|
||||
{
|
||||
type: 'member-add',
|
||||
aci: THEIR_UUID,
|
||||
},
|
||||
{
|
||||
type: 'pending-add-one',
|
||||
serviceId: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
},
|
||||
mentionsMe: 0,
|
||||
rowid: 1,
|
||||
seenStatus: 0,
|
||||
shouldAffectActivity: 1,
|
||||
shouldAffectPreview: 1,
|
||||
isChangeCreatedByUs: 0,
|
||||
isGroupLeaveEvent: 0,
|
||||
isGroupLeaveEventFromOther: 0,
|
||||
isStory: 0,
|
||||
isTimerChangeFromSync: 0,
|
||||
isUserInitiatedMessage: 1,
|
||||
expiresAt: 9007199254740991,
|
||||
},
|
||||
]);
|
||||
});
|
||||
|
||||
for (const table of ['preKeys', 'signedPreKeys', 'kyberPreKeys']) {
|
||||
// eslint-disable-next-line no-loop-func
|
||||
it(`should migrate ${table}`, () => {
|
||||
insertData(db, table, [
|
||||
{
|
||||
id: `${OUR_UUID}:123`,
|
||||
json: {
|
||||
id: `${OUR_UUID}:123`,
|
||||
ourUuid: OUR_UUID,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: `${OUR_PNI}:456`,
|
||||
json: {
|
||||
id: `${OUR_PNI}:456`,
|
||||
ourUuid: OUR_PNI,
|
||||
},
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, table), [
|
||||
{
|
||||
id: `${OUR_UUID}:123`,
|
||||
json: {
|
||||
id: `${OUR_UUID}:123`,
|
||||
ourServiceId: OUR_UUID,
|
||||
},
|
||||
},
|
||||
{
|
||||
id: `PNI:${OUR_PNI}:456`,
|
||||
json: {
|
||||
id: `PNI:${OUR_PNI}:456`,
|
||||
ourServiceId: `PNI:${OUR_PNI}`,
|
||||
},
|
||||
},
|
||||
]);
|
||||
});
|
||||
}
|
||||
|
||||
it('should migrate jobs', () => {
|
||||
insertData(db, 'conversations', [
|
||||
{
|
||||
id: CONVO_ID,
|
||||
type: 'direct',
|
||||
uuid: UUID,
|
||||
json: {
|
||||
id: CONVO_ID,
|
||||
uuid: UUID,
|
||||
pni: PNI,
|
||||
},
|
||||
},
|
||||
]);
|
||||
|
||||
insertData(db, 'jobs', [
|
||||
{
|
||||
id: 'a',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'DeleteStoryForEveryone',
|
||||
updatedStoryRecipients: [
|
||||
{
|
||||
destinationUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'b',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'ResendRequest',
|
||||
senderUuid: THEIR_UUID,
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'c',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
receipts: [
|
||||
{
|
||||
senderUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'd',
|
||||
queueType: 'read sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
readSyncs: [
|
||||
{
|
||||
senderUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'e',
|
||||
queueType: 'view sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
viewSyncs: [
|
||||
{
|
||||
senderUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'f',
|
||||
queueType: 'view once open sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
viewOnceOpens: [
|
||||
{
|
||||
senderUuid: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'g',
|
||||
queueType: 'single proto',
|
||||
data: {
|
||||
identifier: CONVO_ID,
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
]);
|
||||
updateToVersion(db, 88);
|
||||
assert.deepStrictEqual(getTableData(db, 'jobs'), [
|
||||
{
|
||||
id: 'a',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'DeleteStoryForEveryone',
|
||||
updatedStoryRecipients: [
|
||||
{
|
||||
destinationServiceId: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'b',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'ResendRequest',
|
||||
senderAci: THEIR_UUID,
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'c',
|
||||
queueType: 'conversation',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
receipts: [
|
||||
{
|
||||
senderAci: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'd',
|
||||
queueType: 'read sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
readSyncs: [
|
||||
{
|
||||
senderAci: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'e',
|
||||
queueType: 'view sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
viewSyncs: [
|
||||
{
|
||||
senderAci: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'f',
|
||||
queueType: 'view once open sync',
|
||||
data: {
|
||||
type: 'Receipts',
|
||||
viewOnceOpens: [
|
||||
{
|
||||
senderAci: THEIR_UUID,
|
||||
},
|
||||
],
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
{
|
||||
id: 'g',
|
||||
queueType: 'single proto',
|
||||
data: {
|
||||
serviceId: UUID,
|
||||
},
|
||||
timestamp: 1,
|
||||
},
|
||||
]);
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue