100 lines
3.9 KiB
Python
100 lines
3.9 KiB
Python
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()
|