move sqs unit tests

Author: devksingh4

package.json

@@ -17,7 +17,7 @@
     "lint": "yarn workspaces run lint",
     "prepare": "node .husky/install.mjs || true",
     "typecheck": "yarn workspaces run typecheck",
-    "test:unit": "vitest run tests/unit --config tests/unit/vitest.config.ts && yarn workspace infra-core-ui run test:unit",
+    "test:unit": "cross-env RunEnvironment='dev' vitest run tests/unit --config tests/unit/vitest.config.ts && yarn workspace infra-core-ui run test:unit",
     "test:unit-ui": "yarn test:unit --ui",
     "test:unit-watch": "vitest tests/unit",
     "test:live": "vitest tests/live",

src/api/routes/mobileWallet.ts

@@ -6,9 +6,12 @@ import {
 } from "../../common/errors/index.js";
 import { z } from "zod";
 import { checkPaidMembership } from "../functions/membership.js";
-import { AvailableSQSFunctions, SQSPayload } from "common/types/sqsMessage.js";
+import {
+  AvailableSQSFunctions,
+  SQSPayload,
+} from "../../common/types/sqsMessage.js";
 import { SendMessageCommand, SQSClient } from "@aws-sdk/client-sqs";
-import { genericConfig } from "common/config.js";
+import { genericConfig } from "../../common/config.js";
 import { zodToJsonSchema } from "zod-to-json-schema";
 
 const queuedResponseJsonSchema = zodToJsonSchema(

src/api/sqs/handlers.ts

@@ -4,12 +4,15 @@ import {
   runEnvironment,
   SQSHandlerFunction,
 } from "./index.js";
-import { getEntraIdToken, getUserProfile } from "api/functions/entraId.js";
+import {
+  getEntraIdToken,
+  getUserProfile,
+} from "../../api/functions/entraId.js";
 import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
-import { environmentConfig, genericConfig } from "common/config.js";
+import { environmentConfig, genericConfig } from "../../common/config.js";
 import { SecretsManagerClient } from "@aws-sdk/client-secrets-manager";
-import { issueAppleWalletMembershipCard } from "api/functions/mobileWallet.js";
-import { generateMembershipEmailCommand } from "api/functions/ses.js";
+import { issueAppleWalletMembershipCard } from "../../api/functions/mobileWallet.js";
+import { generateMembershipEmailCommand } from "../../api/functions/ses.js";
 import { SESClient } from "@aws-sdk/client-ses";
 
 export const emailMembershipPassHandler: SQSHandlerFunction<

src/api/sqs/index.ts

@@ -13,12 +13,12 @@ import { logger } from "./logger.js";
 import { z, ZodError } from "zod";
 import pino from "pino";
 import { emailMembershipPassHandler, pingHandler } from "./handlers.js";
-import { ValidationError } from "common/errors/index.js";
+import { ValidationError } from "../../common/errors/index.js";
 import { SecretsManagerClient } from "@aws-sdk/client-secrets-manager";
 import { SESClient } from "@aws-sdk/client-ses";
 import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
-import { RunEnvironment } from "common/roles.js";
-import { environmentConfig } from "common/config.js";
+import { RunEnvironment } from "../../common/roles.js";
+import { environmentConfig } from "../../common/config.js";
 
 export type SQSFunctionPayloadTypes = {
   [K in keyof typeof sqsPayloadSchemas]: SQSHandlerFunction<K>;

tests/unit/mobileWallet.test.ts

@@ -1,12 +1,6 @@
-import { afterAll, expect, test, beforeEach, vi } from "vitest";
+import { afterAll, expect, test, beforeEach, vi, describe } from "vitest";
 import init from "../../src/api/index.js";
-import { describe } from "node:test";
 import { EntraFetchError } from "../../src/common/errors/index.js";
-import { mockClient } from "aws-sdk-client-mock";
-import { issueAppleWalletMembershipCard } from "../../src/api/functions/mobileWallet.js";
-import { SendRawEmailCommand, SESClient } from "@aws-sdk/client-ses";
-
-const sesMock = mockClient(SESClient);
 
 vi.mock("../../src/api/functions/membership.js", () => {
   return {
@@ -70,13 +64,11 @@ describe("Mobile wallet pass issuance", async () => {
     await response.json();
   });
   test("Test that passes will be issued for members", async () => {
-    sesMock.on(SendRawEmailCommand).resolvesOnce({}).rejects();
     const response = await app.inject({
       method: "POST",
       url: "/api/v1/mobileWallet/[email protected]",
     });
     expect(response.statusCode).toBe(202);
-    expect(issueAppleWalletMembershipCard).toHaveBeenCalledOnce();
   });
   afterAll(async () => {
     await app.close();

tests/unit/sqs/emailMembershipPass.test.ts

@@ -0,0 +1,50 @@
+import { expect, test, beforeEach, vi } from "vitest";
+vi.mock("../../../src/api/functions/mobileWallet.js", () => ({
+  issueAppleWalletMembershipCard: vi.fn().mockResolvedValue(new ArrayBuffer()),
+}))
+
+vi.mock("../../../src/api/functions/entraId.js", () => ({
+  getEntraIdToken: vi.fn().mockResolvedValue("atokenofalltime"),
+  getUserProfile: vi.fn().mockImplementation(async (_token: string, email: string) => {
+    console.log('getUserProfile called with:', email);
+    if (email === "[email protected]") {
+      return { displayName: "John Doe" };
+    }
+    throw new EntraFetchError({
+      message: "User not found",
+      email,
+    });
+  }),
+  resolveEmailToOid: vi.fn().mockResolvedValue("12345"),
+}));
+
+vi.mock("../../api/functions/mobileWallet.js", () => ({
+  issueAppleWalletMembershipCard: vi.fn().mockResolvedValue(new ArrayBuffer(0)),
+}));
+
+vi.stubEnv("RunEnvironment", 'dev');
+
+import { describe } from "node:test";
+import { emailMembershipPassHandler } from "../../../src/api/sqs/handlers.js";
+import { EntraFetchError } from "../../../src/common/errors/index.js";
+import { getEntraIdToken, getUserProfile } from "../../../src/api/functions/entraId.js";
+import { issueAppleWalletMembershipCard } from "../../../src/api/functions/mobileWallet.js";
+
+describe("Mobile wallet pass issuance (SQS function)", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  test("Test that passes will be issued for members", async () => {
+    const email = '[email protected]';
+    const logger = { info: vi.fn() } as any;
+    await emailMembershipPassHandler(
+      { email },
+      { reqId: '1', initiator: '1' },
+      logger
+    );
+    expect(getEntraIdToken).toHaveBeenCalled();
+    expect(issueAppleWalletMembershipCard).toHaveBeenCalledOnce();
+    expect(getUserProfile).toHaveBeenCalledWith("atokenofalltime", email);
+  });
+});