mirror of
https://github.com/catlog22/Claude-Code-Workflow.git
synced 2026-02-04 01:40:45 +08:00
c6093ef741
- Implemented CliCommandNode component for executing CLI tools with AI models. - Implemented PromptNode component for constructing AI prompts with context. - Added styling for mode and tool badges in both components. - Enhanced user experience with command and argument previews, execution status, and error handling. test: add comprehensive tests for ask_question tool - Created direct test for ask_question tool execution. - Developed end-to-end tests to validate ask_question tool integration with WebSocket and A2UI surfaces. - Implemented simple and integrated WebSocket tests to ensure proper message handling and surface reception. - Added tool registration test to verify ask_question tool is correctly registered. chore: add WebSocket listener and simulation tests - Added WebSocket listener for A2UI surfaces to facilitate testing. - Implemented frontend simulation test to validate complete flow from backend to frontend. - Created various test scripts to ensure robust testing of ask_question tool functionality.
83 lines
2.6 KiB
JavaScript
83 lines
2.6 KiB
JavaScript
/**
|
|
* Integrated WebSocket test for ask_question tool
|
|
* This test runs a WebSocket client that connects to the running CCW server
|
|
* and listens for A2UI surfaces
|
|
*/
|
|
|
|
import { WebSocket } from 'ws';
|
|
|
|
console.log('=== Integrated WebSocket A2UI Test ===\n');
|
|
|
|
// Step 1: Connect to WebSocket
|
|
console.log('1. Connecting to ws://127.0.0.1:3456/ws...');
|
|
|
|
const ws = new WebSocket('ws://127.0.0.1:3456/ws');
|
|
|
|
const connected = new Promise((resolve, reject) => {
|
|
ws.on('open', () => {
|
|
console.log(' ✓ WebSocket connected!\n');
|
|
resolve(true);
|
|
});
|
|
ws.on('error', (err) => {
|
|
console.log(' ✗ Connection error:', err.message);
|
|
reject(err);
|
|
});
|
|
setTimeout(() => reject(new Error('Connection timeout')), 5000);
|
|
});
|
|
|
|
try {
|
|
await connected;
|
|
|
|
// Step 2: Listen for messages
|
|
console.log('2. Listening for A2UI surface messages...\n');
|
|
console.log(' NOTE: The CCW server running on port 3456 has its own');
|
|
console.log(' wsClients set. This WebSocket IS connected to that server.');
|
|
console.log(' A2UI surfaces will be received when tool is called via');
|
|
console.log(' the same server process.\n');
|
|
|
|
let messageCount = 0;
|
|
|
|
ws.on('message', (data) => {
|
|
messageCount++;
|
|
try {
|
|
const msg = JSON.parse(data.toString());
|
|
console.log(`\n 📨 Message ${messageCount}:`);
|
|
console.log(` Type: ${msg.type}`);
|
|
if (msg.type === 'a2ui-surface') {
|
|
console.log(` Surface ID: ${msg.surfaceUpdate?.surfaceId}`);
|
|
console.log(` Question: ${msg.surfaceUpdate?.initialState?.questionId}`);
|
|
console.log(' ✅ A2UI SURFACE RECEIVED!');
|
|
|
|
// Send answer back
|
|
setTimeout(() => {
|
|
console.log('\n Sending answer...');
|
|
ws.send(JSON.stringify({
|
|
type: 'a2ui-answer',
|
|
questionId: msg.surfaceUpdate?.initialState?.questionId,
|
|
surfaceId: msg.surfaceUpdate?.surfaceId,
|
|
value: true,
|
|
cancelled: false,
|
|
timestamp: new Date().toISOString()
|
|
}));
|
|
console.log(' ✓ Answer sent!\n');
|
|
}, 500);
|
|
}
|
|
} catch (e) {
|
|
console.log(` Raw: ${data.toString().substring(0, 100)}`);
|
|
}
|
|
});
|
|
|
|
// Keep connection open for 30 seconds
|
|
console.log('3. Waiting 30 seconds for A2UI messages...');
|
|
console.log(' To trigger, call ask_question tool through CCW server\n');
|
|
|
|
await new Promise(resolve => setTimeout(resolve, 30000));
|
|
|
|
console.log(`\n4. Test completed. Received ${messageCount} messages total.`);
|
|
ws.close();
|
|
|
|
} catch (err) {
|
|
console.error('Test failed:', err.message);
|
|
process.exit(1);
|
|
}
|