Cursor Proxmox MCP

""" Tests for VM console operations. """ import pytest from unittest.mock import Mock, patch from proxmox_mcp.tools.console import VMConsoleManager @pytest.fixture def mock_proxmox(): """Fixture to create a mock ProxmoxAPI instance.""" mock = Mock() # Setup chained mock calls mock.nodes.return_value.qemu.return_value.status.current.get.return_value = { "status": "running" } # Mock the exec endpoint to return a proper dictionary with pid mock.nodes.return_value.qemu.return_value.agent.return_value.post.return_value = { "pid": 12345 } # Mock the exec-status endpoint mock.nodes.return_value.qemu.return_value.agent.return_value.get.return_value = { "out-data": "command output", "err-data": "", "exitcode": 0, "exited": 1 } return mock @pytest.fixture def vm_console(mock_proxmox): """Fixture to create a VMConsoleManager instance.""" return VMConsoleManager(mock_proxmox) @pytest.mark.asyncio async def test_execute_command_success(vm_console, mock_proxmox): """Test successful command execution.""" result = await vm_console.execute_command("node1", "100", "ls -l") assert result["success"] is True assert result["output"] == "command output" assert result["error"] == "" assert result["exit_code"] == 0 # Verify correct API calls mock_proxmox.nodes.assert_called_with("node1") mock_proxmox.nodes.return_value.qemu.assert_called_with("100") @pytest.mark.asyncio async def test_execute_command_vm_not_running(vm_console, mock_proxmox): """Test command execution on stopped VM.""" mock_proxmox.nodes.return_value.qemu.return_value.status.current.get.return_value = { "status": "stopped" } with pytest.raises(ValueError, match="not running"): await vm_console.execute_command("node1", "100", "ls -l") @pytest.mark.asyncio async def test_execute_command_vm_not_found(vm_console, mock_proxmox): """Test command execution on non-existent VM.""" mock_proxmox.nodes.return_value.qemu.return_value.status.current.get.side_effect = \ Exception("VM not found") with pytest.raises(ValueError, match="not found"): await vm_console.execute_command("node1", "100", "ls -l") @pytest.mark.asyncio async def test_execute_command_failure(vm_console, mock_proxmox): """Test command execution failure.""" # Mock the exec endpoint to raise an exception mock_proxmox.nodes.return_value.qemu.return_value.agent.return_value.post.side_effect = \ Exception("Command failed") with pytest.raises(RuntimeError, match="Failed to execute command"): await vm_console.execute_command("node1", "100", "ls -l") @pytest.mark.asyncio async def test_execute_command_with_error_output(vm_console, mock_proxmox): """Test command execution with error output.""" # Override the default mock for this test mock_proxmox.nodes.return_value.qemu.return_value.agent.return_value.post.return_value = { "pid": 12345 } mock_proxmox.nodes.return_value.qemu.return_value.agent.return_value.get.return_value = { "out-data": "", "err-data": "command error", "exitcode": 1, "exited": 1 } result = await vm_console.execute_command("node1", "100", "invalid-command") assert result["success"] is True # Success refers to API call, not command assert result["output"] == "" assert result["error"] == "command error" assert result["exit_code"] == 1