iframe-proxy | Sunbelt Computer Software

97 lines (69 loc) · 2.68 KB

import socket
import time
from time import sleep
from typing import TYPE_CHECKING, Any, Callable
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
if TYPE_CHECKING:
    from appium.webdriver.webdriver import WebDriver
    from appium.webdriver.webelement import WebElement
class NoAvailablePortError(Exception):
def get_available_from_port_range(from_port: int, to_port: int) -> int:
    """Returns available local port number.
        from_port: The start port to search
        to_port: The end port to search
    Returns:
        int: available local port number which are found first
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    for port in range(from_port, to_port):
        try:
            if sock.connect_ex(('localhost', port)) != 0:
                return port
        finally:
            sock.close()
    raise NoAvailablePortError(f'No available port between {from_port} and {to_port}')
def is_ci() -> bool:
    """Returns if current execution is running on CI
    Returns:
        `True` if current executions is on CI
    return os.getenv('CI', 'false') == 'true'
def wait_for_condition(method: Callable, timeout_sec: float = 5, interval_sec: float = 1) -> Any:
    """Wait while `method` returns the built-in objects considered false
    https://docs.python.org/3/library/stdtypes.html#truth-value-testing
        method: The target method to be waited
        timeout: The timeout to be waited (sec.)
        interval_sec: The interval for wait (sec.)
    Returns:
        Any: value which `method` returns
    Raises:
        ValueError: When interval isn't more than 0
    if interval_sec < 0:
        raise ValueError('interval_sec needs to be not less than 0')
    started = time.time()
    while time.time() - started <= timeout_sec:
        result = method()
        if result:
            break
        sleep(interval_sec)
    return result
def wait_for_element(driver: 'WebDriver', locator: str, value: str, timeout_sec: float = 10) -> 'WebElement':
    """Wait until the element located
        driver: WebDriver instance
        locator: Locator like WebDriver, Mobile JSON Wire Protocol
            (e.g. `appium.webdriver.common.mobileby.MobileBy.ACCESSIBILITY_ID`)
        value: Query value to locator
        timeout_sec: Maximum time to wait the element. If time is over, `TimeoutException` is thrown
    Raises:
        `selenium.common.exceptions.TimeoutException`
    Returns:
        The found WebElement
    return WebDriverWait(driver, timeout_sec).until(EC.presence_of_element_located((locator, value)))

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

test_helper.py

Latest commit

History

test_helper.py

File metadata and controls