| ... | ... | @@ -0,0 +1,110 @@ |
| 1 | import { USAGE_CONTRACT_VERSION } from "@t3tools/contracts"; |
| 2 | import { mergeUsage } from "@t3tools/shared/usageMerge"; |
| 3 | import { renderToStaticMarkup } from "react-dom/server"; |
| 4 | import { beforeEach, describe, expect, it, vi } from "vite-plus/test"; |
| 5 | |
| 6 | const testState = vi.hoisted(() => ({ |
| 7 | useUsage: vi.fn(), |
| 8 | })); |
| 9 | |
| 10 | vi.mock("react", async (importOriginal) => { |
| 11 | const actual = await importOriginal<typeof import("react")>(); |
| 12 | return { |
| 13 | ...actual, |
| 14 | useState: vi.fn((initial: unknown) => [ |
| 15 | typeof initial === "function" |
| 16 | ? { |
| 17 | days: 1, |
| 18 | window: { |
| 19 | sinceDay: "2026-08-10", |
| 20 | untilDay: "2026-08-11", |
| 21 | timeZone: "UTC", |
| 22 | resolution: "hour", |
| 23 | sinceTime: "2026-08-10T12:37:00.000Z", |
| 24 | untilTime: "2026-08-11T12:37:00.000Z", |
| 25 | }, |
| 26 | } |
| 27 | : initial === "model" |
| 28 | ? "time" |
| 29 | : initial, |
| 30 | vi.fn(), |
| 31 | ]), |
| 32 | }; |
| 33 | }); |
| 34 | |
| 35 | vi.mock("../../env", () => ({ isElectron: false })); |
| 36 | vi.mock("../../state/usage", () => ({ useUsage: testState.useUsage })); |
| 37 | vi.mock("../ui/button", () => ({ Button: "button" })); |
| 38 | vi.mock("../ui/scroll-area", () => ({ ScrollArea: "div" })); |
| 39 | vi.mock("../ui/select", () => ({ |
| 40 | Select: "div", |
| 41 | SelectItem: "div", |
| 42 | SelectPopup: "div", |
| 43 | SelectTrigger: "div", |
| 44 | SelectValue: "div", |
| 45 | })); |
| 46 | vi.mock("../ui/sidebar", () => ({ SidebarInset: "div" })); |
| 47 | vi.mock("../ui/toggle-group", () => ({ Toggle: "button", ToggleGroup: "div" })); |
| 48 | vi.mock("../WorkspaceBreadcrumb", () => ({ |
| 49 | WorkspaceBreadcrumb: "div", |
| 50 | WorkspaceBreadcrumbItem: "div", |
| 51 | WorkspaceBreadcrumbSeparator: "span", |
| 52 | })); |
| 53 | vi.mock("../WorkspacePageContainer", () => ({ WorkspacePageContainer: "main" })); |
| 54 | vi.mock("../WorkspacePageHeader", () => ({ WorkspacePageHeader: "header" })); |
| 55 | vi.mock("./UsageProviderChart", () => ({ UsageProviderChart: "div" })); |
| 56 | vi.mock("./usageProviders", () => ({ |
| 57 | PROVIDER_ORDER: ["codex", "claude"], |
| 58 | PROVIDER_PRESENTATION: { |
| 59 | codex: { color: "white", label: "Codex", mark: "span" }, |
| 60 | claude: { color: "orange", label: "Claude Code", mark: "span" }, |
| 61 | }, |
| 62 | })); |
| 63 | |
| 64 | import { UsagePage } from "./UsagePage"; |
| 65 | |
| 66 | const providerTotals = (codex: number, claude: number) => |
| 67 | new Map([ |
| 68 | ["codex", { costUsd: codex, totalTokens: codex * 1_000 }], |
| 69 | ["claude", { costUsd: claude, totalTokens: claude * 1_000 }], |
| 70 | ] as const); |
| 71 | |
| 72 | beforeEach(() => { |
| 73 | testState.useUsage.mockReturnValue({ |
| 74 | merged: { |
| 75 | ...mergeUsage([], USAGE_CONTRACT_VERSION), |
| 76 | hourly: [ |
| 77 | { |
| 78 | day: "2026-08-10", |
| 79 | hourStart: "2026-08-10T13:37:00.000Z", |
| 80 | costUsd: 13, |
| 81 | totalTokens: 13_000, |
| 82 | byProvider: providerTotals(7, 6), |
| 83 | }, |
| 84 | { |
| 85 | day: "2026-08-11", |
| 86 | hourStart: "2026-08-11T11:37:00.000Z", |
| 87 | costUsd: 11, |
| 88 | totalTokens: 11_000, |
| 89 | byProvider: providerTotals(6, 5), |
| 90 | }, |
| 91 | ], |
| 92 | }, |
| 93 | environments: [], |
| 94 | isPending: false, |
| 95 | isPartial: false, |
| 96 | refresh: vi.fn(), |
| 97 | }); |
| 98 | }); |
| 99 | |
| 100 | describe("UsagePage hourly breakdown", () => { |
| 101 | it("keeps recent activity visible first without empty hourly rows", () => { |
| 102 | const markup = renderToStaticMarkup(<UsagePage />); |
| 103 | const body = markup.match(/<tbody>(.*?)<\/tbody>/)?.[1] ?? ""; |
| 104 | |
| 105 | expect(body.match(/<tr/g)).toHaveLength(2); |
| 106 | expect(body).toContain("$11.00"); |
| 107 | expect(body).toContain("$13.00"); |
| 108 | expect(body.indexOf("$11.00")).toBeLessThan(body.indexOf("$13.00")); |
| 109 | }); |
| 110 | }); |