From b0b7f15598f12105a44af7cbf08b8e29aad8dcbc Mon Sep 17 00:00:00 2001
From: HexLab98 <liruixinch@outlook.com>
Date: Thu, 23 Jul 2026 08:33:34 +0700
Subject: [PATCH] test(computer-use): assert exact capture skips X11
 active-window probe

Cover exact_target selection and capture(pid=, window_id=) so the
xprop helper is not invoked on the hot path.
---
 .../test_computer_use_cua_backend_linux.py    | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/tests/tools/test_computer_use_cua_backend_linux.py b/tests/tools/test_computer_use_cua_backend_linux.py
index a7d6722f96d9..5ceab6796458 100644
--- a/tests/tools/test_computer_use_cua_backend_linux.py
+++ b/tests/tools/test_computer_use_cua_backend_linux.py
@@ -151,3 +151,48 @@ def test_explicit_app_capture_skips_active_window_fallback():
 
     assert target["window_id"] == 33554439
     active.assert_not_called()
+
+
+def test_exact_target_selection_skips_active_window_fallback():
+    from tools.computer_use.cua_backend import _select_capture_target
+
+    windows = _normalized_windows()[:1]
+
+    with patch("tools.computer_use.cua_backend.sys.platform", "linux"), patch(
+        "tools.computer_use.cua_backend._linux_x11_active_window_id",
+        return_value=84043449,
+    ) as active:
+        target = _select_capture_target(
+            windows, app_requested=False, exact_target=True
+        )
+
+    assert target["window_id"] == 33554439
+    active.assert_not_called()
+
+
+def test_exact_pid_window_capture_does_not_probe_x11_active_window():
+    """capture_after / exact pid+window_id must not pay for an xprop probe."""
+    from unittest.mock import MagicMock
+
+    from tools.computer_use.cua_backend import CuaDriverBackend
+
+    backend = CuaDriverBackend()
+    session = MagicMock()
+    session.call_tool.return_value = {
+        "data": "✅ Chrome — 0 elements",
+        "images": [],
+        "structuredContent": {"elements": []},
+        "isError": False,
+    }
+    backend._session = session
+
+    with patch("tools.computer_use.cua_backend.sys.platform", "linux"), patch(
+        "tools.computer_use.cua_backend._linux_x11_active_window_id",
+        return_value=999,
+    ) as active:
+        backend.capture(mode="ax", pid=1816017, window_id=60817412)
+
+    assert backend._active_pid == 1816017
+    assert backend._active_window_id == 60817412
+    active.assert_not_called()
+    assert all(c.args[0] != "list_windows" for c in session.call_tool.call_args_list)