feat(tools): add double_tap tool with humanize hook
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -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
|
||||
@@ -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,
|
||||
}
|
||||
Reference in New Issue
Block a user