cherry-studio/asr-server/test.js

/**
 * ASR服务器测试脚本
 * 用于测试ASR服务器是否正常工作
 */

const WebSocket = require('ws');
const http = require('http');

// 测试HTTP服务器
console.log('测试HTTP服务器...');
http.get('http://localhost:34515', (res) => {
  console.log(`HTTP状态码: ${res.statusCode}`);
  
  let data = '';
  res.on('data', (chunk) => {
    data += chunk;
  });
  
  res.on('end', () => {
    console.log('HTTP响应接收完成');
    console.log(`响应长度: ${data.length} 字节`);
    console.log('HTTP测试完成');
    
    // 测试WebSocket
    testWebSocket();
  });
}).on('error', (err) => {
  console.error('HTTP测试失败:', err.message);
});

// 测试WebSocket
function testWebSocket() {
  console.log('\n测试WebSocket...');
  const ws = new WebSocket('ws://localhost:34515');
  
  ws.on('open', () => {
    console.log('WebSocket连接已打开');
    
    // 发送身份识别消息
    ws.send(JSON.stringify({
      type: 'identify',
      role: 'electron'
    }));
    
    // 发送测试消息
    setTimeout(() => {
      console.log('发送测试消息...');
      ws.send(JSON.stringify({
        type: 'test',
        message: '这是一条测试消息'
      }));
    }, 1000);
    
    // 关闭连接
    setTimeout(() => {
      console.log('关闭WebSocket连接...');
      ws.close();
      console.log('测试完成');
    }, 2000);
  });
  
  ws.on('message', (data) => {
    console.log(`收到WebSocket消息: ${data}`);
  });
  
  ws.on('error', (error) => {
    console.error('WebSocket测试失败:', error.message);
  });
}