nicegui/tests/conftest.py

import datetime
import importlib
import os
from typing import Callable, Generator

import icecream
import pytest

if True:  # NOTE: prevent formatter from sorting the imports (nicegui must come before justpy)
    from nicegui import ui, globals

from justpy.htmlcomponents import JustpyBaseComponent, WebPage
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

from .screen import Screen

icecream.install()


@pytest.fixture
def chrome_options(chrome_options: webdriver.ChromeOptions) -> webdriver.ChromeOptions:
    chrome_options.add_argument('headless')
    chrome_options.add_argument('disable-gpu')
    chrome_options.add_argument('window-size=600x600')
    #chrome_options.add_experimental_option('w3c', False)
    return chrome_options


@pytest.fixture
def capabilities(capabilities):
    capabilities['goog:loggingPrefs'] = {'browser': 'ALL'}
    return capabilities


@pytest.fixture
def selenium(selenium: webdriver.Chrome) -> webdriver.Chrome:
    selenium.implicitly_wait(0.1)
    selenium.set_page_load_timeout(1)
    return selenium


@pytest.fixture(autouse=True)
def reset_globals() -> Generator[None, None, None]:
    WebPage.sockets.clear()
    WebPage.instances.clear()
    WebPage.next_page_id = 0
    JustpyBaseComponent.instances.clear()
    JustpyBaseComponent.next_id = 0
    importlib.reload(globals)


@pytest.fixture()
def screen(selenium: webdriver.Chrome) -> Generator[Screen, None, None]:
    screen = Screen(selenium)
    yield screen
    screen.stop_server()


@pytest.fixture
def screenshot(selenium: webdriver.Chrome) -> Callable[[str], None]:
    # original taken from https://github.com/theserverlessway/pytest-chrome/blob/master/tests/conftest.py
    def shot(name: str = '') -> None:
        directory = 'screenshots'
        if not os.path.exists(directory):
            os.makedirs(directory)
        identifier = datetime.datetime.now().isoformat()
        if name:
            identifier = f'{identifier}-{name}'
        filename = f'{directory}/{identifier}.png'
        print(f'Storing Screenshot to {filename}')
        selenium.get_screenshot_as_file(filename)
    return shot