From ce93b79a99979aa9defd1b3538f85742b2161f68 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dafydd=20Ll=C5=B7r=20Pearson?= <DafyddLlyr@gmail.com>
Date: Wed, 27 Nov 2024 11:24:20 +0000
Subject: [PATCH] test: Hook

---
 .../FeeBreakdown/useFeeBreakdown.test.ts      | 116 ++++++++++++++++++
 1 file changed, 116 insertions(+)
 create mode 100644 editor.planx.uk/src/@planx/components/Pay/Public/FeeBreakdown/useFeeBreakdown.test.ts

diff --git a/editor.planx.uk/src/@planx/components/Pay/Public/FeeBreakdown/useFeeBreakdown.test.ts b/editor.planx.uk/src/@planx/components/Pay/Public/FeeBreakdown/useFeeBreakdown.test.ts
new file mode 100644
index 0000000000..4e4c042d5a
--- /dev/null
+++ b/editor.planx.uk/src/@planx/components/Pay/Public/FeeBreakdown/useFeeBreakdown.test.ts
@@ -0,0 +1,116 @@
+import { logger } from "airbrake";
+import { useStore } from "pages/FlowEditor/lib/store";
+import { vi } from "vitest";
+
+import { useFeeBreakdown } from "./useFeeBreakdown";
+
+vi.mock("pages/FlowEditor/lib/store", () => ({
+  useStore: vi.fn(),
+}));
+
+vi.mock("airbrake", () => ({
+  logger: {
+    notify: vi.fn(),
+  },
+}));
+
+describe("useFeeBreakdown() hook", () => {
+  describe("valid data", () => {
+    it("returns a fee breakdown for number inputs", () => {
+      const mockPassportData = {
+        "application.fee.calculated": 1000,
+        "application.fee.payable": 800,
+        "application.fee.payable.vat": 160,
+      };
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, "test-session"]);
+
+      const result = useFeeBreakdown();
+
+      expect(result).toEqual({
+        applicationFee: 1000,
+        total: 800,
+        reduction: 200,
+        vat: 160,
+      });
+    });
+
+    it("returns a fee breakdown for number tuple inputs", () => {
+      const mockPassportData = {
+        "application.fee.calculated": [1000],
+        "application.fee.payable": [800],
+        "application.fee.payable.vat": [160],
+      };
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, "test-session"]);
+
+      const result = useFeeBreakdown();
+
+      expect(result).toEqual({
+        applicationFee: 1000,
+        total: 800,
+        reduction: 200,
+        vat: 160,
+      });
+    });
+  });
+
+  describe("invalid inputs", () => {
+    it("returns undefined for missing data", () => {
+      const mockPassportData = {};
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, "test-session"]);
+
+      const result = useFeeBreakdown();
+
+      expect(result).toBeUndefined();
+    });
+
+    it("returns undefined for partial data", () => {
+      const mockPassportData = {
+        "application.fee.calculated": [1000],
+        "application.fee.payable.vat": [160],
+      };
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, "test-session"]);
+
+      const result = useFeeBreakdown();
+
+      expect(result).toBeUndefined();
+    });
+
+    it("returns undefined for incorrect data", () => {
+      const mockPassportData = {
+        "application.fee.calculated": "some string",
+        "application.fee.payable": [800, 700],
+        "application.fee.payable.vat": false,
+      };
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, "test-session"]);
+
+      const result = useFeeBreakdown();
+
+      expect(result).toBeUndefined();
+    });
+
+    it("calls Airbrake if invalid inputs are provided", () => {
+      const mockPassportData = {};
+      const mockSessionId = "test-session";
+
+      vi.mocked(useStore).mockReturnValue([mockPassportData, mockSessionId]);
+      const loggerSpy = vi.spyOn(logger, "notify");
+
+      const result = useFeeBreakdown();
+
+      expect(result).toBeUndefined();
+
+      expect(loggerSpy).toHaveBeenCalledWith(
+        expect.stringContaining(mockSessionId),
+      );
+
+      expect(loggerSpy).toHaveBeenCalledWith(
+        expect.stringContaining("ZodError"),
+      );
+    });
+  });
+});