File size: 3,202 Bytes
fd2c364 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | import { renderHook, waitFor } from "@testing-library/react";
import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { describe, expect, it, vi, beforeEach } from "vitest";
import React from "react";
import SettingsService from "#/api/settings-service/settings-service.api";
const trackMcpConfigUpdatedMock = vi.fn();
vi.mock("#/hooks/use-tracking", () => ({
useTracking: () => ({
trackMcpConfigUpdated: trackMcpConfigUpdatedMock,
}),
}));
const useSettingsMock = vi.fn();
vi.mock("#/hooks/query/use-settings", () => ({
useSettings: () => useSettingsMock(),
}));
import { useSaveSettings } from "#/hooks/mutation/use-save-settings";
const createWrapper = () => {
const queryClient = new QueryClient({
defaultOptions: { queries: { retry: false }, mutations: { retry: false } },
});
return ({ children }: { children: React.ReactNode }) =>
React.createElement(QueryClientProvider, { client: queryClient }, children);
};
describe("useSaveSettings - MCP tracking", () => {
beforeEach(() => {
vi.clearAllMocks();
vi.spyOn(SettingsService, "saveSettings").mockResolvedValue(true);
// Default: no current mcp_config so any incoming config is treated as new.
useSettingsMock.mockReturnValue({ data: {} });
});
it("calls trackMcpConfigUpdated with server counts when mcp_config changes", async () => {
const { result } = renderHook(() => useSaveSettings(), {
wrapper: createWrapper(),
});
await result.current.mutateAsync({
mcp_config: {
sse1: { transport: "sse", url: "http://sse1" },
sse2: { transport: "sse", url: "http://sse2" },
stdio1: { transport: "stdio", command: "cmd", args: [] },
},
});
await waitFor(() => {
expect(trackMcpConfigUpdatedMock).toHaveBeenCalledWith({
sseServersCount: 2,
stdioServersCount: 1,
});
});
});
it("does not call trackMcpConfigUpdated when mcp_config is absent in the update", async () => {
const { result } = renderHook(() => useSaveSettings(), {
wrapper: createWrapper(),
});
await result.current.mutateAsync({ user_consents_to_analytics: true });
expect(trackMcpConfigUpdatedMock).not.toHaveBeenCalled();
});
it("does not call trackMcpConfigUpdated when mcp_config reference is unchanged", async () => {
const sharedConfig = {};
useSettingsMock.mockReturnValue({
data: { mcp_config: sharedConfig },
});
const { result } = renderHook(() => useSaveSettings(), {
wrapper: createWrapper(),
});
// Passing the same object reference — should not trigger tracking.
await result.current.mutateAsync({ mcp_config: sharedConfig });
expect(trackMcpConfigUpdatedMock).not.toHaveBeenCalled();
});
it("counts zero servers correctly when server arrays are empty", async () => {
const { result } = renderHook(() => useSaveSettings(), {
wrapper: createWrapper(),
});
await result.current.mutateAsync({
mcp_config: {},
});
await waitFor(() => {
expect(trackMcpConfigUpdatedMock).toHaveBeenCalledWith({
sseServersCount: 0,
stdioServersCount: 0,
});
});
});
});
|