diff --git a/tests/test_double_tap.py b/tests/test_double_tap.py new file mode 100644 index 0000000..8294c66 --- /dev/null +++ b/tests/test_double_tap.py @@ -0,0 +1,38 @@ +from __future__ import annotations + +from device.manager import DeviceManager +from tests.fakes import FakeDriver + + +def _connected(driver: FakeDriver) -> DeviceManager: + manager = DeviceManager() + manager.register_device("phone", lambda: driver) + manager.connect("phone", max_retries=1) + return manager + + +def test_double_tap_disabled_passes_exact(monkeypatch): + monkeypatch.setenv("APEX_HUMANIZE_ENABLED", "false") + from tools.double_tap import double_tap + + driver = FakeDriver() + res = double_tap(50, 60, interval_ms=90, manager=_connected(driver)) + assert driver.calls[-1] == ("double_tap", (50, 60, 90)) + assert res["action"] == "double_tap" + + +def test_double_tap_enabled_jitters_within_radius(monkeypatch): + import random + + monkeypatch.setenv("APEX_HUMANIZE_ENABLED", "true") + from tools.humanize import set_rng + from tools.double_tap import double_tap + + set_rng(random.Random(8)) + try: + driver = FakeDriver() + double_tap(50, 60, manager=_connected(driver)) + finally: + set_rng(None) + px, py, _ = driver.calls[-1][1] + assert abs(px - 50) <= 5.0 and abs(py - 60) <= 5.0 diff --git a/tools/double_tap.py b/tools/double_tap.py new file mode 100644 index 0000000..b9ce09f --- /dev/null +++ b/tools/double_tap.py @@ -0,0 +1,30 @@ +from __future__ import annotations + +from device.manager import DeviceManager +from tools._device import get_driver +from tools.humanize import get_rng, jitter_duration, jitter_point, load_humanize_config + + +def double_tap( + x: float, + y: float, + *, + interval_ms: int = 80, + device_id: str | None = None, + manager: DeviceManager | None = None, +) -> dict[str, object]: + cfg = load_humanize_config() + px, py = x, y + interval = interval_ms + if cfg.enabled: + rng = get_rng() + px, py = jitter_point(x, y, radius=cfg.tap_radius_px, rng=rng) + interval = jitter_duration(interval_ms, spread=cfg.duration_spread, rng=rng) + get_driver(device_id, manager=manager).double_tap(px, py, interval) + return { + "ok": True, + "action": "double_tap", + "x": px, + "y": py, + "interval_ms": interval, + }