chore: Test performance (#5786)

server/queues/tasks/CleanupDeletedDocumentsTask.test.ts

@@ -1,11 +1,8 @@
 import { subDays } from "date-fns";
 import { Document } from "@server/models";
 import { buildDocument, buildTeam } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import CleanupDeletedDocumentsTask from "./CleanupDeletedDocumentsTask";
 
-setupTestDatabase();
-
 describe("CleanupDeletedDocumentsTask", () => {
   it("should not destroy documents not deleted", async () => {
     const team = await buildTeam();

server/queues/tasks/CleanupDemotedUserTask.test.ts

@@ -6,11 +6,8 @@ import {
   buildWebhookSubscription,
   buildViewer,
 } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import CleanupDemotedUserTask from "./CleanupDemotedUserTask";
 
-setupTestDatabase();
-
 describe("CleanupDemotedUserTask", () => {
   it("should delete api keys for suspended user", async () => {
     const admin = await buildAdmin();

server/queues/tasks/CleanupExpiredFileOperationsTask.test.ts

@@ -2,11 +2,8 @@ import { subDays } from "date-fns";
 import { FileOperationState, FileOperationType } from "@shared/types";
 import { FileOperation } from "@server/models";
 import { buildFileOperation, buildTeam } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import CleanupExpiredFileOperationsTask from "./CleanupExpiredFileOperationsTask";
 
-setupTestDatabase();
-
 describe("CleanupExpiredFileOperationsTask", () => {
   it("should expire exports older than 15 days ago", async () => {
     const team = await buildTeam();

server/queues/tasks/DetachDraftsFromCollectionTask.test.ts

@@ -1,10 +1,7 @@
 import { Document } from "@server/models";
 import { buildCollection, buildDocument } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import DetachDraftsFromCollectionTask from "./DetachDraftsFromCollectionTask";
 
-setupTestDatabase();
-
 describe("DetachDraftsFromCollectionTask", () => {
   const ip = "127.0.0.1";
   it("should detach drafts from deleted collection", async () => {

server/queues/tasks/DocumentPublishedNotificationsTask.test.ts

@@ -5,13 +5,10 @@ import {
   buildCollection,
   buildUser,
 } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import DocumentPublishedNotificationsTask from "./DocumentPublishedNotificationsTask";
 
 const ip = "127.0.0.1";
 
-setupTestDatabase();
-
 beforeEach(async () => {
   jest.resetAllMocks();
 });

server/queues/tasks/ErrorTimedOutFileOperationsTask.test.ts

@@ -2,11 +2,8 @@ import { subDays } from "date-fns";
 import { FileOperationState, FileOperationType } from "@shared/types";
 import { FileOperation } from "@server/models";
 import { buildFileOperation, buildTeam } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import ErrorTimedOutFileOperationsTask from "./ErrorTimedOutFileOperationsTask";
 
-setupTestDatabase();
-
 describe("ErrorTimedOutFileOperationsTask", () => {
   it("should error exports older than 12 hours", async () => {
     const team = await buildTeam();

server/queues/tasks/ImportMarkdownZipTask.test.ts

@@ -2,11 +2,8 @@ import fs from "fs";
 import path from "path";
 import { FileOperation } from "@server/models";
 import { buildFileOperation } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import ImportMarkdownZipTask from "./ImportMarkdownZipTask";
 
-setupTestDatabase();
-
 describe("ImportMarkdownZipTask", () => {
   it("should import the documents, attachments", async () => {
     const fileOperation = await buildFileOperation();

server/queues/tasks/ImportNotionTask.test.ts

@@ -2,11 +2,8 @@ import fs from "fs";
 import path from "path";
 import { FileOperation } from "@server/models";
 import { buildFileOperation } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import ImportNotionTask from "./ImportNotionTask";
 
-setupTestDatabase();
-
 describe("ImportNotionTask", () => {
   it("should import successfully from a Markdown export", async () => {
     const fileOperation = await buildFileOperation();

server/queues/tasks/InviteReminderTask.test.ts

@@ -1,11 +1,8 @@
 import { subDays } from "date-fns";
 import InviteReminderEmail from "@server/emails/templates/InviteReminderEmail";
 import { buildInvite } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import InviteReminderTask from "./InviteReminderTask";
 
-setupTestDatabase();
-
 describe("InviteReminderTask", () => {
   it("should not destroy documents not deleted", async () => {
     const spy = jest.spyOn(InviteReminderEmail.prototype, "schedule");

server/queues/tasks/RevisionCreatedNotificationsTask.test.ts

@@ -6,13 +6,10 @@ import {
   Revision,
 } from "@server/models";
 import { buildDocument, buildUser } from "@server/test/factories";
-import { setupTestDatabase } from "@server/test/support";
 import RevisionCreatedNotificationsTask from "./RevisionCreatedNotificationsTask";
 
 const ip = "127.0.0.1";
 
-setupTestDatabase();
-
 beforeEach(async () => {
   jest.resetAllMocks();
 });

server/queues/tasks/index.ts

@@ -1,10 +1,12 @@
 import path from "path";
 import { glob } from "glob";
+import env from "@server/env";
 import Logger from "@server/logging/Logger";
 import { requireDirectory } from "@server/utils/fs";
 import BaseTask from "./BaseTask";
 
 const tasks = {};
+const rootDir = env.ENVIRONMENT === "test" ? "" : "build";
 
 requireDirectory<{ default: BaseTask<any> }>(__dirname).forEach(
   ([module, id]) => {
@@ -16,7 +18,7 @@ requireDirectory<{ default: BaseTask<any> }>(__dirname).forEach(
 );
 
 glob
-  .sync("build/plugins/*/server/tasks/!(*.test).js")
+  .sync(path.join(rootDir, "plugins/*/server/tasks/!(*.test).[jt]s"))
   .forEach((filePath: string) => {
     // eslint-disable-next-line @typescript-eslint/no-var-requires
     const task = require(path.join(process.cwd(), filePath)).default;