Files
agentic-mobile-control/tests/test_wda_integration.py
T

29 lines
686 B
Python

from __future__ import annotations
import os
import pytest
from core.wda_driver import WDADriver, WDADriverConfig
@pytest.mark.integration
def test_wda_driver_screenshot_against_real_device() -> None:
server_url = os.getenv("APEX_WDA_SERVER_URL")
if not server_url:
pytest.skip("set APEX_WDA_SERVER_URL to run WDA hardware integration")
driver = WDADriver(
WDADriverConfig(
server_url=server_url,
udid=os.getenv("APEX_WDA_UDID") or None,
device_name=os.getenv("APEX_WDA_DEVICE_NAME") or None,
)
)
driver.connect()
try:
assert driver.screenshot()
finally:
driver.disconnect()