From d68c42ebfdf0056c0943535329787b981bee6787 Mon Sep 17 00:00:00 2001
From: Thomas Walker <thomas.e.walker88@gmail.com>
Date: Thu, 9 Jan 2025 15:27:13 -0500
Subject: [PATCH] feat(test) Email handler test (#998)

* feat(test) Email handler test

* remove a console.log
---
 lib/lambda/processEmailsHandlers.test.ts | 134 +++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100644 lib/lambda/processEmailsHandlers.test.ts

diff --git a/lib/lambda/processEmailsHandlers.test.ts b/lib/lambda/processEmailsHandlers.test.ts
new file mode 100644
index 000000000..91017c964
--- /dev/null
+++ b/lib/lambda/processEmailsHandlers.test.ts
@@ -0,0 +1,134 @@
+import { describe, it, expect, vi } from "vitest";
+import { Context } from "aws-lambda";
+import { SESClient } from "@aws-sdk/client-ses";
+import { handler } from "./processEmails";
+import { KafkaRecord, KafkaEvent } from "shared-types";
+import { Authority } from "shared-types";
+
+const nms = "new-medicaid-submission";
+const ncs = "new-chip-submission";
+const tempExtension = "temp-extension";
+const withdrawPackage = "withdraw-package";
+const contractingInitial = "contracting-initial";
+const capitatedInitial = "capitated-initial";
+
+describe("process emails  Handler", () => {
+  it.each([
+    [`should send an email for ${nms} with ${Authority.MED_SPA}`, Authority.MED_SPA, nms],
+    [`should send an email for ${nms} with ${Authority.CHIP_SPA}`, Authority.CHIP_SPA, nms],
+    [`should send an email for ${nms} with ${Authority["1915b"]}`, Authority["1915b"], nms],
+    [`should send an email for ${nms} with ${Authority["1915c"]}`, Authority["1915c"], nms],
+    [`should send an email for ${ncs} with ${Authority.MED_SPA}`, Authority.MED_SPA, ncs],
+    [`should send an email for ${ncs} with ${Authority.CHIP_SPA}`, Authority.CHIP_SPA, ncs],
+    [`should send an email for ${ncs} with ${Authority["1915b"]}`, Authority["1915b"], ncs],
+    [`should send an email for ${ncs} with ${Authority["1915c"]}`, Authority["1915c"], ncs],
+    [
+      `should send an email for ${tempExtension} with ${Authority.MED_SPA}`,
+      Authority.MED_SPA,
+      tempExtension,
+    ],
+    [
+      `should send an email for ${tempExtension} with ${Authority.CHIP_SPA}`,
+      Authority.CHIP_SPA,
+      tempExtension,
+    ],
+    [
+      `should send an email for ${tempExtension} with ${Authority["1915b"]}`,
+      Authority["1915b"],
+      tempExtension,
+    ],
+    [
+      `should send an email for ${tempExtension} with ${Authority["1915c"]}`,
+      Authority["1915c"],
+      tempExtension,
+    ],
+    [
+      `should send an email for ${withdrawPackage} with ${Authority.MED_SPA}`,
+      Authority.MED_SPA,
+      withdrawPackage,
+    ],
+    [
+      `should send an email for ${withdrawPackage} with ${Authority.CHIP_SPA}`,
+      Authority.CHIP_SPA,
+      withdrawPackage,
+    ],
+    [
+      `should send an email for ${withdrawPackage} for ${ncs} with ${Authority["1915b"]}`,
+      Authority["1915b"],
+      withdrawPackage,
+    ],
+    [
+      `should send an email for ${withdrawPackage} with ${Authority["1915c"]}`,
+      Authority["1915c"],
+      withdrawPackage,
+    ],
+    [
+      `should send an email for ${contractingInitial} with ${Authority.MED_SPA}`,
+      Authority.MED_SPA,
+      contractingInitial,
+    ],
+    [
+      `should send an email for ${contractingInitial} with ${Authority.CHIP_SPA}`,
+      Authority.CHIP_SPA,
+      contractingInitial,
+    ],
+    [
+      `should send an email for ${contractingInitial} with ${Authority["1915b"]}`,
+      Authority["1915b"],
+      contractingInitial,
+    ],
+    [
+      `should send an email for ${contractingInitial} with ${Authority["1915c"]}`,
+      Authority["1915c"],
+      contractingInitial,
+    ],
+    [
+      `should send an email for ${capitatedInitial} with ${Authority.MED_SPA}`,
+      Authority.MED_SPA,
+      capitatedInitial,
+    ],
+    [
+      `should send an email for ${capitatedInitial} with ${Authority.CHIP_SPA}`,
+      Authority.CHIP_SPA,
+      capitatedInitial,
+    ],
+    [
+      `should send an email for ${capitatedInitial} with ${Authority["1915b"]}`,
+      Authority["1915b"],
+      capitatedInitial,
+    ],
+    [
+      `should send an email for ${capitatedInitial} with ${Authority["1915c"]}`,
+      Authority["1915c"],
+      capitatedInitial,
+    ],
+  ])("%s", async (_, auth, eventType) => {
+    const callback = vi.fn();
+    const secSPY = vi.spyOn(SESClient.prototype, "send");
+    const mockEvent: KafkaEvent = {
+      records: {
+        "mock-topic": [
+          {
+            key: Buffer.from("VA").toString("base64"),
+            value: Buffer.from(
+              JSON.stringify({
+                origin: "mako",
+                event: eventType,
+                authority: auth,
+              }),
+            ).toString("base64"),
+            headers: {},
+            timestamp: 1732645041557,
+            offset: "0",
+            partition: 0,
+            topic: "mock-topic",
+          } as unknown as KafkaRecord,
+        ],
+      },
+      eventSource: "",
+      bootstrapServers: "",
+    };
+    await handler(mockEvent, {} as Context, callback);
+    expect(secSPY).toHaveBeenCalledTimes(2);
+  });
+});