@dxos/react-ui 0.1.35 → 0.1.36-next.ef27157

package/src/playwright/invitations-manager.ts

@@ -3,28 +3,29 @@
 //
 
 import { expect } from 'chai';
-import type { Browser, ConsoleMessage, Page } from 'playwright';
+import type { Browser, ConsoleMessage } from 'playwright';
 import waitForExpect from 'wait-for-expect';
 
 import { Trigger } from '@dxos/async';
-import type { InvitationsOptions } from '@dxos/client';
-import { ConnectionState } from '@dxos/protocols/proto/dxos/client/services';
+import { ConnectionState, Invitation } from '@dxos/protocols/proto/dxos/client/services';
 import { setupPage } from '@dxos/test/playwright';
 
+import { ShellManager } from '../testing';
+
 // TODO(wittjosiah): Factor out.
 // TODO(burdon): No hard-coding of ports; reconcile all DXOS tools ports.
 const storybookUrl = (storyId: string) => `http://localhost:9009/iframe.html?id=${storyId}&viewMode=story`;
 
 export type PanelType = number | 'identity' | 'devices' | 'spaces' | 'join';
 
-export class InvitationsManager {
-  page!: Page;
-
+export class InvitationsManager extends ShellManager {
   private _initialized = false;
   private _invitationCode = new Trigger<string>();
-  private _authenticationCode = new Trigger<string>();
+  private _authCode = new Trigger<string>();
 
-  constructor(private readonly _browser: Browser) {}
+  constructor(private readonly _browser: Browser) {
+    super();
+  }
 
   async init() {
     if (this._initialized) {
@@ -45,24 +46,12 @@ export class InvitationsManager {
 
   async getDisplayName(id: number) {
     // TODO(wittjosiah): Update id.
-    return this._peer(id).getByTestId('identity-list-item').nth(0).textContent();
+    return this.peer(id).getByTestId('identity-list-item').nth(0).textContent();
   }
 
   async getSpaceName(id: number, nth: number) {
     // TODO(wittjosiah): Update id.
-    return this._peer(id).getByTestId('space-list-item').nth(nth).textContent();
-  }
-
-  authenticatorIsVisible(id: number, type: 'device' | 'space') {
-    // TODO(wittjosiah): Update id.
-    return this._peer(id)
-      .getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`)
-      .isVisible();
-  }
-
-  invitationFailed(id: number) {
-    // TODO(wittjosiah): Update id.
-    return this._peer(id).getByTestId('invitation-rescuer-reset').isVisible();
+    return this.peer(id).getByTestId('space-list-item').nth(nth).textContent();
   }
 
   // Actions
@@ -77,7 +66,7 @@ export class InvitationsManager {
   }
 
   async openPanel(id: number, panel: PanelType) {
-    const peer = this._peer(id);
+    const peer = this.peer(id);
 
     if (typeof panel === 'number') {
       // TODO(wittjosiah): Update id.
@@ -104,7 +93,7 @@ export class InvitationsManager {
   }
 
   async createIdentity(id: number) {
-    const createIdentity = this._peer(id).getByTestId('invitations.create-identity');
+    const createIdentity = this.peer(id).getByTestId('invitations.create-identity');
     // TODO(wittjosiah): Clicking on buttons wrapped in tooltips is flaky in webkit playwright.
     await createIdentity.click();
     await waitForExpect(async () => {
@@ -113,12 +102,12 @@ export class InvitationsManager {
   }
 
   async createSpace(id: number) {
-    await this._peer(id).getByTestId('invitations.create-space').click();
+    await this.peer(id).getByTestId('invitations.create-space').click();
   }
 
-  async createInvitation(id: number, type: 'device' | 'space', options?: InvitationsOptions): Promise<string> {
+  async createInvitation(id: number, type: 'device' | 'space', options?: Partial<Invitation>): Promise<string> {
     if (!options) {
-      const peer = this._peer(id);
+      const peer = this.peer(id);
       this._invitationCode = new Trigger<string>();
       await peer.getByTestId(`${type}s-panel.create-invitation`).click();
       return this._invitationCode.wait();
@@ -139,7 +128,7 @@ export class InvitationsManager {
   }
 
   async acceptInvitation(id: number, type: 'device' | 'space', invitation: string) {
-    const peer = this._peer(id);
+    const peer = this.peer(id);
     // TODO(wittjosiah): Update ids.
     if (type === 'device') {
       await peer.getByTestId('join-identity').click();
@@ -148,56 +137,12 @@ export class InvitationsManager {
     await this.page.keyboard.press('Enter');
   }
 
-  async cancelInvitation(id: number, type: 'device' | 'space', kind: 'host' | 'guest') {
-    const peer = this._peer(id);
-    if (kind === 'guest') {
-      await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-authenticator-cancel`).click();
-    }
-  }
-
-  async invitationInputContinue(id: number, type: 'device' | 'space') {
-    // TODO(wittjosiah): Update ids.
-    await this._peer(id)
-      .getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-input-continue`)
-      .click();
-  }
-
-  async getAuthenticationCode(): Promise<string> {
-    this._authenticationCode = new Trigger<string>();
-    return await this._authenticationCode.wait();
-  }
-
-  async authenticateInvitation(id: number, type: 'device' | 'space', authenticationCode: string) {
-    const peer = this._peer(id);
-    // TODO(wittjosiah): Update ids.
-    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).type(authenticationCode);
-    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-authenticator-next`).click();
-  }
-
-  async clearAuthenticationCode(id: number, type: 'device' | 'space') {
-    const peer = this._peer(id);
-    // TODO(wittjosiah): Update ids.
-    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).fill('');
-    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).focus();
-  }
-
-  async resetInvitation(id: number) {
-    const peer = this._peer(id);
-    // TODO(wittjosiah): Update ids.
-    await peer.getByTestId('invitation-rescuer-reset').click();
-  }
-
-  // TODO(wittjosiah): Remove.
-  async doneInvitation(id: number, type: 'device' | 'space') {
-    const peer = this._peer(id);
-    // TODO(wittjosiah): Update ids.
-    await peer
-      .getByTestId(type === 'device' ? 'halo-invitation-accepted-done' : 'space-invitation-accepted-done')
-      .click();
+  async getAuthCode(): Promise<string> {
+    this._authCode = new Trigger<string>();
+    return await this._authCode.wait();
   }
 
-  // TODO(wittjosiah): Peer manager?
-  private _peer(id: number) {
+  peer(id: number) {
     return this.page.getByTestId(`peer-${id}`);
   }
 
@@ -206,8 +151,8 @@ export class InvitationsManager {
       const json = JSON.parse(message.text());
       if (json.invitationCode) {
         this._invitationCode.wake(json.invitationCode);
-      } else if (json.authenticationCode) {
-        this._authenticationCode.wake(json.authenticationCode);
+      } else if (json.authCode) {
+        this._authCode.wake(json.authCode);
       }
     } catch {}
   }

package/src/playwright/invitations.spec.ts

@@ -6,8 +6,7 @@ import { test } from '@playwright/test';
 import { expect } from 'chai';
 
 import { sleep } from '@dxos/async';
-import { ConnectionState } from '@dxos/protocols/proto/dxos/client/services';
-import { AuthMethod } from '@dxos/protocols/proto/dxos/halo/invitations';
+import { ConnectionState, Invitation } from '@dxos/protocols/proto/dxos/client/services';
 
 import { InvitationsManager } from './invitations-manager';
 
@@ -26,12 +25,9 @@ test.describe('Invitations', () => {
       const invitation = await manager.createInvitation(0, 'device');
 
       await manager.openPanel(1, 'identity');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'device', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'device', authenticationCode);
-      await manager.doneInvitation(1, 'device');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'device', invitation)]);
+      await manager.authenticateInvitation('device', authCode, manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
@@ -40,11 +36,11 @@ test.describe('Invitations', () => {
     test.skip('no auth method', async () => {
       await manager.createIdentity(0);
       await manager.openPanel(0, 'devices');
-      const invitation = await manager.createInvitation(0, 'device', { authMethod: AuthMethod.NONE });
+      const invitation = await manager.createInvitation(0, 'device', { authMethod: Invitation.AuthMethod.NONE });
 
       await manager.openPanel(1, 'identity');
       await manager.acceptInvitation(1, 'device', invitation);
-      await manager.doneInvitation(1, 'device');
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
@@ -56,14 +52,11 @@ test.describe('Invitations', () => {
       const invitation = await manager.createInvitation(0, 'device');
 
       await manager.openPanel(1, 'identity');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'device', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'device', '000000');
-      await manager.clearAuthenticationCode(1, 'device');
-      await manager.authenticateInvitation(1, 'device', authenticationCode);
-      await manager.doneInvitation(1, 'device');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'device', invitation)]);
+      await manager.authenticateInvitation('device', '000000', manager.peer(1));
+      await manager.clearAuthCode('device', manager.peer(1));
+      await manager.authenticateInvitation('device', authCode, manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
@@ -82,12 +75,9 @@ test.describe('Invitations', () => {
 
       // TODO(wittjosiah): Retry here.
 
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'device', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'device', authenticationCode);
-      await manager.doneInvitation(1, 'device');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'device', invitation)]);
+      await manager.authenticateInvitation('device', authCode, manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
@@ -102,20 +92,18 @@ test.describe('Invitations', () => {
 
       await manager.openPanel(1, 'identity');
       await manager.acceptInvitation(1, 'device', invitation);
-      await manager.cancelInvitation(1, 'device', 'guest');
-      await manager.resetInvitation(1);
-      await manager.invitationInputContinue(1, 'device');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.clearAuthenticationCode(1, 'device')
-      ]);
-      await manager.authenticateInvitation(1, 'device', authenticationCode);
-      await manager.doneInvitation(1, 'device');
+      await manager.cancelInvitation('device', 'guest', manager.peer(1));
+      await manager.resetInvitation(manager.peer(1));
+      await manager.invitationInputContinue('device', manager.peer(1));
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.clearAuthCode('device', manager.peer(1))]);
+      await manager.authenticateInvitation('device', authCode, manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
 
-    test('recover from network failure during invitation', async () => {
+    // TODO(wittjosiah): Propagating network failure is flaky.
+    test.skip('recover from network failure during invitation', async () => {
       await manager.createIdentity(0);
       await manager.openPanel(0, 'devices');
       const invitation = await manager.createInvitation(0, 'device');
@@ -126,14 +114,11 @@ test.describe('Invitations', () => {
       await manager.setConnectionState(0, ConnectionState.OFFLINE);
       await sleep(100);
       await manager.setConnectionState(0, ConnectionState.ONLINE);
-      await manager.resetInvitation(1);
-      await manager.invitationInputContinue(1, 'device');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.clearAuthenticationCode(1, 'device')
-      ]);
-      await manager.authenticateInvitation(1, 'device', authenticationCode);
-      await manager.doneInvitation(1, 'device');
+      await manager.resetInvitation(manager.peer(1));
+      await manager.invitationInputContinue('device', manager.peer(1));
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.clearAuthCode('device', manager.peer(1))]);
+      await manager.authenticateInvitation('device', authCode, manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(1));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
     });
@@ -146,20 +131,20 @@ test.describe('Invitations', () => {
 
       await manager.openPanel(1, 'identity');
       await manager.openPanel(2, 'identity');
-      const [authenticationCode1] = await Promise.all([
-        manager.getAuthenticationCode(),
+      const [authCode1] = await Promise.all([
+        manager.getAuthCode(),
         manager.acceptInvitation(1, 'device', invitation1)
       ]);
       // Prevent auth code from being reused.
       await sleep(100);
-      const [authenticationCode2] = await Promise.all([
-        manager.getAuthenticationCode(),
+      const [authCode2] = await Promise.all([
+        manager.getAuthCode(),
         manager.acceptInvitation(2, 'device', invitation2)
       ]);
-      await manager.authenticateInvitation(1, 'device', authenticationCode1);
-      await manager.authenticateInvitation(2, 'device', authenticationCode2);
-      await manager.doneInvitation(1, 'device');
-      await manager.doneInvitation(2, 'device');
+      await manager.authenticateInvitation('device', authCode1, manager.peer(1));
+      await manager.authenticateInvitation('device', authCode2, manager.peer(2));
+      await manager.doneInvitation('device', manager.peer(1));
+      await manager.doneInvitation('device', manager.peer(2));
 
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(1));
       expect(await manager.getDisplayName(0)).to.equal(await manager.getDisplayName(2));
@@ -176,12 +161,9 @@ test.describe('Invitations', () => {
 
       await manager.createIdentity(1);
       await manager.openPanel(1, 'join');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'space', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'space', invitation)]);
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -191,12 +173,12 @@ test.describe('Invitations', () => {
       await manager.createIdentity(0);
       await manager.createSpace(0);
       await manager.openPanel(0, 0);
-      const invitation = await manager.createInvitation(0, 'space', { authMethod: AuthMethod.NONE });
+      const invitation = await manager.createInvitation(0, 'space', { authMethod: Invitation.AuthMethod.NONE });
 
       await manager.createIdentity(1);
       await manager.openPanel(1, 'join');
       await manager.acceptInvitation(1, 'space', invitation);
-      await manager.doneInvitation(1, 'space');
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -210,17 +192,14 @@ test.describe('Invitations', () => {
 
       await manager.createIdentity(1);
       await manager.openPanel(1, 'join');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'space', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'space', '000000');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'space', invitation)]);
+      await manager.authenticateInvitation('space', '000000', manager.peer(1));
 
-      expect(await manager.authenticatorIsVisible(1, 'space')).to.be.true;
+      expect(await manager.authenticatorIsVisible('space', manager.peer(1))).to.be.true;
 
-      await manager.clearAuthenticationCode(1, 'space');
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      await manager.clearAuthCode('space', manager.peer(1));
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -236,24 +215,21 @@ test.describe('Invitations', () => {
       await manager.openPanel(1, 'join');
       await manager.acceptInvitation(1, 'space', invitation);
 
-      await manager.authenticateInvitation(1, 'space', '000001');
-      await manager.clearAuthenticationCode(1, 'space');
-      await manager.authenticateInvitation(1, 'space', '000002');
-      await manager.clearAuthenticationCode(1, 'space');
-      await manager.authenticateInvitation(1, 'space', '000003');
+      await manager.authenticateInvitation('space', '000001', manager.peer(1));
+      await manager.clearAuthCode('space', manager.peer(1));
+      await manager.authenticateInvitation('space', '000002', manager.peer(1));
+      await manager.clearAuthCode('space', manager.peer(1));
+      await manager.authenticateInvitation('space', '000003', manager.peer(1));
 
-      expect(await manager.invitationFailed(1)).to.be.true;
+      expect(await manager.invitationFailed(manager.peer(1))).to.be.true;
 
-      await manager.resetInvitation(1);
-      await manager.invitationInputContinue(1, 'space');
+      await manager.resetInvitation(manager.peer(1));
+      await manager.invitationInputContinue('space', manager.peer(1));
 
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.clearAuthenticationCode(1, 'space')
-      ]);
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.clearAuthCode('space', manager.peer(1))]);
 
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -272,12 +248,9 @@ test.describe('Invitations', () => {
 
       // TODO(wittjosiah): Retry here.
 
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'space', invitation)
-      ]);
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'space', invitation)]);
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -295,23 +268,21 @@ test.describe('Invitations', () => {
       await manager.createIdentity(1);
       await manager.openPanel(1, 'join');
       await manager.acceptInvitation(1, 'space', invitation);
-      await manager.cancelInvitation(1, 'space', 'guest');
-      await manager.resetInvitation(1);
-      await manager.invitationInputContinue(1, 'space');
+      await manager.cancelInvitation('space', 'guest', manager.peer(1));
+      await manager.resetInvitation(manager.peer(1));
+      await manager.invitationInputContinue('space', manager.peer(1));
 
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.clearAuthenticationCode(1, 'space')
-      ]);
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.clearAuthCode('space', manager.peer(1))]);
 
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
     });
 
-    test('recover from network failure during invitation', async () => {
+    // TODO(wittjosiah): Propagating network failure is flaky.
+    test.skip('recover from network failure during invitation', async () => {
       await manager.createIdentity(0);
       await manager.createSpace(0);
       await manager.openPanel(0, 0);
@@ -324,14 +295,11 @@ test.describe('Invitations', () => {
       await manager.setConnectionState(0, ConnectionState.OFFLINE);
       await sleep(100);
       await manager.setConnectionState(0, ConnectionState.ONLINE);
-      await manager.resetInvitation(1);
-      await manager.invitationInputContinue(1, 'space');
-      const [authenticationCode] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.clearAuthenticationCode(1, 'space')
-      ]);
-      await manager.authenticateInvitation(1, 'space', authenticationCode);
-      await manager.doneInvitation(1, 'space');
+      await manager.resetInvitation(manager.peer(1));
+      await manager.invitationInputContinue('space', manager.peer(1));
+      const [authCode] = await Promise.all([manager.getAuthCode(), manager.clearAuthCode('space', manager.peer(1))]);
+      await manager.authenticateInvitation('space', authCode, manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(1));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));
@@ -348,20 +316,14 @@ test.describe('Invitations', () => {
       await manager.createIdentity(2);
       await manager.openPanel(1, 'join');
       await manager.openPanel(2, 'join');
-      const [authenticationCode1] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(1, 'space', invitation1)
-      ]);
+      const [authCode1] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(1, 'space', invitation1)]);
       // Prevent auth code from being reused.
       await sleep(100);
-      const [authenticationCode2] = await Promise.all([
-        manager.getAuthenticationCode(),
-        manager.acceptInvitation(2, 'space', invitation2)
-      ]);
-      await manager.authenticateInvitation(1, 'space', authenticationCode1);
-      await manager.authenticateInvitation(2, 'space', authenticationCode2);
-      await manager.doneInvitation(1, 'space');
-      await manager.doneInvitation(2, 'space');
+      const [authCode2] = await Promise.all([manager.getAuthCode(), manager.acceptInvitation(2, 'space', invitation2)]);
+      await manager.authenticateInvitation('space', authCode1, manager.peer(1));
+      await manager.authenticateInvitation('space', authCode2, manager.peer(2));
+      await manager.doneInvitation('space', manager.peer(1));
+      await manager.doneInvitation('space', manager.peer(2));
 
       await manager.openPanel(0, 'spaces');
       expect(await manager.getSpaceName(0, 0)).to.equal(await manager.getSpaceName(1, 0));

package/src/testing/index.ts

@@ -0,0 +1,5 @@
+//
+// Copyright 2023 DXOS.org
+//
+
+export * from './shell-manager';

package/src/testing/shell-manager.ts

@@ -0,0 +1,59 @@
+//
+// Copyright 2023 DXOS.org
+//
+
+import type { Locator, Page, FrameLocator } from 'playwright';
+
+type Scope = Locator | FrameLocator | Page;
+
+export class ShellManager {
+  page!: Page;
+
+  authenticatorIsVisible(type: 'device' | 'space', scope?: Scope) {
+    return (scope || this.page).getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).isVisible();
+  }
+
+  invitationFailed(scope?: Scope) {
+    return (scope || this.page).getByTestId('invitation-rescuer-reset').isVisible();
+  }
+
+  async inputInvitation(type: 'device' | 'space', invitation: string, scope?: Scope) {
+    await (scope || this.page).getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-input`).type(invitation);
+    await this.page.keyboard.press('Enter');
+  }
+
+  async invitationInputContinue(type: 'device' | 'space', scope?: Scope) {
+    await (scope || this.page).getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-input-continue`).click();
+  }
+
+  async cancelInvitation(type: 'device' | 'space', kind: 'host' | 'guest', scope?: Scope) {
+    if (kind === 'guest') {
+      await (scope || this.page)
+        .getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-authenticator-cancel`)
+        .click();
+    }
+  }
+
+  async authenticateInvitation(type: 'device' | 'space', authCode: string, scope?: Scope) {
+    const peer = scope || this.page;
+    // TODO(wittjosiah): Update ids.
+    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).type(authCode);
+    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-invitation-authenticator-next`).click();
+  }
+
+  async clearAuthCode(type: 'device' | 'space', scope?: Scope) {
+    const peer = scope || this.page;
+    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).fill('');
+    await peer.getByTestId(`${type === 'device' ? 'halo' : 'space'}-auth-code-input`).focus();
+  }
+
+  async resetInvitation(scope?: Scope) {
+    await (scope || this.page).getByTestId('invitation-rescuer-reset').click();
+  }
+
+  async doneInvitation(type: 'device' | 'space', scope?: Scope) {
+    await (scope || this.page)
+      .getByTestId(type === 'device' ? 'halo-invitation-accepted-done' : 'space-invitation-accepted-done')
+      .click();
+  }
+}

package/src/util/invitationStatusValue.ts

@@ -16,6 +16,7 @@ export const invitationStatusValue = new Map<Invitation.State, number>([
   [Invitation.State.INIT, 0],
   [Invitation.State.CONNECTING, 1],
   [Invitation.State.CONNECTED, 2],
-  [Invitation.State.AUTHENTICATING, 3],
-  [Invitation.State.SUCCESS, 4]
+  [Invitation.State.READY_FOR_AUTHENTICATION, 3],
+  [Invitation.State.AUTHENTICATING, 4],
+  [Invitation.State.SUCCESS, 5]
 ]);