From 3f7d4a4cd9a06f33bb51cbe678ab06dbe456dda6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karol=20S=C3=B3jko?= <karolsojko@protonmail.com>
Date: Mon, 17 Oct 2022 15:07:30 +0200
Subject: [PATCH] feat: add workspaces workflow e2e tests (#1810)

* feat: add workspaces workflow e2e tests

* fix: collection finding data for assertions
---
 packages/snjs/mocha/test.html          |   1 +
 packages/snjs/mocha/workspaces.test.js | 103 +++++++++++++++++++++++++
 2 files changed, 104 insertions(+)
 create mode 100644 packages/snjs/mocha/workspaces.test.js

diff --git a/packages/snjs/mocha/test.html b/packages/snjs/mocha/test.html
index cf4718eb0..81e4f541f 100644
--- a/packages/snjs/mocha/test.html
+++ b/packages/snjs/mocha/test.html
@@ -92,6 +92,7 @@
   <script type="module" src="files.test.js"></script>
   <script type="module" src="session.test.js"></script>
   <script type="module" src="subscriptions.test.js"></script>
+  <script type="module" src="workspaces.test.js"></script>
   <script type="module">
     mocha.run();
   </script>
diff --git a/packages/snjs/mocha/workspaces.test.js b/packages/snjs/mocha/workspaces.test.js
new file mode 100644
index 000000000..b647d11a4
--- /dev/null
+++ b/packages/snjs/mocha/workspaces.test.js
@@ -0,0 +1,103 @@
+import * as Factory from './lib/factory.js'
+chai.use(chaiAsPromised)
+const expect = chai.expect
+
+describe('workspaces', function () {
+  this.timeout(Factory.TwentySecondTimeout)
+
+  let ownerContext
+  let owner
+  let inviteeContext
+  let invitee
+
+  afterEach(async function () {
+    await Factory.safeDeinit(ownerContext.application)
+    await Factory.safeDeinit(inviteeContext.application)
+    localStorage.clear()
+  })
+
+  beforeEach(async function () {
+    localStorage.clear()
+
+    ownerContext = await Factory.createAppContextWithFakeCrypto()
+    await ownerContext.launch()
+    const ownerRegistrationResponse = await ownerContext.register()
+    owner = ownerRegistrationResponse.user
+
+    inviteeContext = await Factory.createAppContextWithFakeCrypto()
+    await inviteeContext.launch()
+    const inviteeRegistrationResponse = await inviteeContext.register()
+    invitee = inviteeRegistrationResponse.user
+  })
+
+  it('should create workspaces for user', async () => {
+    await ownerContext.application.workspaceManager.createWorkspace({
+      workspaceType: 'team',
+      encryptedWorkspaceKey: 'foo',
+      encryptedPrivateKey: 'bar',
+      publicKey: 'buzz',
+      workspaceName: 'Acme Team',
+    })
+
+    await ownerContext.application.workspaceManager.createWorkspace({
+      workspaceType: 'private',
+      encryptedWorkspaceKey: 'foo',
+      encryptedPrivateKey: 'bar',
+      publicKey: 'buzz',
+    })
+
+    const { ownedWorkspaces, joinedWorkspaces } = await ownerContext.application.workspaceManager.listWorkspaces()
+
+    expect(joinedWorkspaces.length).to.equal(0)
+
+    expect(ownedWorkspaces.length).to.equal(2)
+  })
+
+  it('should allow inviting and adding users to a workspace', async () => {
+    const { uuid } = await ownerContext.application.workspaceManager.createWorkspace({
+      workspaceType: 'team',
+      encryptedWorkspaceKey: 'foo',
+      encryptedPrivateKey: 'bar',
+      publicKey: 'buzz',
+      workspaceName: 'Acme Team',
+    })
+
+    const result = await ownerContext.application.workspaceManager.inviteToWorkspace({
+      inviteeEmail: 'test@standardnotes.com',
+      workspaceUuid: uuid,
+      accessLevel: 'read-only'
+    })
+
+    await inviteeContext.application.workspaceManager.acceptInvite({
+      inviteUuid: result.uuid,
+      userUuid: invitee.uuid,
+      publicKey: 'foobar',
+      encryptedPrivateKey: 'buzz',
+    })
+
+    let listUsersResult = await inviteeContext.application.workspaceManager.listWorkspaceUsers({ workspaceUuid: uuid })
+    let inviteeAssociation = listUsersResult.users.find(user => user.userDisplayName === 'test@standardnotes.com')
+
+    expect(inviteeAssociation.userUuid).to.equal(invitee.uuid)
+    expect(inviteeAssociation.status).to.equal('pending-keyshare')
+
+    await ownerContext.application.workspaceManager.initiateKeyshare({
+      workspaceUuid: inviteeAssociation.workspaceUuid,
+      userUuid: inviteeAssociation.userUuid,
+      encryptedWorkspaceKey: 'foobarbuzz',
+    })
+
+    listUsersResult = await inviteeContext.application.workspaceManager.listWorkspaceUsers({ workspaceUuid: uuid })
+    inviteeAssociation = listUsersResult.users.find(user => user.userDisplayName === 'test@standardnotes.com')
+
+    expect(inviteeAssociation.userUuid).to.equal(invitee.uuid)
+    expect(inviteeAssociation.status).to.equal('active')
+    expect(inviteeAssociation.encryptedWorkspaceKey).to.equal('foobarbuzz')
+
+    const { ownedWorkspaces, joinedWorkspaces } = await inviteeContext.application.workspaceManager.listWorkspaces()
+
+    expect(joinedWorkspaces.length).to.equal(1)
+
+    expect(ownedWorkspaces.length).to.equal(0)
+  })
+})