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
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();
unmount();

View File

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

View File

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