Function test_automatic_reconnect

src/py/reactpy/tests/test_client.py:16–70

(browser: Browser)

async def test_automatic_reconnect(browser: Browser):
    port = find_available_port("localhost")
    page = await browser.new_page()

    # we need to wait longer here because the automatic reconnect is not instant
    page.set_default_timeout(10000)

    @reactpy.component
    def SomeComponent():
        count, incr_count = use_counter(0)
        return reactpy.html._(
            reactpy.html.p({"data_count": count, "id": "count"}, "count", count),
            reactpy.html.button(
                {"on_click": lambda e: incr_count(), "id": "incr"}, "incr"
            ),
        )

    async def get_count():
        # need to refetch element because may unmount on reconnect
        count = await page.wait_for_selector("#count")
        return await count.get_attribute("data-count")

    async with AsyncExitStack() as exit_stack:
        server = await exit_stack.enter_async_context(BackendFixture(port=port))
        display = await exit_stack.enter_async_context(
            DisplayFixture(server, driver=page)
        )

        await display.show(SomeComponent)

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

        for i in range(3):
            await poll(get_count).until_equals(str(i))
            await incr.click()

    # the server is disconnected but the last view state is still shown
    await page.wait_for_selector("#count")

    async with AsyncExitStack() as exit_stack:
        server = await exit_stack.enter_async_context(BackendFixture(port=port))
        display = await exit_stack.enter_async_context(
            DisplayFixture(server, driver=page)
        )

        # use mount instead of show to avoid a page refresh
        display.backend.mount(SomeComponent)

        for i in range(3):
            await poll(get_count).until_equals(str(i))

            # need to refetch element because may unmount on reconnect
            incr = await page.wait_for_selector("#incr")

            await incr.click()


async def test_style_can_be_changed(display: DisplayFixture):