diff --git a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.activity.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.activity.test.tsx index ca7adf07941..4a8af9eff22 100644 --- a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.activity.test.tsx +++ b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.activity.test.tsx @@ -28,6 +28,12 @@ vi.mock("@/app/(dashboard)/router-settings/_components/general_settings", () => vi.mock("./PromptCompressionTab", () => ({ __esModule: true, default: () =>
})); +vi.mock("@/app/(dashboard)/hooks/useAuthorized", () => ({ + default: () => ({ accessToken: "test-token", userId: "u1", userRole: "proxy_admin" }), +})); + +vi.mock("@/app/(dashboard)/hooks/useIsOrgAdmin", () => ({ default: () => false })); + import CostOptimizationView from "./CostOptimizationView"; const singlePage = { diff --git a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.test.tsx index c6d5a410418..5285495efa4 100644 --- a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.test.tsx +++ b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/CostOptimizationView.test.tsx @@ -7,6 +7,8 @@ vi.mock("@/app/(dashboard)/hooks/useAuthorized", () => ({ default: useAuthorizedMock, })); +vi.mock("@/app/(dashboard)/hooks/useIsOrgAdmin", () => ({ default: () => false })); + vi.mock("./UsageTab", () => ({ __esModule: true, default: () => })); vi.mock("./PromptCompressionTab", () => ({ __esModule: true, default: () => })); vi.mock("./PromptCachingTab", () => ({ __esModule: true, default: () => }));