fix unit tests

Author: devksingh4

tests/unit/entraGroupManagement.test.ts

@@ -69,13 +69,16 @@ describe("Test Modify Group and List Group Routes", () => {
       "[email protected]",
       "test-group-id",
       EntraGroupActions.ADD,
+      expect.any(Object), // Matches any object
     );
+
     expect(modifyGroup).toHaveBeenNthCalledWith(
       2,
       "ey.test.token",
       "[email protected]",
       "test-group-id",
       EntraGroupActions.REMOVE,
+      expect.any(Object), // Matches any object
     );
     expect(response.body.success).toEqual([
       { email: "[email protected]" },

tests/unit/mobileWallet.test.ts

@@ -13,19 +13,6 @@ import { v4 as uuidv4 } from "uuid";
 const smMock = mockClient(SecretsManagerClient);
 const sqsMock = mockClient(SQSClient);
 
-vi.mock("../../src/api/functions/membership.js", () => {
-  return {
-    checkPaidMembership: vi.fn(
-      (_endpoint: string, _log: any, netId: string) => {
-        if (netId === "valid") {
-          return true;
-        }
-        return false;
-      },
-    ),
-  };
-});
-
 vi.mock("../../src/api/functions/entraId.js", () => {
   return {
     getEntraIdToken: vi.fn().mockImplementation(async () => {

tests/unit/vitest.setup.ts

@@ -44,3 +44,21 @@ vi.mock(
     };
   },
 );
+
+vi.mock(
+  import("../../src/api/functions/membership.js"),
+  async (importOriginal) => {
+    const mod = await importOriginal();
+    return {
+      ...mod,
+      checkPaidMembershipFromTable: vi.fn(async (netId, _dynamoClient) => {
+        switch (netId) {
+          case "valid":
+            return true;
+          default:
+            return false;
+        }
+      }),
+    };
+  },
+);