outline/server/queues/tasks/CleanupDeletedDocumentsTask.test.ts

import { subDays } from "date-fns";
import { Document } from "@server/models";
import { buildDocument, buildTeam } from "@server/test/factories";
import CleanupDeletedDocumentsTask from "./CleanupDeletedDocumentsTask";

const props = {
  limit: 100,
  partition: {
    partitionIndex: 0,
    partitionCount: 1,
  },
};

describe("CleanupDeletedDocumentsTask", () => {
  it("should not destroy documents not deleted", async () => {
    const team = await buildTeam();
    await buildDocument({
      teamId: team.id,
      publishedAt: new Date(),
    });

    const task = new CleanupDeletedDocumentsTask();
    await task.perform(props);

    expect(
      await Document.unscoped().count({
        where: {
          teamId: team.id,
        },
        paranoid: false,
      })
    ).toEqual(1);
  });

  it("should not destroy documents deleted less than 30 days ago", async () => {
    const team = await buildTeam();
    await buildDocument({
      teamId: team.id,
      publishedAt: new Date(),
      deletedAt: subDays(new Date(), 25),
    });

    const task = new CleanupDeletedDocumentsTask();
    await task.perform(props);

    expect(
      await Document.unscoped().count({
        where: {
          teamId: team.id,
        },
        paranoid: false,
      })
    ).toEqual(1);
  });

  it("should destroy documents deleted more than 30 days ago", async () => {
    const team = await buildTeam();
    await buildDocument({
      teamId: team.id,
      publishedAt: new Date(),
      deletedAt: subDays(new Date(), 60),
    });

    const task = new CleanupDeletedDocumentsTask();
    await task.perform(props);

    expect(
      await Document.unscoped().count({
        where: {
          teamId: team.id,
        },
        paranoid: false,
      })
    ).toEqual(0);
  });

  it("should destroy draft documents deleted more than 30 days ago", async () => {
    const team = await buildTeam();
    await buildDocument({
      teamId: team.id,
      publishedAt: undefined,
      deletedAt: subDays(new Date(), 60),
    });

    const task = new CleanupDeletedDocumentsTask();
    await task.perform(props);

    expect(
      await Document.unscoped().count({
        where: {
          teamId: team.id,
        },
        paranoid: false,
      })
    ).toEqual(0);
  });
});