✨ Handling quota limit error (#437)

2024-03-26 09:04:36 +01:00
parent 389d42fe4f
commit cb5a2ad3b7
14 changed files with 235 additions and 35 deletions
@@ -233,8 +233,14 @@ export default class UserApi {
  };

  async createDocumentFromFilename(
-    file_name: "sample.png",
-    parent_id = "root"
+    file_name:
+      | "sample.png"
+      | "sample.doc"
+      | "sample.pdf"
+      | "sample.zip"
+      | "sample.mp4"
+      | "sample.gif",
+    parent_id = "root",
  ) {
    const file = await this.uploadFile(file_name);

@@ -0,0 +1,116 @@
+import { afterEach, beforeEach, describe, expect, it } from "@jest/globals";
+import { init, TestPlatform } from "../setup";
+import UserApi from "../common/user-api";
+import config from "config";
+import { e2e_createDocumentFile, e2e_createVersion } from "./utils";
+import { deserialize } from "class-transformer";
+import { ResourceUpdateResponse } from "../../../src/utils/types";
+
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-ignore
+jest.mock("config");
+
+describe("the Drive feature", () => {
+  const filesUrl = "/internal/services/files/v1";
+  let platform: TestPlatform;
+  let currentUser: UserApi;
+  let configHasSpy: jest.SpyInstance;
+  let configGetSpy: jest.SpyInstance;
+
+  beforeEach(async () => {
+    configHasSpy = jest.spyOn(config, "has");
+    configGetSpy = jest.spyOn(config, "get");
+
+    configHasSpy.mockImplementation((setting: string) => {
+      const value = jest.requireActual("config").has(setting);
+      return value;
+    });
+    configGetSpy.mockImplementation((setting: string) => {
+      if (setting === "drive.featureUserQuota") {
+        return true;
+      }
+      if (setting === "drive.defaultUserQuota") {
+        return 2000000;
+      }
+      return jest.requireActual("config").get(setting);
+    });
+    platform = await init({
+      services: [
+        "webserver",
+        "database",
+        "applications",
+        "search",
+        "storage",
+        "message-queue",
+        "user",
+        "search",
+        "files",
+        "websocket",
+        "messages",
+        "auth",
+        "realtime",
+        "channels",
+        "counter",
+        "statistics",
+        "platform-services",
+        "documents",
+      ],
+    });
+    currentUser = await UserApi.getInstance(platform);
+  });
+
+  afterEach(async () => {
+    await platform?.tearDown();
+    platform = null;
+    configGetSpy.mockRestore();
+  });
+
+  it("did create the drive item with size under quota", async () => {
+    const result = await currentUser.uploadFileAndCreateDocument("sample.doc");
+    expect(result).toBeDefined();
+  });
+
+  it("did not upload the drive item with size above quota", async () => {
+    const item = await currentUser.uploadFile("sample.mp4");
+    expect(item).toBeDefined();
+    const result: any = await currentUser.createDocumentFromFile(item);
+    expect(result).toBeDefined();
+    expect(result.statusCode).toBe(403);
+    expect(result.error).toBe("Forbidden");
+    expect(result.message).toContain("Not enough space");
+    const fileDownloadResponse = await platform.app.inject({
+      method: "GET",
+      url: `${filesUrl}/companies/${platform.workspace.company_id}/files/${item.id}/download`,
+    });
+    // make sure the file was removed
+    expect(fileDownloadResponse).toBeTruthy();
+    expect(fileDownloadResponse.statusCode).toBe(404);
+  });
+
+  it("did create a version for a drive item", async () => {
+    const item = await currentUser.createDefaultDocument();
+    const fileUploadResponse = await e2e_createDocumentFile(
+      platform,
+      "../common/assets/sample.mp4",
+    );
+    const fileUploadResult: any = deserialize<ResourceUpdateResponse<File>>(
+      ResourceUpdateResponse,
+      fileUploadResponse.body,
+    );
+    const file_metadata = { external_id: fileUploadResult.resource.id };
+
+    const result: any = await e2e_createVersion(platform, item.id, {
+      filename: "file2",
+      file_metadata,
+    });
+    expect(result).toBeDefined();
+    expect(result.statusCode).toBe(403);
+    const fileDownloadResponse = await platform.app.inject({
+      method: "GET",
+      url: `${filesUrl}/companies/${platform.workspace.company_id}/files/${item.id}/download`,
+    });
+    // make sure the file was removed
+    expect(fileDownloadResponse).toBeTruthy();
+    expect(fileDownloadResponse.statusCode).toBe(404);
+  });
+});
@@ -54,8 +54,9 @@ export const e2e_createVersion = async (
  });
 };

-export const e2e_createDocumentFile = async (platform: TestPlatform) => {
-  const filePath = `${__dirname}/assets/test.txt`;
+export const e2e_createDocumentFile = async (platform: TestPlatform, documentPath?: string) => {
+  const subFilePath = documentPath ?? "assets/test.txt";
+  const filePath = `${__dirname}/${subFilePath}`;
  const token = await platform.auth.getJWTToken();
  const form = formAutoContent({ file: fs.createReadStream(filePath) });
  form.headers["authorization"] = `Bearer ${token}`;