test-command-execution.js

Go to the documentation of this file.

/**
 * Test script to verify command execution works end-to-end
 * Tests the execute_script functionality added to agent v2
 */
 
const WebSocket = require('ws');
 
// Agent UUID from the VM
const agentUuid = '2c3e2d7a-c521-4844-b44b-a30b4c2ec4d8';
const wsUrl = 'wss://rmm-psa-backend-t9f7k.ondigitalocean.app/api/ws';
 
console.log('Connecting to WebSocket as backend...');
const ws = new WebSocket(`${wsUrl}?agent_uuid=${agentUuid}`);
 
ws.on('open', () => {
  console.log('✅ Connected to WebSocket');
  
  // Send a test script execution
  const testCommand = {
    type: 'execute_script',
    execution_id: 'test-' + Date.now(),
    script_id: 'test-script-1',
    code: 'echo "Hello from Agent v2!"; echo "Current directory: $PWD"; whoami'
  };
  
  console.log('\n📤 Sending command execution:', testCommand);
  ws.send(JSON.stringify(testCommand));
  
  // Timeout after 30 seconds
  setTimeout(() => {
    console.log('\n⏱️  Test timeout - closing connection');
    ws.close();
    process.exit(0);
  }, 30000);
});
 
ws.on('message', (data) => {
  try {
    const message = JSON.parse(data.toString());
    
    switch (message.type) {
      case 'script_output':
        console.log(`📝 Output [${message.execution_id}]:`, message.line);
        break;
        
      case 'script_complete':
        console.log(`✅ Script completed [${message.execution_id}] - Exit code: ${message.exit_code}`);
        setTimeout(() => {
          ws.close();
          process.exit(0);
        }, 1000);
        break;
        
      case 'script_error':
        console.log(`❌ Script error [${message.execution_id}]:`, message.error);
        setTimeout(() => {
          ws.close();
          process.exit(1);
        }, 1000);
        break;
        
      default:
        console.log('📨 Received:', message.type);
    }
  } catch (err) {
    console.error('Error parsing message:', err);
  }
});
 
ws.on('error', (error) => {
  console.error('❌ WebSocket error:', error.message);
  process.exit(1);
});
 
ws.on('close', () => {
  console.log('🔌 Connection closed');
});