basic unit tests

Author: devksingh4

tests/unit/entraGroupManagement.test.ts

@@ -0,0 +1,171 @@
+import { afterAll, expect, test, beforeEach, vi } from "vitest";
+import { mockClient } from "aws-sdk-client-mock";
+import init from "../../src/api/index.js";
+import { createJwt } from "./auth.test.js";
+import supertest from "supertest";
+import { describe } from "node:test";
+import {
+  GetSecretValueCommand,
+  SecretsManagerClient,
+} from "@aws-sdk/client-secrets-manager";
+import { EntraGroupError } from "../../src/common/errors/index.js";
+
+// Mock required dependencies - their real impl's are defined in the beforeEach section.
+vi.mock("../../src/api/functions/entraId.js", () => {
+  return {
+    ...vi.importActual("../../src/api/functions/entraId.js"),
+    getEntraIdToken: vi.fn().mockImplementation(async () => {
+      return "";
+    }),
+    modifyGroup: vi.fn().mockImplementation(async () => {
+      return "";
+    }),
+    resolveEmailToOid: vi.fn().mockImplementation(async () => {
+      return "";
+    }),
+    listGroupMembers: vi.fn().mockImplementation(async () => {
+      return "";
+    }),
+  };
+});
+
+import {
+  modifyGroup,
+  listGroupMembers,
+  getEntraIdToken,
+  resolveEmailToOid,
+} from "../../src/api/functions/entraId.js";
+import { EntraGroupActions } from "../../src/common/types/iam.js";
+
+const smMock = mockClient(SecretsManagerClient);
+const app = await init();
+
+describe("Test Modify Group and List Group Routes", () => {
+  beforeEach(() => {
+    vi.resetAllMocks();
+    smMock.on(GetSecretValueCommand).resolves({
+      SecretString: JSON.stringify({ jwt_key: "test_jwt_key" }),
+    });
+  });
+
+  test("Modify group: Add and remove members", async () => {
+    const testJwt = createJwt();
+    await app.ready();
+
+    const response = await supertest(app.server)
+      .patch("/api/v1/iam/groups/test-group-id")
+      .set("authorization", `Bearer ${testJwt}`)
+      .send({
+        add: ["[email protected]"],
+        remove: ["[email protected]"],
+      });
+
+    expect(response.statusCode).toBe(202);
+    expect(modifyGroup).toHaveBeenCalledTimes(2);
+    expect(modifyGroup).toHaveBeenNthCalledWith(
+      1,
+      "ey.test.token",
+      "[email protected]",
+      "test-group-id",
+      EntraGroupActions.ADD,
+    );
+    expect(modifyGroup).toHaveBeenNthCalledWith(
+      2,
+      "ey.test.token",
+      "[email protected]",
+      "test-group-id",
+      EntraGroupActions.REMOVE,
+    );
+    expect(response.body.success).toEqual([
+      { email: "[email protected]" },
+      { email: "[email protected]" },
+    ]);
+    expect(response.body.failure).toEqual([]);
+  });
+
+  test("Modify group: Fail for invalid email domain", async () => {
+    const testJwt = createJwt();
+    await app.ready();
+
+    const response = await supertest(app.server)
+      .patch("/api/v1/iam/groups/test-group-id")
+      .set("authorization", `Bearer ${testJwt}`)
+      .send({
+        add: ["[email protected]"],
+        remove: [],
+      });
+
+    expect(response.statusCode).toBe(202);
+    expect(modifyGroup).toHaveBeenCalledTimes(1);
+    expect(response.body.success).toEqual([]);
+    expect(response.body.failure).toEqual([
+      {
+        email: "[email protected]",
+        message:
+          "User's domain must be illinois.edu to be added or removed from the group.",
+      },
+    ]);
+  });
+
+  test("List group members: Happy path", async () => {
+    const testJwt = createJwt();
+    await app.ready();
+
+    const response = await supertest(app.server)
+      .get("/api/v1/iam/groups/test-group-id")
+      .set("authorization", `Bearer ${testJwt}`);
+
+    expect(response.statusCode).toBe(200);
+    expect(listGroupMembers).toHaveBeenCalledWith(
+      "ey.test.token",
+      "test-group-id",
+    );
+    expect(response.body).toEqual([
+      { name: "John Doe", email: "[email protected]" },
+      { name: "Jane Doe", email: "[email protected]" },
+    ]);
+  });
+
+  afterAll(async () => {
+    await app.close();
+  });
+  beforeEach(() => {
+    vi.resetAllMocks();
+    vi.useFakeTimers();
+    (getEntraIdToken as any).mockImplementation(async () => {
+      return "ey.test.token";
+    });
+    (modifyGroup as any).mockImplementation(
+      async (_token, email, group, _action) => {
+        if (!email.endsWith("@illinois.edu")) {
+          throw new EntraGroupError({
+            code: 400,
+            message:
+              "User's domain must be illinois.edu to be added or removed from the group.",
+            group,
+          });
+        }
+        return true;
+      },
+    );
+    (resolveEmailToOid as any).mockImplementation(async (_token, email) => {
+      if (email === "[email protected]") {
+        throw new Error("User not found");
+      }
+      return "mocked-oid";
+    });
+    (listGroupMembers as any).mockImplementation(async (_token, group) => {
+      if (group === "nonexistent-group-id") {
+        throw new EntraGroupError({
+          code: 404,
+          message: "Group not found.",
+          group,
+        });
+      }
+      return [
+        { name: "John Doe", email: "[email protected]" },
+        { name: "Jane Doe", email: "[email protected]" },
+      ];
+    });
+  });
+});