feat: 票据合规预审助手首版 v1.0.0（verify-invoice-default）

examples/real_desktop_rpa/tests/test_selector_contract.py

@@ -0,0 +1,99 @@
+from __future__ import annotations
+
+import sys
+import unittest
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+
+ROOT = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT / "scripts"))
+
+from service.desktop_session import (  # noqa: E402
+    DesktopRpaError,
+    WindowCandidate,
+    activate_window,
+    resolve_unique_window,
+)
+from service import uia_locator  # noqa: E402
+from util.selectors import ElementSpec, MAIN_WINDOW  # noqa: E402
+
+
+class SelectorContractTest(unittest.TestCase):
+    def test_unique_window_is_required(self) -> None:
+        candidate = WindowCandidate(1, 10, "WXWork", "WeWorkWindow", "企业微信", True)
+        self.assertEqual(
+            resolve_unique_window(MAIN_WINDOW, candidates=[candidate]),
+            candidate,
+        )
+
+    def test_duplicate_windows_fail_instead_of_taking_first(self) -> None:
+        candidates = [
+            WindowCandidate(1, 10, "WXWork", "WeWorkWindow", "企业微信", True),
+            WindowCandidate(2, 11, "WXWork", "WeWorkWindow", "企业微信", True),
+        ]
+        with self.assertRaises(DesktopRpaError) as ctx:
+            resolve_unique_window(MAIN_WINDOW, candidates=candidates)
+        self.assertEqual(ctx.exception.code, "ELEMENT_AMBIGUOUS")
+
+    def test_hidden_window_is_not_a_candidate(self) -> None:
+        hidden = WindowCandidate(1, 10, "WXWork", "WeWorkWindow", "企业微信", False)
+        with self.assertRaises(DesktopRpaError) as ctx:
+            resolve_unique_window(MAIN_WINDOW, candidates=[hidden])
+        self.assertEqual(ctx.exception.code, "WINDOW_NOT_FOUND")
+
+    def test_activate_without_maximize_skips_restore_when_zoomed(self) -> None:
+        window = WindowCandidate(42, 10, "WXWork", "WeWorkWindow", "企业微信", True)
+        with patch("service.desktop_session.user32") as user32:
+            user32.IsIconic.return_value = False
+            user32.IsZoomed.return_value = True
+            user32.SetForegroundWindow.return_value = True
+            activate_window(window, maximize=False)
+            user32.ShowWindow.assert_not_called()
+            user32.SetForegroundWindow.assert_called_once_with(42)
+
+    def test_activate_maximize_when_not_zoomed(self) -> None:
+        window = WindowCandidate(42, 10, "WXWork", "WeWorkWindow", "企业微信", True)
+        with patch("service.desktop_session.user32") as user32:
+            user32.IsIconic.return_value = False
+            user32.IsZoomed.return_value = False
+            user32.SetForegroundWindow.return_value = True
+            activate_window(window, maximize=True)
+            user32.ShowWindow.assert_called()
+            user32.SetForegroundWindow.assert_called_once_with(42)
+
+    def test_locate_unique_matches_child_value(self) -> None:
+        row = MagicMock()
+        row.Name = "status_layout"
+        row.AutomationId = ""
+        row.ControlTypeName = "TreeControl"
+        row.ClassName = ""
+        label = MagicMock()
+        label.Name = "main_chattext"
+        label.GetChildren.return_value = []
+        row.GetChildren.return_value = [label]
+
+        root = MagicMock()
+        root.Name = "wlist_view"
+        root.AutomationId = ""
+        root.ControlTypeName = "TreeControl"
+        root.ClassName = ""
+        root.GetChildren.return_value = [row]
+
+        with patch.object(uia_locator, "control_value", side_effect=lambda c: "新的客户" if c is label else ""):
+            spec = ElementSpec(
+                "new_customer_row",
+                name="status_layout",
+                ancestor_name="wlist_view",
+                child_value_name="main_chattext",
+                value="新的客户",
+            )
+            # ancestor is root itself when searching from root with name wlist_view
+            found = uia_locator.locate_unique(root, ElementSpec("wlist", name="wlist_view"))
+            self.assertIs(found, root)
+            found_row = uia_locator.locate_unique(root, spec)
+            self.assertIs(found_row, row)
+
+
+if __name__ == "__main__":
+    unittest.main()