86 lines
2.0 KiB
JavaScript
86 lines
2.0 KiB
JavaScript
/**
|
|
* Test client for SSE (Server-Sent Events) streaming connection
|
|
*/
|
|
|
|
import { EventSource } from 'eventsource';
|
|
import fetch from 'node-fetch';
|
|
|
|
const SERVER_URL = 'http://localhost:3001';
|
|
const SSE_URL = `${SERVER_URL}/sse`;
|
|
const MESSAGES_URL = `${SERVER_URL}/messages`;
|
|
|
|
console.log('🔌 Testing SSE streaming connection...');
|
|
|
|
let sessionId = null;
|
|
|
|
// Create SSE connection
|
|
const eventSource = new EventSource(SSE_URL);
|
|
|
|
eventSource.onopen = () => {
|
|
console.log('✅ SSE connection established!');
|
|
};
|
|
|
|
eventSource.onmessage = (event) => {
|
|
try {
|
|
const data = JSON.parse(event.data);
|
|
console.log('📨 Received SSE message:', JSON.stringify(data, null, 2));
|
|
|
|
// Extract session ID from the first message
|
|
if (data.sessionId && !sessionId) {
|
|
sessionId = data.sessionId;
|
|
console.log(`🆔 Session ID: ${sessionId}`);
|
|
|
|
// Test listing tools after connection is established
|
|
setTimeout(() => testListTools(), 1000);
|
|
}
|
|
} catch (error) {
|
|
console.log('📨 Received SSE data:', event.data);
|
|
}
|
|
};
|
|
|
|
eventSource.onerror = (error) => {
|
|
console.error('❌ SSE error:', error);
|
|
};
|
|
|
|
async function testListTools() {
|
|
if (!sessionId) {
|
|
console.error('❌ No session ID available');
|
|
return;
|
|
}
|
|
|
|
console.log('📋 Testing tools/list via SSE...');
|
|
|
|
const message = {
|
|
jsonrpc: '2.0',
|
|
id: 1,
|
|
method: 'tools/list',
|
|
params: {},
|
|
};
|
|
|
|
try {
|
|
const response = await fetch(MESSAGES_URL, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'X-Session-ID': sessionId,
|
|
},
|
|
body: JSON.stringify(message),
|
|
});
|
|
|
|
if (!response.ok) {
|
|
console.error('❌ Failed to send message:', response.status, response.statusText);
|
|
} else {
|
|
console.log('✅ Message sent successfully');
|
|
}
|
|
} catch (error) {
|
|
console.error('❌ Error sending message:', error);
|
|
}
|
|
}
|
|
|
|
// Close connection after 10 seconds
|
|
setTimeout(() => {
|
|
console.log('⏰ Closing SSE connection...');
|
|
eventSource.close();
|
|
process.exit(0);
|
|
}, 10000);
|