vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-04-30 12:40:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

test(cli): cover MCP resource edge cases

Browse source
This commit is contained in:
liqoingyu 2026-01-19 11:46:48 +08:00
parent 5b2dc78897
commit 5087426af7
1 changed files with 97 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

97
packages/cli/src/ui/hooks/atCommandProcessor.test.ts
View file

@ -386,6 +386,103 @@ describe('handleAtCommand', () => {
expect(mockAddItem).not.toHaveBeenCalled();
});
it('should not expand MCP resources in untrusted folders', async () => {
(mockConfig as unknown as { getMcpServers: () => unknown }).getMcpServers =
() =>
({
github: {},
}) as unknown;
const configWithTrust = mockConfig as unknown as {
isTrustedFolder: () => boolean;
};
configWithTrust.isTrustedFolder = () => false;
const readMcpResourceSpy = vi.spyOn(registry, 'readMcpResource');
const query = 'Show me the data from @github: repos/owner/repo/issues';
const result = await handleAtCommand({
query,
config: mockConfig,
addItem: mockAddItem,
onDebugMessage: mockOnDebugMessage,
messageId: 1004,
signal: abortController.signal,
});
expect(result).toEqual({
processedQuery: null,
shouldProceed: false,
});
expect(readMcpResourceSpy).toHaveBeenCalledWith(
'github',
'github://repos/owner/repo/issues',
);
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
type: 'tool_group',
tools: [
expect.objectContaining({
status: ToolCallStatus.Error,
resultDisplay: expect.stringContaining('untrusted'),
}),
],
}),
1004,
);
});
it('should preserve trailing punctuation after an MCP resource reference', async () => {
(mockConfig as unknown as { getMcpServers: () => unknown }).getMcpServers =
() =>
({
github: {},
}) as unknown;
vi.spyOn(registry, 'readMcpResource').mockResolvedValue({
contents: [
{
uri: 'github://repos/owner/repo/issues',
mimeType: 'application/json',
text: '{"ok":true}',
},
],
} as unknown as Awaited<ReturnType<ToolRegistry['readMcpResource']>>);
const query = 'Show me the data from @github: repos/owner/repo/issues.';
const result = await handleAtCommand({
query,
config: mockConfig,
addItem: mockAddItem,
onDebugMessage: mockOnDebugMessage,
messageId: 1005,
signal: abortController.signal,
});
expect(result).toEqual({
processedQuery: [
{ text: 'Show me the data from @github:repos/owner/repo/issues.' },
{ text: '\n--- Content from referenced MCP resources ---' },
{ text: '\nContent from @github:repos/owner/repo/issues:\n' },
{ text: '{"ok":true}' },
{ text: '\n--- End of MCP resource content ---' },
],
shouldProceed: true,
});
expect(registry.readMcpResource).toHaveBeenCalledWith(
'github',
'github://repos/owner/repo/issues',
);
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
type: 'tool_group',
tools: [expect.objectContaining({ status: ToolCallStatus.Success })],
}),
1005,
);
});
it('should handle query with text before and after @command', async () => {
const fileContent = 'Markdown content.';
const filePath = await createTestFile(