File size: 7,561 Bytes
097fb32 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 | /**
* 端到端测试：向真实 Cursor2API 服务发送请求
*
* 测试场景：
* 1. 简单请求能正常返回
* 2. 带工具的多轮长对话触发压缩
* 3. 验证 stop_reason 正确
*/
const API_URL = 'http://localhost:3010/v1/messages';
interface TestResult {
name: string;
passed: boolean;
detail: string;
}
const results: TestResult[] = [];
function assert(name: string, condition: boolean, detail = '') {
results.push({ name, passed: condition, detail });
console.log(condition ? ` ✅ ${name}` : ` ❌ ${name}: ${detail}`);
}
// 构造一个模拟 Claude Code 的长对话请求（带很多轮工具交互历史）
function buildLongToolRequest(turnCount: number) {
const messages: any[] = [];
// 模拟多轮工具交互历史
for (let i = 0; i < turnCount; i++) {
if (i === 0) {
// 第一轮：用户发起请求
messages.push({
role: 'user',
content: 'Help me analyze the project structure. Read the main entry file first.'
});
} else {
// 工具结果
messages.push({
role: 'user',
content: [
{
type: 'tool_result',
tool_use_id: `tool_${i}`,
content: `File content of module${i}.ts:\n` +
`import { something } from './utils';\n\n` +
`export class Module${i} {\n` +
Array.from({length: 30}, (_, j) => ` method${j}() { return ${j}; }`).join('\n') +
`\n}\n`
}
]
});
}
// 助手的工具调用
messages.push({
role: 'assistant',
content: [
{ type: 'text', text: `Let me check module${i + 1}.` },
{
type: 'tool_use',
id: `tool_${i + 1}`,
name: 'Read',
input: { file_path: `src/module${i + 1}.ts` }
}
]
});
}
// 最后一轮工具结果
messages.push({
role: 'user',
content: [
{
type: 'tool_result',
tool_use_id: `tool_${turnCount}`,
content: 'File not found: src/module' + turnCount + '.ts'
}
]
});
return {
model: 'claude-sonnet-4-20250514',
max_tokens: 4096,
stream: false,
system: 'You are a helpful coding assistant.',
tools: [
{
name: 'Read',
description: 'Read a file from disk',
input_schema: {
type: 'object',
properties: {
file_path: { type: 'string', description: 'Path to the file' }
},
required: ['file_path']
}
},
{
name: 'Bash',
description: 'Execute a shell command',
input_schema: {
type: 'object',
properties: {
command: { type: 'string', description: 'The command to execute' }
},
required: ['command']
}
}
],
messages
};
}
async function runTests() {
console.log('\n=== 测试 1：基本请求 ===');
try {
const resp = await fetch(API_URL, {
method: 'POST',
headers: { 'Content-Type': 'application/json', 'x-api-key': 'test' },
body: JSON.stringify({
model: 'claude-sonnet-4-20250514',
max_tokens: 1024,
stream: false,
messages: [{ role: 'user', content: 'Say "hello" in one word.' }]
})
});
assert('服务器响应', resp.ok, `status=${resp.status}`);
const data = await resp.json();
assert('返回 message 类型', data.type === 'message', `type=${data.type}`);
assert('stop_reason 是 end_turn', data.stop_reason === 'end_turn', `stop_reason=${data.stop_reason}`);
assert('有 content', data.content?.length > 0, `content=${JSON.stringify(data.content)}`);
console.log(` 📝 响应: ${data.content?.[0]?.text?.substring(0, 100)}`);
} catch (e: any) {
assert('基本请求', false, e.message);
}
console.log('\n=== 测试 2：长对话工具请求（触发压缩）===');
try {
const longReq = buildLongToolRequest(18); // 18 轮 → 37 条消息
console.log(` 📊 发送 ${longReq.messages.length} 条消息...`);
const resp = await fetch(API_URL, {
method: 'POST',
headers: { 'Content-Type': 'application/json', 'x-api-key': 'test' },
body: JSON.stringify(longReq)
});
assert('长对话服务器响应', resp.ok, `status=${resp.status}`);
const data = await resp.json();
assert('长对话返回 message', data.type === 'message', `type=${data.type}`);
assert('长对话有 content', data.content?.length > 0);
// 检查 stop_reason
const validStops = ['end_turn', 'tool_use', 'max_tokens'];
assert('stop_reason 合法', validStops.includes(data.stop_reason), `stop_reason=${data.stop_reason}`);
console.log(` 📝 stop_reason: ${data.stop_reason}`);
console.log(` 📝 content blocks: ${data.content?.length}`);
if (data.content?.[0]?.text) {
console.log(` 📝 响应片段: ${data.content[0].text.substring(0, 150)}...`);
}
} catch (e: any) {
assert('长对话请求', false, e.message);
}
console.log('\n=== 测试 3：流式请求 ===');
try {
const resp = await fetch(API_URL, {
method: 'POST',
headers: { 'Content-Type': 'application/json', 'x-api-key': 'test' },
body: JSON.stringify({
model: 'claude-sonnet-4-20250514',
max_tokens: 1024,
stream: true,
messages: [{ role: 'user', content: 'Say "world" in one word.' }]
})
});
assert('流式响应 200', resp.ok, `status=${resp.status}`);
assert('Content-Type 是 SSE', resp.headers.get('content-type')?.includes('text/event-stream') ?? false);
const body = await resp.text();
const events = body.split('\n').filter(l => l.startsWith('event:'));
assert('有 SSE 事件', events.length > 0, `events=${events.length}`);
assert('包含 message_start', body.includes('message_start'));
assert('包含 message_stop', body.includes('message_stop'));
// 检查 stop_reason
const deltaMatch = body.match(/"stop_reason"\s*:\s*"([^"]+)"/);
if (deltaMatch) {
assert('流式 stop_reason 合法', ['end_turn', 'tool_use', 'max_tokens'].includes(deltaMatch[1]), `stop_reason=${deltaMatch[1]}`);
}
console.log(` 📝 SSE 事件数: ${events.length}`);
} catch (e: any) {
assert('流式请求', false, e.message);
}
// 总结
const passed = results.filter(r => r.passed).length;
const failed = results.filter(r => !r.passed).length;
console.log(`\n=== 端到端结果: ${passed} 通过, ${failed} 失败 ===\n`);
}
runTests().catch(console.error);
|