diff --git a/client/src/components/Chat/Messages/Content/__tests__/ContentParts.integration.test.tsx b/client/src/components/Chat/Messages/Content/__tests__/ContentParts.integration.test.tsx index 555bf7c4f6..050804db0a 100644 --- a/client/src/components/Chat/Messages/Content/__tests__/ContentParts.integration.test.tsx +++ b/client/src/components/Chat/Messages/Content/__tests__/ContentParts.integration.test.tsx @@ -1,8 +1,8 @@ import React from 'react'; import { RecoilRoot } from 'recoil'; import { ContentTypes } from 'librechat-data-provider'; -import type { TAttachment, TMessageContentParts } from 'librechat-data-provider'; import { fireEvent, render, screen } from '@testing-library/react'; +import type { TAttachment, TMessageContentParts } from 'librechat-data-provider'; import ContentParts from '../ContentParts'; jest.mock('~/hooks', () => ({ @@ -20,9 +20,13 @@ jest.mock('~/hooks', () => ({ scheduleMessageContentLayoutReconcile: jest.fn(() => jest.fn()), })); -jest.mock('~/hooks/MCP', () => ({ - useMCPIconMap: () => new Map(), -})); +jest.mock('~/hooks/MCP', () => { + const mcpServerNames: string[] = []; + return { + useMCPIconMap: () => new Map(), + useMCPServerNames: () => mcpServerNames, + }; +}); jest.mock('../ToolOutput', () => ({ StackedToolIcons: () => , diff --git a/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx b/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx index 81622392a3..1e9bd72273 100644 --- a/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx +++ b/client/src/components/Chat/Messages/Content/__tests__/ToolCall.test.tsx @@ -33,9 +33,13 @@ jest.mock('~/hooks', () => ({ }), })); -jest.mock('~/hooks/MCP', () => ({ - useMCPIconMap: () => new Map(), -})); +jest.mock('~/hooks/MCP', () => { + const mcpServerNames: string[] = []; + return { + useMCPIconMap: () => new Map(), + useMCPServerNames: () => mcpServerNames, + }; +}); jest.mock('~/components/Chat/Messages/Content/MessageContent', () => ({ __esModule: true, diff --git a/client/src/components/Chat/Messages/Content/__tests__/ToolCallGroup.test.tsx b/client/src/components/Chat/Messages/Content/__tests__/ToolCallGroup.test.tsx index 10d61e4482..2328d916f5 100644 --- a/client/src/components/Chat/Messages/Content/__tests__/ToolCallGroup.test.tsx +++ b/client/src/components/Chat/Messages/Content/__tests__/ToolCallGroup.test.tsx @@ -32,9 +32,13 @@ jest.mock('~/hooks', () => ({ scheduleMessageContentLayoutReconcile: jest.fn(() => jest.fn()), })); -jest.mock('~/hooks/MCP', () => ({ - useMCPIconMap: () => new Map(), -})); +jest.mock('~/hooks/MCP', () => { + const mcpServerNames: string[] = []; + return { + useMCPIconMap: () => new Map(), + useMCPServerNames: () => mcpServerNames, + }; +}); jest.mock('../ToolOutput', () => ({ StackedToolIcons: ({ toolNames }: { toolNames: string[] }) => (