fix: failed unit tests

This commit is contained in:
mingholy.lmh
2025-09-15 14:38:02 +08:00
parent e9b423b43a
commit 62d858f344
3 changed files with 27 additions and 15 deletions

View File

@@ -339,7 +339,7 @@ describe('AuthDialog', () => {
// Should show error message instead of calling onSelect // Should show error message instead of calling onSelect
expect(lastFrame()).toContain( expect(lastFrame()).toContain(
'You must select an auth method to proceed. Press Ctrl+C twice to exit.', 'You must select an auth method to proceed. Press Ctrl+C again to exit.',
); );
expect(onSelect).not.toHaveBeenCalled(); expect(onSelect).not.toHaveBeenCalled();
unmount(); unmount();

View File

@@ -64,6 +64,7 @@ async function createMockConfig(
getTool: vi.fn(), getTool: vi.fn(),
getFunctionDeclarations: vi.fn().mockReturnValue([]), getFunctionDeclarations: vi.fn().mockReturnValue([]),
getFunctionDeclarationsFiltered: vi.fn().mockReturnValue([]), getFunctionDeclarationsFiltered: vi.fn().mockReturnValue([]),
getAllToolNames: vi.fn().mockReturnValue([]),
...toolRegistryMocks, ...toolRegistryMocks,
} as unknown as ToolRegistry; } as unknown as ToolRegistry;
@@ -86,25 +87,36 @@ const createMockStream = (
if (response === 'stop') { if (response === 'stop') {
// When stopping, the model might return text, but the subagent logic primarily cares about the absence of functionCalls. // When stopping, the model might return text, but the subagent logic primarily cares about the absence of functionCalls.
yield { yield {
candidates: [ type: 'chunk',
{ value: {
content: { candidates: [
parts: [{ text: 'Done.' }], {
content: {
parts: [{ text: 'Done.' }],
},
}, },
}, ],
], },
}; };
} else if (response.length > 0) { } else if (response.length > 0) {
yield { functionCalls: response }; yield {
type: 'chunk',
value: {
functionCalls: response,
},
};
} else { } else {
yield { yield {
candidates: [ type: 'chunk',
{ value: {
content: { candidates: [
parts: [{ text: 'Done.' }], {
content: {
parts: [{ text: 'Done.' }],
},
}, },
}, ],
], },
}; // Handle empty array also as stop }; // Handle empty array also as stop
} }
})(); })();

View File

@@ -66,8 +66,8 @@ describe('FileSearch', () => {
expect(results).toEqual([ expect(results).toEqual([
'src/', 'src/',
'.qwenignore',
'.gitignore', '.gitignore',
'.qwenignore',
'src/not-ignored.js', 'src/not-ignored.js',
]); ]);
}); });