workavia-drive/twake/backend/node/test/e2e/messages/messages.messages.spec.ts

import "reflect-metadata";
import { afterEach, describe, expect, it } from "@jest/globals";
import { init, TestPlatform } from "../setup";
import {
  ResourceListResponse,
  ResourceUpdateResponse,
  User,
  Workspace,
} from "../../../src/utils/types";
import { deserialize } from "class-transformer";
import { ParticipantObject, Thread } from "../../../src/services/messages/entities/threads";
import {
  createMessage,
  createParticipant,
  e2e_createChannel,
  e2e_createMessage,
  e2e_createThread,
} from "./utils";
import { Message } from "../../../src/services/messages/entities/messages";
import { v1 as uuidv1 } from "uuid";
import { MessageWithReplies } from "../../../src/services/messages/types";
import { TestDbService } from "../utils.prepare.db";
import gr from "../../../src/services/global-resolver";
import { ChannelVisibility, WorkspaceExecutionContext } from "../../../src/services/channels/types";
import { ChannelSaveOptions } from "../../../src/services/channels/web/types";
import { ChannelUtils, get as getChannelUtils } from "../channels/utils";

describe("The Messages feature", () => {
  const url = "/internal/services/messages/v1";
  let platform: TestPlatform;
  let testDbService: TestDbService;
  let channelUtils: ChannelUtils;

  function getContext(user?: User): WorkspaceExecutionContext {
    return {
      workspace: platform.workspace,
      user: user || platform.currentUser,
    };
  }

  beforeAll(async () => {
    platform = await init({
      services: [
        "webserver",
        "database",
        "search",
        "storage",
        "message-queue",
        "applications",
        "user",
        "websocket",
        "webserver",
        "messages",
        "files",
        "auth",
        "search",
        "realtime",
        "channels",
        "counter",
        "statistics",
        "platform-services",
      ],
    });
    await gr.database.getConnector().drop();

    channelUtils = getChannelUtils(platform);
  });

  afterEach(async () => {
    await platform.tearDown();
  });

  describe("On user use messages in a thread", () => {
    it("should create a message in a thread", async () => {
      const response = await e2e_createThread(
        platform,
        [],
        createMessage({ text: "Initial thread message" }),
      );
      const result: ResourceUpdateResponse<Thread> = deserialize(
        ResourceUpdateResponse,
        response.body,
      );
      const threadId = result.resource.id;

      await e2e_createMessage(platform, threadId, createMessage({ text: "Reply 1" }));

      await e2e_createMessage(platform, threadId, createMessage({ text: "Reply 2" }));

      const jwtToken = await platform.auth.getJWTToken();
      const listResponse = await platform.app.inject({
        method: "GET",
        url: `${url}/companies/${platform.workspace.company_id}/threads/${threadId}/messages`,
        headers: {
          authorization: `Bearer ${jwtToken}`,
        },
      });

      const listResult: ResourceListResponse<Thread> = deserialize(
        ResourceListResponse,
        listResponse.body,
      );

      expect(listResponse.statusCode).toBe(200);
      expect(listResult.resources.length).toBe(3);
    });
  });

  describe("Inbox", () => {
    it("Should get recent user messages", async done => {
      const directChannelIn = channelUtils.getDirectChannel();
      const members = [platform.currentUser.id, uuidv1()];
      const directWorkspace: Workspace = {
        company_id: platform.workspace.company_id,
        workspace_id: ChannelVisibility.DIRECT,
      };

      await Promise.all([
        gr.services.channels.channels.save<ChannelSaveOptions>(
          directChannelIn,
          {
            members,
          },
          { ...getContext(), ...{ workspace: directWorkspace } },
        ),
      ]);

      const recipient: ParticipantObject = {
        created_at: 0,
        created_by: "",
        type: "channel",
        company_id: directChannelIn.company_id,
        workspace_id: "direct",
        id: directChannelIn.id,
      };

      for (let i = 0; i < 6; i++) {
        const response = await e2e_createThread(
          platform,
          [recipient],
          createMessage({ text: `Initial thread message ${i}` }),
        );
        const result: ResourceUpdateResponse<Thread> = deserialize(
          ResourceUpdateResponse,
          response.body,
        );
        const threadId = result.resource.id;

        const replies = [];
        for (let j = 0; j < 3; j++) {
          replies.push(
            e2e_createMessage(
              platform,
              threadId,
              createMessage({ text: `Reply ${j} to message ${i}` }),
            ),
          );
        }
        await Promise.all(replies);
      }

      const jwtToken = await platform.auth.getJWTToken({ sub: members[0] });
      const response = await platform.app.inject({
        method: "GET",
        url: `${url}/companies/${platform.workspace.company_id}/inbox?limit=5`,
        headers: {
          authorization: `Bearer ${jwtToken}`,
        },
      });

      expect(response.statusCode).toEqual(200);

      const inbox: ResourceListResponse<MessageWithReplies> = deserialize(
        ResourceListResponse,
        response.body,
      );

      expect(inbox.resources.length).toEqual(5);

      for (const resource of inbox.resources) {
        expect(resource).toMatchObject({
          id: expect.any(String),
          thread_id: expect.any(String),
          created_at: expect.any(Number),
          updated_at: expect.any(Number),
          user_id: platform.currentUser.id,
          application_id: null,
          text: expect.any(String),
          cache: {
            company_id: directChannelIn.company_id,
            channel_id: directChannelIn.id,
          },
        });
      }

      done();
    });
  });
});