from __future__ import annotations

import os

import pytest

from driver.android_driver import AndroidDriver, AndroidDriverConfig


@pytest.mark.integration
def test_android_driver_screenshot_against_real_device() -> None:
    server_url = os.getenv("APEX_ANDROID_SERVER_URL")
    if not server_url:
        pytest.skip("set APEX_ANDROID_SERVER_URL to run Android hardware integration")

    driver = AndroidDriver(
        AndroidDriverConfig(
            server_url=server_url,
            udid=os.getenv("APEX_ANDROID_UDID") or None,
            device_name=os.getenv("APEX_ANDROID_DEVICE_NAME") or None,
        )
    )
    driver.connect()
    try:
        assert driver.screenshot()
    finally:
        driver.disconnect()