chore: import upstream snapshot with attribution

libs/cua-driver/python/tests/test_wrapper.py

@@ -0,0 +1,85 @@
+"""Tests for the cua-driver Python wrapper."""
+
+import subprocess
+import sys
+from pathlib import Path
+from unittest.mock import Mock, patch
+
+import pytest
+
+
+def test_get_binary_path():
+    """Test that get_binary_path returns a valid path."""
+    from cua_driver.wrapper import get_binary_path
+
+    # This will raise FileNotFoundError if binary doesn't exist
+    # In CI, we need to build the package first for this to pass
+    try:
+        binary_path = get_binary_path()
+        assert binary_path.exists()
+        assert binary_path.name in ("cua-driver", "cua-driver.exe")
+    except FileNotFoundError:
+        # Expected in development without building
+        pytest.skip("Binary not bundled yet (run build_wheel.py first)")
+
+
+def test_run_cua_driver_version(monkeypatch):
+    """Test running cua-driver --version through the wrapper."""
+    from cua_driver.wrapper import run_cua_driver, get_binary_path
+
+    try:
+        binary_path = get_binary_path()
+    except FileNotFoundError:
+        pytest.skip("Binary not bundled yet")
+
+    # Run with --version
+    exit_code = run_cua_driver(["--version"])
+    assert exit_code == 0
+
+
+def test_wrapper_preserves_exit_code():
+    """Test that the wrapper preserves the binary's exit code."""
+    from cua_driver.wrapper import run_cua_driver, get_binary_path
+
+    try:
+        binary_path = get_binary_path()
+    except FileNotFoundError:
+        pytest.skip("Binary not bundled yet")
+
+    # Invalid command should return non-zero
+    exit_code = run_cua_driver(["--this-flag-does-not-exist"])
+    assert exit_code != 0
+
+
+@patch("cua_driver.wrapper.subprocess.run")
+@patch("cua_driver.wrapper.get_binary_path")
+def test_subprocess_args(mock_get_binary, mock_run):
+    """Test that subprocess is called with correct arguments."""
+    from cua_driver.wrapper import run_cua_driver
+
+    mock_binary = Path("/fake/path/cua-driver")
+    mock_get_binary.return_value = mock_binary
+    mock_run.return_value = Mock(returncode=0)
+
+    run_cua_driver(["mcp", "--help"])
+
+    mock_run.assert_called_once()
+    call_args = mock_run.call_args
+    assert call_args[0][0] == [str(mock_binary), "mcp", "--help"]
+    assert call_args[1]["stdin"] == sys.stdin
+    assert call_args[1]["stdout"] == sys.stdout
+    assert call_args[1]["stderr"] == sys.stderr
+
+
+@patch("cua_driver.wrapper.subprocess.run")
+@patch("cua_driver.wrapper.get_binary_path")
+def test_keyboard_interrupt_handling(mock_get_binary, mock_run):
+    """Test that KeyboardInterrupt returns exit code 130."""
+    from cua_driver.wrapper import run_cua_driver
+
+    mock_binary = Path("/fake/path/cua-driver")
+    mock_get_binary.return_value = mock_binary
+    mock_run.side_effect = KeyboardInterrupt()
+
+    exit_code = run_cua_driver(["mcp"])
+    assert exit_code == 130