fix(usage): treat usage_script credentials as explicit overrides (#4654)

* fix(usage): treat usage_script credentials as explicit overrides * fix: treat usage script credentials as explicit overrides
2026-07-24 12:44:18 +08:00 · 2026-06-28 17:37:58 +08:00
parent eb847642d0
commit d1f6c74be8
6 changed files with 711 additions and 17 deletions
@@ -0,0 +1,127 @@
+import type { ReactNode } from "react";
+import { act, renderHook } from "@testing-library/react";
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { useUpdateProviderMutation } from "@/lib/query/mutations";
+import { usageKeys } from "@/lib/query/usage";
+import type { Provider } from "@/types";
+
+const apiMocks = vi.hoisted(() => ({
+  update: vi.fn(),
+}));
+
+vi.mock("@/lib/api", () => ({
+  providersApi: {
+    update: (...args: unknown[]) => apiMocks.update(...args),
+  },
+  sessionsApi: {},
+  settingsApi: {},
+}));
+
+vi.mock("@/hooks/useHermes", () => ({
+  invalidateHermesProviderCaches: vi.fn(),
+}));
+
+vi.mock("@/hooks/useOpenClaw", () => ({
+  openclawKeys: {
+    health: ["openclaw", "health"],
+  },
+}));
+
+vi.mock("react-i18next", () => ({
+  useTranslation: () => ({
+    t: (_key: string, options?: { defaultValue?: string }) =>
+      options?.defaultValue ?? _key,
+  }),
+}));
+
+vi.mock("sonner", () => ({
+  toast: {
+    success: vi.fn(),
+    error: vi.fn(),
+  },
+}));
+
+function createWrapper() {
+  const queryClient = new QueryClient({
+    defaultOptions: {
+      queries: { retry: false },
+      mutations: { retry: false },
+    },
+  });
+  const invalidateSpy = vi.spyOn(queryClient, "invalidateQueries");
+
+  const wrapper = ({ children }: { children: ReactNode }) => (
+    <QueryClientProvider client={queryClient}>{children}</QueryClientProvider>
+  );
+
+  return { wrapper, invalidateSpy };
+}
+
+function createProvider(overrides: Partial<Provider> = {}): Provider {
+  return {
+    id: "provider-1",
+    name: "Test Provider",
+    settingsConfig: {},
+    ...overrides,
+  };
+}
+
+beforeEach(() => {
+  apiMocks.update.mockReset().mockResolvedValue(true);
+});
+
+describe("useUpdateProviderMutation", () => {
+  it("invalidates the updated provider usage query", async () => {
+    const { wrapper, invalidateSpy } = createWrapper();
+    const provider = createProvider({ id: "provider-b" });
+    const { result } = renderHook(() => useUpdateProviderMutation("codex"), {
+      wrapper,
+    });
+
+    await act(async () => {
+      await result.current.mutateAsync({ provider });
+    });
+
+    expect(apiMocks.update).toHaveBeenCalledWith(provider, "codex", undefined);
+    expect(invalidateSpy).toHaveBeenCalledWith({
+      queryKey: ["providers", "codex"],
+    });
+    expect(invalidateSpy).toHaveBeenCalledWith({
+      queryKey: usageKeys.script("provider-b", "codex"),
+    });
+    expect(invalidateSpy).not.toHaveBeenCalledWith({
+      queryKey: usageKeys.all,
+    });
+  });
+
+  it("also invalidates the previous usage query when provider id changes", async () => {
+    const { wrapper, invalidateSpy } = createWrapper();
+    const provider = createProvider({ id: "provider-new" });
+    const { result } = renderHook(() => useUpdateProviderMutation("openclaw"), {
+      wrapper,
+    });
+
+    await act(async () => {
+      await result.current.mutateAsync({
+        provider,
+        originalId: "provider-old",
+      });
+    });
+
+    expect(apiMocks.update).toHaveBeenCalledWith(
+      provider,
+      "openclaw",
+      "provider-old",
+    );
+    expect(invalidateSpy).toHaveBeenCalledWith({
+      queryKey: usageKeys.script("provider-new", "openclaw"),
+    });
+    expect(invalidateSpy).toHaveBeenCalledWith({
+      queryKey: usageKeys.script("provider-old", "openclaw"),
+    });
+    expect(invalidateSpy).not.toHaveBeenCalledWith({
+      queryKey: usageKeys.all,
+    });
+  });
+});