feat(server): add invitation status to getInviteInfo response (#11158)

close CLOUD-182

packages/backend/server/src/__tests__/e2e/workspace/invite.spec.ts

@@ -1,9 +1,12 @@
 import {
   acceptInviteByInviteIdMutation,
+  createInviteLinkMutation,
+  getInviteInfoQuery,
   getMembersByWorkspaceIdQuery,
   inviteByEmailMutation,
   leaveWorkspaceMutation,
   revokeMemberPermissionMutation,
+  WorkspaceInviteLinkExpireTime,
   WorkspaceMemberStatus,
 } from '@affine/graphql';
 import { faker } from '@faker-js/faker';
@@ -32,6 +35,34 @@ e2e('should invite a user', async t => {
   const invitationNotification = app.queue.last('notification.sendInvitation');
   t.is(invitationNotification.payload.inviterId, owner.id);
   t.is(invitationNotification.payload.inviteId, result.invite);
+
+  // invitation status is pending
+  const { getInviteInfo } = await app.gql({
+    query: getInviteInfoQuery,
+    variables: {
+      inviteId: result.invite,
+    },
+  });
+  t.is(getInviteInfo.status, WorkspaceMemberStatus.Pending);
+
+  // u2 accept invite
+  await app.switchUser(u2);
+  await app.gql({
+    query: acceptInviteByInviteIdMutation,
+    variables: {
+      workspaceId: workspace.id,
+      inviteId: result.invite,
+    },
+  });
+
+  // invitation status is accepted
+  const { getInviteInfo: getInviteInfo2 } = await app.gql({
+    query: getInviteInfoQuery,
+    variables: {
+      inviteId: result.invite,
+    },
+  });
+  t.is(getInviteInfo2.status, WorkspaceMemberStatus.Accepted);
 });
 
 e2e('should leave a workspace', async t => {
@@ -290,7 +321,7 @@ e2e('should limit member count correctly', async t => {
   const workspace = await app.create(Mockers.Workspace, {
     owner: { id: owner.id },
   });
-  await Promise.allSettled(
+  await Promise.all(
     Array.from({ length: 10 }).map(async () => {
       const user = await app.signup();
       await app.create(Mockers.WorkspaceUser, {
@@ -312,3 +343,62 @@ e2e('should limit member count correctly', async t => {
   t.is(result.workspace.memberCount, 11);
   t.is(result.workspace.members.length, 10);
 });
+
+e2e('should get invite link info with status', async t => {
+  const owner = await app.signup();
+
+  const workspace = await app.create(Mockers.Workspace, {
+    owner: { id: owner.id },
+  });
+
+  await app.login(owner);
+  const { createInviteLink } = await app.gql({
+    query: createInviteLinkMutation,
+    variables: {
+      workspaceId: workspace.id,
+      expireTime: WorkspaceInviteLinkExpireTime.OneDay,
+    },
+  });
+  t.truthy(createInviteLink, 'failed to create invite link');
+  const link = createInviteLink.link;
+  const inviteId = link.split('/').pop()!;
+
+  // owner/member see accept status
+  const { getInviteInfo } = await app.gql({
+    query: getInviteInfoQuery,
+    variables: {
+      inviteId,
+    },
+  });
+  t.truthy(getInviteInfo, 'failed to get invite info');
+  t.is(getInviteInfo.status, WorkspaceMemberStatus.Accepted);
+
+  // non-member see null status
+  await app.signup();
+  const { getInviteInfo: getInviteInfo2 } = await app.gql({
+    query: getInviteInfoQuery,
+    variables: {
+      inviteId,
+    },
+  });
+  t.truthy(getInviteInfo2, 'failed to get invite info');
+  t.is(getInviteInfo2.status, null);
+
+  // pending-member see under review status
+  await app.signup();
+  await app.gql({
+    query: acceptInviteByInviteIdMutation,
+    variables: {
+      workspaceId: workspace.id,
+      inviteId,
+    },
+  });
+  const { getInviteInfo: getInviteInfo3 } = await app.gql({
+    query: getInviteInfoQuery,
+    variables: {
+      inviteId,
+    },
+  });
+  t.truthy(getInviteInfo3, 'failed to get invite info');
+  t.is(getInviteInfo3.status, WorkspaceMemberStatus.UnderReview);
+});

packages/backend/server/src/__tests__/team.e2e.ts

@@ -544,27 +544,34 @@ test('should be able to invite by link', async t => {
   const [teamInviteId, teamInvite, acceptTeamInvite] =
     await createInviteLink(tws);
 
+  const member = await app.signup();
   {
     // check invite link
-    app.switchUser(owner);
+    app.switchUser(member);
     const info = await getInviteInfo(app, inviteId);
     t.is(info.workspace.id, ws.id, 'should be able to get invite info');
+    t.falsy(info.status);
 
     // check team invite link
     const teamInfo = await getInviteInfo(app, teamInviteId);
     t.is(teamInfo.workspace.id, tws.id, 'should be able to get invite info');
+    t.falsy(info.status);
   }
 
   {
     // invite link
     for (const [i] of Array.from({ length: 5 }).entries()) {
       const user = await invite(`test${i}@affine.pro`);
-      const status = (await models.workspaceUser.get(ws.id, user.id))?.status;
+      const role = await models.workspaceUser.get(ws.id, user.id);
+      t.truthy(role);
+      const status = role!.status;
       t.is(
         status,
         WorkspaceMemberStatus.UnderReview,
         'should be able to check status'
       );
+      const info = await getInviteInfo(app, role!.id);
+      t.is(info.status, WorkspaceMemberStatus.UnderReview);
     }
 
     await t.throwsAsync(

packages/backend/server/src/__tests__/utils/invite.ts

@@ -163,6 +163,7 @@ export async function getInviteInfo(
           name
           avatarUrl
         }
+        status
       }
     }
   `);