mirrors/signal-desktop
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

Use untagged pnis in storage service

Browse source
This commit is contained in:
Fedor Indutny 2023-09-28 01:14:55 +02:00 • committed by GitHub
parent 283ef57779
commit eb7942dd1e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 530 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

175
ts/test-node/sql/migration_960_test.ts Normal file
View file

@ -0,0 +1,175 @@
// 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 OUR_ACI = generateGuid();
const OUR_UNPREFIXED_PNI = generateGuid();
const OUR_PREFIXED_PNI = `PNI:${OUR_UNPREFIXED_PNI}`;
describe('SQL/updateToSchemaVersion960', () => {
let db: Database;
beforeEach(() => {
db = new SQL(':memory:');
updateToVersion(db, 950);
insertData(db, 'items', [
{
id: 'uuid_id',
json: {
id: 'uuid_id',
value: `${OUR_ACI}.1`,
},
},
{
id: 'pni',
json: {
id: 'pni',
value: OUR_UNPREFIXED_PNI,
},
},
]);
});
afterEach(() => {
db.close();
});
it('should migrate our conversation', () => {
insertData(db, 'conversations', [
{
id: CONVO_ID,
type: 'direct',
serviceId: OUR_ACI,
json: {
id: CONVO_ID,
serviceId: OUR_ACI,
pni: OUR_UNPREFIXED_PNI,
},
},
]);
updateToVersion(db, 960);
assert.deepStrictEqual(getTableData(db, 'conversations'), [
{
id: CONVO_ID,
type: 'direct',
serviceId: OUR_ACI,
json: {
id: CONVO_ID,
serviceId: OUR_ACI,
pni: OUR_PREFIXED_PNI,
},
},
]);
});
it('should migrate items', () => {
insertData(db, 'items', [
{
id: 'registrationIdMap',
json: {
id: 'registrationIdMap',
value: {
[OUR_ACI]: 123,
[OUR_UNPREFIXED_PNI]: 456,
},
},
},
{
id: 'identityKeyMap',
json: {
id: 'identityKeyMap',
value: {
[OUR_ACI]: {},
[OUR_UNPREFIXED_PNI]: {},
},
},
},
]);
updateToVersion(db, 960);
assert.deepStrictEqual(getTableData(db, 'items'), [
{
id: 'uuid_id',
json: {
id: 'uuid_id',
value: `${OUR_ACI}.1`,
},
},
{
id: 'pni',
json: {
id: 'pni',
value: OUR_PREFIXED_PNI,
},
},
{
id: 'registrationIdMap',
json: {
id: 'registrationIdMap',
value: {
[OUR_ACI]: 123,
[OUR_PREFIXED_PNI]: 456,
},
},
},
{
id: 'identityKeyMap',
json: {
id: 'identityKeyMap',
value: {
[OUR_ACI]: {},
[OUR_PREFIXED_PNI]: {},
},
},
},
]);
});
for (const table of ['preKeys', 'signedPreKeys', 'kyberPreKeys']) {
// eslint-disable-next-line no-loop-func
it(`should migrate ${table}`, () => {
insertData(db, table, [
{
id: `${OUR_ACI}:123`,
json: {
id: `${OUR_ACI}:123`,
ourServiceId: OUR_ACI,
},
},
{
id: `${OUR_UNPREFIXED_PNI}:456`,
json: {
id: `${OUR_UNPREFIXED_PNI}:456`,
ourServiceId: OUR_UNPREFIXED_PNI,
},
},
]);
updateToVersion(db, 960);
assert.deepStrictEqual(getTableData(db, table), [
{
id: `${OUR_ACI}:123`,
json: {
id: `${OUR_ACI}:123`,
ourServiceId: OUR_ACI,
},
ourServiceId: OUR_ACI,
},
{
id: `${OUR_PREFIXED_PNI}:456`,
json: {
id: `${OUR_PREFIXED_PNI}:456`,
ourServiceId: OUR_PREFIXED_PNI,
},
ourServiceId: OUR_PREFIXED_PNI,
},
]);
});
}
});