laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add some backend var state inheritance tests (#2685)

Browse Source
This commit is contained in:
benedikt-bartscher 2024-02-29 00:11:14 +01:00 committed by GitHub
parent e70c149752
commit c465b94016
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
integration/test_state_inheritance.py
View File

@ -77,19 +77,29 @@ def StateInheritance():
) )
class Base1(rx.State, Mixin): class Base1(rx.State, Mixin):
_base1: str = "_base1"
base1: str = "base1" base1: str = "base1"
@rx.var @rx.var
def computed_basevar(self) -> str: def computed_basevar(self) -> str:
return "computed_basevar1" return "computed_basevar1"
@rx.var
def computed_backend_vars_base1(self) -> str:
return self._base1
class Base2(rx.State): class Base2(rx.State):
_base2: str = "_base2"
base2: str = "base2" base2: str = "base2"
@rx.var @rx.var
def computed_basevar(self) -> str: def computed_basevar(self) -> str:
return "computed_basevar2" return "computed_basevar2"
@rx.var
def computed_backend_vars_base2(self) -> str:
return self._base2
class Child1(Base1, OtherMixin): class Child1(Base1, OtherMixin):
pass pass
@ -97,12 +107,17 @@ def StateInheritance():
pass pass
class Child3(Child2): class Child3(Child2):
_child3: str = "_child3"
child3: str = "child3" child3: str = "child3"
@rx.var @rx.var
def computed_childvar(self) -> str: def computed_childvar(self) -> str:
return "computed_childvar" return "computed_childvar"
@rx.var
def computed_backend_vars_child3(self) -> str:
return f"{self._base2}.{self._child3}"
def index() -> rx.Component: def index() -> rx.Component:
return rx.vstack( return rx.vstack(
rx.chakra.input( rx.chakra.input(
@ -118,9 +133,15 @@ def StateInheritance():
on_click=Base1.on_click_mixin, # type: ignore on_click=Base1.on_click_mixin, # type: ignore
id="base1-mixin-btn", id="base1-mixin-btn",
), ),
rx.heading(
Base1.computed_backend_vars_base1, id="base1-computed_backend_vars"
),
# Base 2 # Base 2
rx.heading(Base2.computed_basevar, id="base2-computed_basevar"), rx.heading(Base2.computed_basevar, id="base2-computed_basevar"),
rx.heading(Base2.base2, id="base2-base2"), rx.heading(Base2.base2, id="base2-base2"),
rx.heading(
Base2.computed_backend_vars_base2, id="base2-computed_backend_vars"
),
# Child 1 # Child 1
rx.heading(Child1.computed_basevar, id="child1-computed_basevar"), rx.heading(Child1.computed_basevar, id="child1-computed_basevar"),
rx.heading(Child1.computed_mixin, id="child1-computed_mixin"), rx.heading(Child1.computed_mixin, id="child1-computed_mixin"),
@ -169,6 +190,9 @@ def StateInheritance():
on_click=Child3.on_click_other_mixin, # type: ignore on_click=Child3.on_click_other_mixin, # type: ignore
id="child3-other-mixin-btn", id="child3-other-mixin-btn",
), ),
rx.heading(
Child3.computed_backend_vars_child3, id="child3-computed_backend_vars"
),
) )
app = rx.App() app = rx.App()
@ -262,6 +286,11 @@ def test_state_inheritance(
base1_base1 = driver.find_element(By.ID, "base1-base1") base1_base1 = driver.find_element(By.ID, "base1-base1")
assert base1_base1.text == "base1" assert base1_base1.text == "base1"
base1_computed_backend_vars = driver.find_element(
By.ID, "base1-computed_backend_vars"
)
assert base1_computed_backend_vars.text == "_base1"
# Base 2 # Base 2
base2_computed_basevar = driver.find_element(By.ID, "base2-computed_basevar") base2_computed_basevar = driver.find_element(By.ID, "base2-computed_basevar")
assert base2_computed_basevar.text == "computed_basevar2" assert base2_computed_basevar.text == "computed_basevar2"
@ -269,6 +298,11 @@ def test_state_inheritance(
base2_base2 = driver.find_element(By.ID, "base2-base2") base2_base2 = driver.find_element(By.ID, "base2-base2")
assert base2_base2.text == "base2" assert base2_base2.text == "base2"
base2_computed_backend_vars = driver.find_element(
By.ID, "base2-computed_backend_vars"
)
assert base2_computed_backend_vars.text == "_base2"
# Child 1 # Child 1
child1_computed_basevar = driver.find_element(By.ID, "child1-computed_basevar") child1_computed_basevar = driver.find_element(By.ID, "child1-computed_basevar")
assert child1_computed_basevar.text == "computed_basevar1" assert child1_computed_basevar.text == "computed_basevar1"
@ -332,6 +366,11 @@ def test_state_inheritance(
child3_other_mixin = driver.find_element(By.ID, "child3-other_mixin") child3_other_mixin = driver.find_element(By.ID, "child3-other_mixin")
assert child3_other_mixin.text == "other_mixin" assert child3_other_mixin.text == "other_mixin"
child3_computed_backend_vars = driver.find_element(
By.ID, "child3-computed_backend_vars"
)
assert child3_computed_backend_vars.text == "_base2._child3"
# Event Handler Tests # Event Handler Tests
raises_alert(driver, "base1-mixin-btn") raises_alert(driver, "base1-mixin-btn")
raises_alert(driver, "child2-mixin-btn") raises_alert(driver, "child2-mixin-btn")