vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-07-09 17:19:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

test(e2e): stabilize plan mode tool-control assertion (#6176)

Browse source
This commit is contained in:
易良 2026-07-02 20:07:11 +08:00 • committed by GitHub
parent 250ead34d7
commit 75645caf47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 40 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

45
integration-tests/sdk-typescript/tool-control.test.ts
View file

@ -19,6 +19,7 @@ import {
type SDKMessage,
type SDKUserMessage,
} from '@qwen-code/sdk';
import { fakeToolCall, startFakeOpenAIServer } from '../fake-openai-server.js';
import {
SDKTestHelper,
extractText,
@ -1030,12 +1031,35 @@ describe('Tool Control Parameters (E2E)', () => {
await helper.createFile('test.txt', 'original');
const canUseToolCalls: string[] = [];
const fakeServer = await startFakeOpenAIServer(({ requestIndex }) => {
if (requestIndex === 0) {
return {
toolCalls: [
fakeToolCall('read_file', {
file_path: helper.getPath('test.txt'),
}),
],
};
}
return { content: 'Plan: leave the file unchanged.' };
});
const q = query({
prompt: 'Read test.txt and write "modified" to it.',
options: {
...SHARED_TEST_OPTIONS,
cwd: testDir,
model: 'fake-model',
env: {
NO_PROXY: '127.0.0.1,localhost',
no_proxy: '127.0.0.1,localhost',
OPENAI_API_KEY: 'fake-key',
OPENAI_BASE_URL: fakeServer.baseUrl,
OPENAI_MODEL: 'fake-model',
QWEN_MODEL: 'fake-model',
},
authType: 'openai',
permissionMode: 'plan',
// allowedTools should be overridden by plan mode
allowedTools: ['write_file'],
@ -1057,14 +1081,25 @@ describe('Tool Control Parameters (E2E)', () => {
const toolCalls = findToolCalls(messages);
const toolNames = toolCalls.map((tc) => tc.toolUse.name);
// Should be able to read
expect(toolNames).toContain('read_file');
assertSuccessfulCompletion(messages);
// write_file should NOT be called in plan mode
// (plan mode blocks all write operations)
// The AI should respond with a plan instead
// read_file should be allowed in plan mode.
expect(toolNames).toContain('read_file');
const readFileResults = findToolResults(messages, 'read_file');
expect(readFileResults.length).toBeGreaterThan(0);
for (const result of readFileResults) {
expect(result.isError).toBe(false);
expect(result.content).toContain('original');
}
// write_file should NOT be called in plan mode.
// The fake model responds with a plan after reading the file.
expect(toolNames).not.toContain('write_file');
expect(canUseToolCalls.length).toBe(0);
expect(await helper.readFile('test.txt')).toBe('original');
} finally {
await q.close();
await fakeServer.close();
}
},
TEST_TIMEOUT,