Add mock test for safety number dialog

2024-09-09 15:42:33 -07:00 · 2024-09-09 15:42:33 -07:00 · 4f9d383180
commit 4f9d383180
parent ff5b21d2f3
1 changed files with 198 additions and 0 deletions
--- a/ts/test-mock/messaging/safety_number_test.ts
+++ b/ts/test-mock/messaging/safety_number_test.ts
@ -0,0 +1,198 @@
+// Copyright 2024 Signal Messenger, LLC
+// SPDX-License-Identifier: AGPL-3.0-only
+
+import createDebug from 'debug';
+import { StorageState, Proto } from '@signalapp/mock-server';
+import { assert } from 'chai';
+
+import type { App } from '../playwright';
+import { Bootstrap } from '../bootstrap';
+import { MINUTE } from '../../util/durations';
+import { uuidToBytes } from '../../util/uuidToBytes';
+import { MY_STORY_ID } from '../../types/Stories';
+import {
+  clickOnConversation,
+  typeIntoInput,
+  expectSystemMessages,
+} from '../helpers';
+
+export const debug = createDebug('mock:test:safetyNumber');
+
+const IdentifierType = Proto.ManifestRecord.Identifier.Type;
+
+describe('safety number', function (this: Mocha.Suite) {
+  let bootstrap: Bootstrap;
+  let app: App;
+
+  this.timeout(MINUTE);
+  beforeEach(async () => {
+    bootstrap = new Bootstrap();
+    await bootstrap.init();
+
+    const { phone, contacts } = bootstrap;
+    const [alice] = contacts;
+    let state = StorageState.getEmpty();
+
+    state = state.updateAccount({
+      profileKey: phone.profileKey.serialize(),
+      e164: phone.device.number,
+      givenName: phone.profileName,
+      hasSetMyStoriesPrivacy: true,
+    });
+
+    state = state.addContact(alice, {
+      identityKey: alice.publicKey.serialize(),
+      profileKey: alice.profileKey.serialize(),
+    });
+    state = state.pin(alice);
+
+    // Create a story distribution list
+    state = state.addRecord({
+      type: IdentifierType.STORY_DISTRIBUTION_LIST,
+      record: {
+        storyDistributionList: {
+          allowsReplies: true,
+          identifier: uuidToBytes(MY_STORY_ID),
+          isBlockList: false,
+          name: MY_STORY_ID,
+          recipientServiceIds: [alice.device.aci],
+        },
+      },
+    });
+
+    await phone.setStorageState(state);
+
+    app = await bootstrap.link();
+  });
+
+  afterEach(async function (this: Mocha.Context) {
+    if (!bootstrap) {
+      return;
+    }
+
+    await bootstrap.maybeSaveLogs(this.currentTest, app);
+    await app.close();
+    await bootstrap.teardown();
+  });
+
+  async function changeIdentityKey(): Promise<void> {
+    const {
+      phone,
+      contacts: [alice, bob],
+    } = bootstrap;
+
+    await app.waitForStorageService();
+
+    debug('change public key in storage service');
+    let state = await phone.expectStorageState('after link');
+
+    // Break identity key
+    state = state.updateContact(alice, {
+      identityKey: bob.publicKey.serialize(),
+    });
+
+    await phone.setStorageState(state);
+    await phone.sendFetchStorage({
+      timestamp: bootstrap.getTimestamp(),
+    });
+
+    await app.waitForStorageService();
+  }
+
+  it('show safety number change UI on regular send', async () => {
+    const {
+      contacts: [alice],
+    } = bootstrap;
+
+    const window = await app.getWindow();
+
+    await clickOnConversation(window, alice);
+
+    const input = await app.waitForEnabledComposer();
+    await typeIntoInput(input, 'Hello Alice!');
+
+    await changeIdentityKey();
+
+    await expectSystemMessages(window, [
+      /Safety Number has changed/, // Bob's key from storage service
+    ]);
+
+    debug('Sending message');
+    await input.press('Enter');
+
+    debug('Waiting for safety number dialog');
+    const dialog = window.locator(
+      '[data-testid="ConfirmationDialog.SafetyNumberChangeDialog.reviewing"]'
+    );
+    await dialog.locator(`"${alice.profileName}"`).waitFor();
+
+    await expectSystemMessages(window, [
+      /Safety Number has changed/, // Bob's key from storage service
+      /Safety Number has changed/, // Fixed Alice's key from backend
+    ]);
+
+    debug('Confirming send');
+    await dialog.getByRole('button', { name: 'Send anyway' }).click();
+
+    debug('Getting a message');
+    const { body } = await alice.waitForMessage();
+    assert.strictEqual(body, 'Hello Alice!');
+  });
+
+  it('show safety number change UI on story send', async () => {
+    const {
+      contacts: [alice],
+    } = bootstrap;
+    const window = await app.getWindow();
+
+    const storiesPane = window.locator('.Stories');
+    const storiesCreator = window.locator('.StoryCreator');
+
+    await window.getByTestId('NavTabsItem--Stories').click();
+
+    await storiesPane
+      .locator('button.Stories__pane__add-story__button')
+      .click();
+    await storiesPane
+      .locator(
+        '.ContextMenu__popper .Stories__pane__add-story__option--title ' +
+          '>> "Text story"'
+      )
+      .click();
+
+    debug('Focusing textarea');
+    // Note: For some reason `.click()` doesn't work here anymore.
+    await storiesCreator.locator('.TextAttachment').dispatchEvent('click');
+
+    debug('Entering text');
+    await storiesCreator.locator('.TextAttachment__text__textarea').fill('123');
+
+    debug('Clicking "Next"');
+    await storiesCreator
+      .locator('.StoryCreator__toolbar button >> "Next"')
+      .click();
+
+    debug('Selecting "My Story"');
+    await window
+      .locator('.SendStoryModal__distribution-list__name >> "My Story"')
+      .click();
+
+    await changeIdentityKey();
+
+    debug('Hitting Send');
+    await window.locator('button.SendStoryModal__send').click();
+
+    debug('Waiting for safety number dialog');
+    const dialog = window.locator(
+      '[data-testid="ConfirmationDialog.SafetyNumberChangeDialog.reviewing"]'
+    );
+    await dialog.locator(`"${alice.profileName}"`).waitFor();
+
+    debug('Confirming send');
+    await dialog.getByRole('button', { name: 'Send anyway' }).click();
+
+    debug('Getting a story');
+    const { storyMessage } = await alice.waitForStory();
+    assert.strictEqual(storyMessage.textAttachment?.text, '123');
+  });
+});