144 lines
4.2 KiB
TypeScript
144 lines
4.2 KiB
TypeScript
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import { assert } from 'chai';
|
|
|
|
import * as durations from '../../util/durations';
|
|
import type { App, Bootstrap } from './fixtures';
|
|
import { initStorage, debug } from './fixtures';
|
|
|
|
describe('storage service', function needsName() {
|
|
this.timeout(durations.MINUTE);
|
|
|
|
let bootstrap: Bootstrap;
|
|
let app: App;
|
|
|
|
beforeEach(async () => {
|
|
({ bootstrap, app } = await initStorage());
|
|
});
|
|
|
|
afterEach(async function after() {
|
|
if (!bootstrap) {
|
|
return;
|
|
}
|
|
|
|
if (this.currentTest?.state !== 'passed') {
|
|
await bootstrap.saveLogs();
|
|
}
|
|
|
|
await app.close();
|
|
await bootstrap.teardown();
|
|
});
|
|
|
|
it('should handle message request state changes', async () => {
|
|
const { phone, desktop, server } = bootstrap;
|
|
|
|
const initialState = await phone.expectStorageState('initial state');
|
|
|
|
debug('Creating stranger');
|
|
const stranger = await server.createPrimaryDevice({
|
|
profileName: 'Mysterious Stranger',
|
|
});
|
|
|
|
const ourKey = await desktop.popSingleUseKey();
|
|
await stranger.addSingleUseKey(desktop, ourKey);
|
|
|
|
debug('Sending a message from a stranger');
|
|
await stranger.sendText(desktop, 'Hello!', {
|
|
withProfileKey: true,
|
|
timestamp: bootstrap.getTimestamp(),
|
|
});
|
|
|
|
const window = await app.getWindow();
|
|
|
|
const leftPane = window.locator('.left-pane-wrapper');
|
|
const conversationStack = window.locator('.conversation-stack');
|
|
|
|
debug('Opening conversation with a stranger');
|
|
await leftPane
|
|
.locator(
|
|
'_react=ConversationListItem' +
|
|
`[title = ${JSON.stringify(stranger.profileName)}]`
|
|
)
|
|
.click();
|
|
|
|
debug("Verify that we stored stranger's profile key");
|
|
const postMessageState = await phone.waitForStorageState({
|
|
after: initialState,
|
|
});
|
|
{
|
|
assert.strictEqual(postMessageState.version, 2);
|
|
assert.isFalse(postMessageState.getContact(stranger)?.whitelisted);
|
|
assert.strictEqual(
|
|
postMessageState.getContact(stranger)?.profileKey?.length,
|
|
32
|
|
);
|
|
|
|
// ContactRecord
|
|
const { added, removed } = postMessageState.diff(initialState);
|
|
assert.strictEqual(added.length, 1, 'only one record must be added');
|
|
assert.strictEqual(removed.length, 0, 'no records should be removed');
|
|
}
|
|
|
|
debug('Accept conversation from a stranger');
|
|
await conversationStack
|
|
.locator('.module-message-request-actions button >> "Accept"')
|
|
.click();
|
|
|
|
debug('Verify that storage state was updated');
|
|
{
|
|
const nextState = await phone.waitForStorageState({
|
|
after: postMessageState,
|
|
});
|
|
assert.strictEqual(nextState.version, 3);
|
|
assert.isTrue(nextState.getContact(stranger)?.whitelisted);
|
|
|
|
// ContactRecord
|
|
const { added, removed } = nextState.diff(postMessageState);
|
|
assert.strictEqual(added.length, 1, 'only one record must be added');
|
|
assert.strictEqual(
|
|
removed.length,
|
|
1,
|
|
'only one record should be removed'
|
|
);
|
|
}
|
|
|
|
// Stranger should receive our profile key
|
|
{
|
|
const { body, source, dataMessage } = await stranger.waitForMessage();
|
|
assert.strictEqual(body, '', 'profile key message has no body');
|
|
assert.strictEqual(
|
|
source,
|
|
desktop,
|
|
'profile key message has valid source'
|
|
);
|
|
assert.isTrue(
|
|
phone.profileKey
|
|
.serialize()
|
|
.equals(dataMessage.profileKey ?? new Uint8Array(0)),
|
|
'profile key message has correct profile key'
|
|
);
|
|
}
|
|
|
|
debug('Enter message text');
|
|
const composeArea = window.locator(
|
|
'.composition-area-wrapper, ' +
|
|
'.ConversationView__template .react-wrapper'
|
|
);
|
|
const input = composeArea.locator('_react=CompositionInput');
|
|
|
|
await input.type('hello stranger!');
|
|
await input.press('Enter');
|
|
|
|
{
|
|
const { body, source } = await stranger.waitForMessage();
|
|
assert.strictEqual(body, 'hello stranger!', 'text message has body');
|
|
assert.strictEqual(source, desktop, 'text message has valid source');
|
|
}
|
|
|
|
debug('Verifying the final manifest version');
|
|
const finalState = await phone.expectStorageState('consistency check');
|
|
assert.strictEqual(finalState.version, 3);
|
|
});
|
|
});
|