diff --git a/App/components/po/po-line-items-editor.tsx b/App/components/po/po-line-items-editor.tsx index 65f3c93..403ad7a 100644 --- a/App/components/po/po-line-items-editor.tsx +++ b/App/components/po/po-line-items-editor.tsx @@ -20,8 +20,11 @@ const UOM_OPTIONS = [ { value: "mL", label: "mL — Millilitre" }, { value: "m", label: "m — Metre" }, { value: "m2", label: "m² — Sq. Metre" }, - { value: "hr", label: "hr — Hour" }, - { value: "day", label: "day — Day" }, + { value: "hr", label: "hr — Hour" }, + { value: "day", label: "day — Day" }, + { value: "week", label: "week — Week" }, + { value: "month", label: "month — Month" }, + { value: "year", label: "year — Year" }, { value: "lump", label: "lump — Lump Sum" }, { value: "Ltr", label: "Ltr — Litre (alt)" }, ]; diff --git a/App/tests/unit/po-line-items-editor.test.tsx b/App/tests/unit/po-line-items-editor.test.tsx index 1d24a93..53eeb19 100644 --- a/App/tests/unit/po-line-items-editor.test.tsx +++ b/App/tests/unit/po-line-items-editor.test.tsx @@ -93,6 +93,25 @@ describe("LineItemsEditor — edit mode", () => { const lastCall = onChange.mock.calls[onChange.mock.calls.length - 1][0] as LineItemInput[]; expect(lastCall[0].gstRate).toBeCloseTo(0.05); }); + + it("offers month and year as unit-of-measure options", () => { + render(); + const selects = screen.getAllByRole("combobox") as HTMLSelectElement[]; + const unitSelect = selects.find((s) => s.value === "pc")!; + const values = Array.from(unitSelect.options).map((o) => o.value); + expect(values).toContain("month"); + expect(values).toContain("year"); + }); + + it("calls onChange with the selected duration unit", async () => { + const onChange = vi.fn(); + render(); + const selects = screen.getAllByRole("combobox") as HTMLSelectElement[]; + const unitSelect = selects.find((s) => s.value === "pc")!; + fireEvent.change(unitSelect, { target: { value: "year" } }); + const lastCall = onChange.mock.calls[onChange.mock.calls.length - 1][0] as LineItemInput[]; + expect(lastCall[0].unit).toBe("year"); + }); }); // ── Totals calculation (edit mode) ────────────────────────────────────────────