feat(enterprise): add GDPR user data export (async)

2026-08-03 07:46:42 +02:00 · 2026-06-13 21:07:28 +08:00
parent fa7da7ce0c
commit b6a8226837
5 changed files with 337 additions and 0 deletions
@@ -0,0 +1,89 @@
+import { eq } from "drizzle-orm";
+import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { db, schema } from "../../apps/api/src/db/index.js";
+import { buildTestApp, loginAsAdmin, type TestApp } from "./test-server.js";
+
+let testApp: TestApp;
+let adminToken: string;
+
+beforeAll(async () => {
+  testApp = await buildTestApp();
+  adminToken = await loginAsAdmin(testApp.app);
+}, 30_000);
+
+afterAll(async () => {
+  await testApp.cleanup();
+}, 10_000);
+
+describe("GDPR data export", () => {
+  it("returns 403 for POST without enterprise license", async () => {
+    const res = await testApp.app.inject({
+      method: "POST",
+      url: "/api/v1/enterprise/users/some-id/export",
+      headers: { authorization: `Bearer ${adminToken}` },
+    });
+    expect(res.statusCode).toBe(403);
+    const body = JSON.parse(res.body);
+    expect(body.error).toContain("enterprise");
+  });
+
+  it("returns 403 for GET without enterprise license", async () => {
+    const res = await testApp.app.inject({
+      method: "GET",
+      url: "/api/v1/enterprise/users/some-id/export/some-job-id",
+      headers: { authorization: `Bearer ${adminToken}` },
+    });
+    expect(res.statusCode).toBe(403);
+    const body = JSON.parse(res.body);
+    expect(body.error).toContain("enterprise");
+  });
+
+  it("returns 401 for POST without auth", async () => {
+    const res = await testApp.app.inject({
+      method: "POST",
+      url: "/api/v1/enterprise/users/some-id/export",
+    });
+    expect(res.statusCode).toBe(401);
+  });
+
+  it("returns 401 for GET without auth", async () => {
+    const res = await testApp.app.inject({
+      method: "GET",
+      url: "/api/v1/enterprise/users/some-id/export/some-job-id",
+    });
+    expect(res.statusCode).toBe(401);
+  });
+
+  it("returns 403 for non-admin user", async () => {
+    // Create a regular user
+    await testApp.app.inject({
+      method: "POST",
+      url: "/api/auth/register",
+      headers: { authorization: `Bearer ${adminToken}` },
+      payload: {
+        username: "gdprexportuser",
+        password: "TestPass1",
+        role: "user",
+      },
+    });
+    await db
+      .update(schema.users)
+      .set({ mustChangePassword: false })
+      .where(eq(schema.users.username, "gdprexportuser"));
+
+    const loginRes = await testApp.app.inject({
+      method: "POST",
+      url: "/api/auth/login",
+      payload: { username: "gdprexportuser", password: "TestPass1" },
+    });
+    const userToken = JSON.parse(loginRes.body).token;
+
+    const res = await testApp.app.inject({
+      method: "POST",
+      url: "/api/v1/enterprise/users/some-id/export",
+      headers: { authorization: `Bearer ${userToken}` },
+    });
+    // Regular users lack compliance:manage, so they get 403 before the enterprise check
+    expect(res.statusCode).toBe(403);
+  });
+});