Class TestRunCommandSafely

tests/test_helpers.py:258–303

Test the run_command_safely function.

class TestRunCommandSafely:
    """Test the run_command_safely function."""

    @pytest.fixture
    def mock_console(self):
        """Fixture providing a mock console."""
        console = Mock()
        console.write = Mock()
        console.read = Mock()
        return console

    @pytest.mark.asyncio
    async def test_run_command_safely_basic(self, mock_console):
        """Test basic command execution."""
        # Mock console read to return prompt immediately
        mock_console.read.return_value = {
            'data': 'command output\n',
            'prompt': '\x01\x02msf6\x01\x02 \x01\x02> \x01\x02',
            'busy': False
        }
        
        result = await run_command_safely(mock_console, 'help')
        
        mock_console.write.assert_called_once_with('help\n')
        assert 'command output' in result

    @pytest.mark.asyncio
    async def test_run_command_safely_invalid_console(self, mock_console):
        """Test command execution with invalid console."""
        # Remove required methods
        delattr(mock_console, 'write')
        
        with pytest.raises(TypeError, match="Unsupported console object"):
            await run_command_safely(mock_console, 'help')

    @pytest.mark.asyncio
    async def test_run_command_safely_read_error(self, mock_console):
        """Test command execution with read error - should timeout gracefully."""
        mock_console.read.side_effect = Exception("Read failed")

        # Should not raise exception, but timeout and return empty result
        result = await run_command_safely(mock_console, 'help')
        
        # Should return empty string after timeout
        assert isinstance(result, str)
        assert result == ""  # Empty result after timeout


class TestFindAvailablePort: