signal-desktop/ts/test-node/sql/migrateConversationMessages_test.ts
2024-07-22 11:16:33 -07:00

109 lines
2.4 KiB
TypeScript

// Copyright 2024 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import { assert } from 'chai';
import type { WritableDB } from '../../sql/Interface';
import { migrateConversationMessages, setupTests } from '../../sql/Server';
import { createDB, insertData, getTableData } from './helpers';
describe('SQL/migrateConversationMessages', () => {
let db: WritableDB;
beforeEach(() => {
db = createDB();
setupTests(db);
});
afterEach(() => {
db.close();
});
function compactify(
message: Record<string, unknown>
): Record<string, unknown> {
const { id, conversationId, json } = message;
return {
id,
conversationId,
json,
};
}
it('should leave irrelevant messages intact', () => {
insertData(db, 'messages', [
{
id: 'irrelevant',
conversationId: 'other',
json: {
conversationId: 'other',
},
},
]);
migrateConversationMessages(db, 'obsolete', 'current');
assert.deepStrictEqual(getTableData(db, 'messages').map(compactify), [
{
id: 'irrelevant',
conversationId: 'other',
json: {
conversationId: 'other',
},
},
]);
});
it('should update conversationId and send state', () => {
insertData(db, 'messages', [
{
id: 'no-send-state',
conversationId: 'obsolete',
json: {
conversationId: 'obsolete',
body: 'test',
sendStateByConversationId: {
other: 'Failed',
obsolete: 'Read',
},
editHistory: [
{
body: 'test2',
sendStateByConversationId: {
other: 'Failed',
obsolete: 'Read',
},
},
],
},
},
]);
migrateConversationMessages(db, 'obsolete', 'current');
assert.deepStrictEqual(getTableData(db, 'messages').map(compactify), [
{
id: 'no-send-state',
conversationId: 'current',
json: {
body: 'test',
conversationId: 'current',
sendStateByConversationId: {
other: 'Failed',
current: 'Read',
},
editHistory: [
{
body: 'test2',
sendStateByConversationId: {
other: 'Failed',
current: 'Read',
},
},
],
},
},
]);
});
});