project refactor

tests/conftest.py

@@ -0,0 +1,52 @@
+# from nicegui.testing.conftest import *
+from collections.abc import Generator
+import os
+import pytest
+from nicegui.testing import Screen
+from nicegui.testing.conftest import (
+    capabilities,  # noqa: F401
+    driver,  # noqa: F401
+    remove_all_screenshots,  # noqa: F401
+    reset_globals,  # noqa: F401
+    DOWNLOAD_DIR,
+)
+from selenium import webdriver
+
+
+@pytest.fixture
+def chrome_options(chrome_options: webdriver.ChromeOptions) -> webdriver.ChromeOptions:
+    """Configure the Chrome driver options."""
+    chrome_options.add_argument("disable-dev-shm-using")
+    chrome_options.add_argument("no-sandbox")
+    chrome_options.add_argument("headless")
+    # check if we are running on GitHub Actions
+    if "GITHUB_ACTIONS" in os.environ:
+        chrome_options.add_argument("disable-gpu")
+    else:
+        chrome_options.add_argument("--use-gl=angle")
+    chrome_options.add_argument("window-size=1920x1080")
+    chrome_options.add_experimental_option(
+        "prefs",
+        {
+            "download.default_directory": str(DOWNLOAD_DIR),
+            "download.prompt_for_download": False,  # To auto download the file
+            "download.directory_upgrade": True,
+        },
+    )
+    if "CHROME_BINARY_LOCATION" in os.environ:
+        chrome_options.binary_location = os.environ["CHROME_BINARY_LOCATION"]
+    return chrome_options
+
+
+@pytest.fixture
+def screen(
+    driver: webdriver.Chrome,  # noqa: F811
+    request: pytest.FixtureRequest,
+    caplog: pytest.LogCaptureFixture,
+) -> Generator[Screen, None, None]:
+    """Create a new Screen instance."""
+    screen_ = Screen(driver, caplog)
+    yield screen_
+    if screen_.is_open:
+        screen_.shot(request.node.name)
+    screen_.stop_server()