756bf9b0f4
* Move sharding internal to StateManager
Avoid leaking sharding implementation details all over the State class and
breaking the API
* WiP StateManager based sharding
* Copy the state __dict__ when serializing to avoid breaking the instance
* State tests need to pass the correct substate token for redis
* state: when getting parent_state, set top_level=False
ensure that we don't end up with a broken tree
* test_app: get tests passing with redis by passing the correct token
refactor upload tests to suck less
* test_client_storage: look up substate key
* state.py: pass static checks
* test_dynamic_routes: working with redis state shard
* Update the remaining AppHarness tests to pass {token}_{state.get_full_name()}
* test_app: pass all tokens with state suffix
* StateManagerRedis: clean up commentary
147 lines
5.3 KiB
Python
147 lines
5.3 KiB
Python
"""Integration tests for text input and related components."""
|
|
import time
|
|
from typing import Generator
|
|
|
|
import pytest
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.keys import Keys
|
|
|
|
from reflex.testing import AppHarness
|
|
|
|
|
|
def FullyControlledInput():
|
|
"""App using a fully controlled input with implicit debounce wrapper."""
|
|
import reflex as rx
|
|
|
|
class State(rx.State):
|
|
text: str = "initial"
|
|
|
|
app = rx.App(state=rx.State)
|
|
|
|
@app.add_page
|
|
def index():
|
|
return rx.fragment(
|
|
rx.chakra.input(
|
|
value=State.router.session.client_token, is_read_only=True, id="token"
|
|
),
|
|
rx.chakra.input(
|
|
id="debounce_input_input",
|
|
on_change=State.set_text, # type: ignore
|
|
value=State.text,
|
|
),
|
|
rx.chakra.input(value=State.text, id="value_input", is_read_only=True),
|
|
rx.chakra.input(on_change=State.set_text, id="on_change_input"), # type: ignore
|
|
rx.el.input(
|
|
value=State.text,
|
|
id="plain_value_input",
|
|
disabled=True,
|
|
_disabled={"background_color": "#EEE"},
|
|
),
|
|
rx.button("CLEAR", on_click=rx.set_value("on_change_input", "")),
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def fully_controlled_input(tmp_path) -> Generator[AppHarness, None, None]:
|
|
"""Start FullyControlledInput app at tmp_path via AppHarness.
|
|
|
|
Args:
|
|
tmp_path: pytest tmp_path fixture
|
|
|
|
Yields:
|
|
running AppHarness instance
|
|
"""
|
|
with AppHarness.create(
|
|
root=tmp_path,
|
|
app_source=FullyControlledInput, # type: ignore
|
|
) as harness:
|
|
yield harness
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fully_controlled_input(fully_controlled_input: AppHarness):
|
|
"""Type text after moving cursor. Update text on backend.
|
|
|
|
Args:
|
|
fully_controlled_input: harness for FullyControlledInput app
|
|
"""
|
|
assert fully_controlled_input.app_instance is not None, "app is not running"
|
|
driver = fully_controlled_input.frontend()
|
|
|
|
# get a reference to the connected client
|
|
token_input = driver.find_element(By.ID, "token")
|
|
assert token_input
|
|
|
|
# wait for the backend connection to send the token
|
|
token = fully_controlled_input.poll_for_value(token_input)
|
|
assert token
|
|
|
|
async def get_state_text():
|
|
state = await fully_controlled_input.get_state(f"{token}_state.state")
|
|
return state.substates["state"].text
|
|
|
|
# find the input and wait for it to have the initial state value
|
|
debounce_input = driver.find_element(By.ID, "debounce_input_input")
|
|
value_input = driver.find_element(By.ID, "value_input")
|
|
on_change_input = driver.find_element(By.ID, "on_change_input")
|
|
plain_value_input = driver.find_element(By.ID, "plain_value_input")
|
|
clear_button = driver.find_element(By.TAG_NAME, "button")
|
|
assert fully_controlled_input.poll_for_value(debounce_input) == "initial"
|
|
assert fully_controlled_input.poll_for_value(value_input) == "initial"
|
|
assert fully_controlled_input.poll_for_value(plain_value_input) == "initial"
|
|
assert (
|
|
plain_value_input.value_of_css_property("background-color")
|
|
== "rgba(238, 238, 238, 1)"
|
|
)
|
|
|
|
# move cursor to home, then to the right and type characters
|
|
debounce_input.send_keys(Keys.HOME, Keys.ARROW_RIGHT)
|
|
debounce_input.send_keys("foo")
|
|
time.sleep(0.5)
|
|
assert debounce_input.get_attribute("value") == "ifoonitial"
|
|
assert await get_state_text() == "ifoonitial"
|
|
assert fully_controlled_input.poll_for_value(value_input) == "ifoonitial"
|
|
assert fully_controlled_input.poll_for_value(plain_value_input) == "ifoonitial"
|
|
|
|
# clear the input on the backend
|
|
async with fully_controlled_input.modify_state(f"{token}_state.state") as state:
|
|
state.substates["state"].text = ""
|
|
assert await get_state_text() == ""
|
|
assert (
|
|
fully_controlled_input.poll_for_value(
|
|
debounce_input, exp_not_equal="ifoonitial"
|
|
)
|
|
== ""
|
|
)
|
|
|
|
# type more characters
|
|
debounce_input.send_keys("getting testing done")
|
|
time.sleep(0.5)
|
|
assert debounce_input.get_attribute("value") == "getting testing done"
|
|
assert await get_state_text() == "getting testing done"
|
|
assert fully_controlled_input.poll_for_value(value_input) == "getting testing done"
|
|
assert (
|
|
fully_controlled_input.poll_for_value(plain_value_input)
|
|
== "getting testing done"
|
|
)
|
|
|
|
# type into the on_change input
|
|
on_change_input.send_keys("overwrite the state")
|
|
time.sleep(0.5)
|
|
assert debounce_input.get_attribute("value") == "overwrite the state"
|
|
assert on_change_input.get_attribute("value") == "overwrite the state"
|
|
assert await get_state_text() == "overwrite the state"
|
|
assert fully_controlled_input.poll_for_value(value_input) == "overwrite the state"
|
|
assert (
|
|
fully_controlled_input.poll_for_value(plain_value_input)
|
|
== "overwrite the state"
|
|
)
|
|
|
|
clear_button.click()
|
|
time.sleep(0.5)
|
|
assert on_change_input.get_attribute("value") == ""
|
|
# potential bug: clearing the on_change field doesn't itself trigger on_change
|
|
# assert backend_state.text == ""
|
|
# assert debounce_input.get_attribute("value") == ""
|
|
# assert value_input.get_attribute("value") == ""
|