test_pyscript.py

# ruff: noqa: S701
import asyncio
from pathlib import Path

import pytest
from jinja2 import Environment as JinjaEnvironment
from jinja2 import FileSystemLoader as JinjaFileSystemLoader
from requests import request
from starlette.applications import Starlette
from starlette.routing import Route
from starlette.templating import Jinja2Templates

import reactpy
from reactpy import html
from reactpy.executors.asgi.pyscript import ReactPyCsr
from reactpy.testing import BackendFixture, DisplayFixture
from reactpy.testing.common import REACTPY_TESTS_DEFAULT_TIMEOUT


@pytest.fixture(scope="module")
async def display(browser):
    """Override for the display fixture that uses ReactPyMiddleware."""
    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        initial=html.div({"id": "loading"}, "Loading..."),
    )

    async with BackendFixture(app) as server:
        async with DisplayFixture(
            backend=server, browser=browser, timeout=20
        ) as new_display:
            yield new_display


@pytest.fixture(scope="module")
async def multi_file_display(browser):
    """Override for the display fixture that uses ReactPyMiddleware."""
    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "load_first.py",
        Path(__file__).parent / "pyscript_components" / "load_second.py",
        initial=html.div({"id": "loading"}, "Loading..."),
    )

    async with BackendFixture(app) as server:
        async with DisplayFixture(backend=server, browser=browser) as new_display:
            yield new_display


@pytest.fixture(scope="module")
async def jinja_display(browser):
    """Override for the display fixture that uses ReactPyMiddleware."""
    templates = Jinja2Templates(
        env=JinjaEnvironment(
            loader=JinjaFileSystemLoader("tests/templates"),
            extensions=["reactpy.templatetags.ReactPyJinja"],
        )
    )

    async def homepage(request):
        return templates.TemplateResponse(request, "pyscript.html")

    app = Starlette(routes=[Route("/", homepage)])

    async with BackendFixture(app) as server:
        async with DisplayFixture(backend=server, browser=browser) as new_display:
            yield new_display


async def test_root_component(display: DisplayFixture):
    await display.goto("/")

    await display.page.wait_for_selector("#loading")
    await display.page.wait_for_selector("#incr")

    await display.page.click("#incr")
    await display.page.wait_for_selector("#incr[data-count='1']")

    await display.page.click("#incr")
    await display.page.wait_for_selector("#incr[data-count='2']")

    await display.page.click("#incr")
    await display.page.wait_for_selector("#incr[data-count='3']")


async def test_multi_file_components(multi_file_display: DisplayFixture):
    await multi_file_display.goto("/")

    await multi_file_display.page.wait_for_selector("#incr")

    await multi_file_display.page.click("#incr")
    await multi_file_display.page.wait_for_selector("#incr[data-count='1']")

    await multi_file_display.page.click("#incr")
    await multi_file_display.page.wait_for_selector("#incr[data-count='2']")

    await multi_file_display.page.click("#incr")
    await multi_file_display.page.wait_for_selector("#incr[data-count='3']")


def test_bad_file_path():
    with pytest.raises(ValueError):
        ReactPyCsr()


async def test_customized_noscript(tmp_path: Path):
    noscript_file = tmp_path / "noscript.html"
    noscript_file.write_text(
        '<p id="noscript-message">Please enable JavaScript.</p>',
        encoding="utf-8",
    )

    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript=noscript_file,
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            '<noscript><p id="noscript-message">Please enable JavaScript.</p></noscript>'
            in response.text
        )



async def test_customized_noscript_string():
    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript='<p id="noscript-message">Please enable JavaScript.</p>',
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            '<noscript><p id="noscript-message">Please enable JavaScript.</p></noscript>'
            in response.text
        )


async def test_customized_noscript_from_file(tmp_path: Path):
    noscript_file = tmp_path / "noscript.html"
    noscript_file.write_text(
        '<p id="noscript-message">Please enable JavaScript.</p>',
        encoding="utf-8",
    )

    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript=noscript_file,
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            '<noscript><p id="noscript-message">Please enable JavaScript.</p></noscript>'
            in response.text
        )


async def test_customized_noscript_from_string():
    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript='<p id="noscript-message">Please enable JavaScript.</p>',
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            '<noscript><p id="noscript-message">Please enable JavaScript.</p></noscript>'
            in response.text
        )


async def test_customized_noscript_from_component():
    @reactpy.component
    def noscript_message():
        return html.p({"id": "noscript-message"}, "Please enable JavaScript.")

    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript=noscript_message,
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            '<noscript><p id="noscript-message">Please enable JavaScript.</p></noscript>'
            in response.text
        )


async def test_default_noscript_rendered():
    app = ReactPyCsr(Path(__file__).parent / "pyscript_components" / "root.py")

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert "<noscript>Enable JavaScript to view this site.</noscript>" in response.text



async def test_noscript_omitted():
    app = ReactPyCsr(Path(__file__).parent / "pyscript_components" / "root.py")

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert (
            "<noscript>Enable JavaScript to view this site.</noscript>"
            in response.text
        )


async def test_noscript_disabled():
    app = ReactPyCsr(
        Path(__file__).parent / "pyscript_components" / "root.py",
        html_noscript=None,
    )

    async with BackendFixture(app) as server:
        url = f"http://{server.host}:{server.port}"
        response = await asyncio.to_thread(
            request, "GET", url, timeout=REACTPY_TESTS_DEFAULT_TIMEOUT.current
        )
        assert response.status_code == 200
        assert "<noscript>" not in response.text


async def test_jinja_template_tag(jinja_display: DisplayFixture):
    await jinja_display.goto("/")

    await jinja_display.page.wait_for_selector("#loading")
    await jinja_display.page.wait_for_selector("#incr")

    await jinja_display.page.click("#incr")
    await jinja_display.page.wait_for_selector("#incr[data-count='1']")

    await jinja_display.page.click("#incr")
    await jinja_display.page.wait_for_selector("#incr[data-count='2']")

    await jinja_display.page.click("#incr")
    await jinja_display.page.wait_for_selector("#incr[data-count='3']")