from __future__ import annotations

import os

import pytest

from driver.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()