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

extend state inheritance tests with non-computed vars (#2425)

Browse Source
This commit is contained in:
benedikt-bartscher 2024-01-23 21:20:34 +01:00 committed by GitHub
parent 0b1b8ee639
commit 6401dc6a7f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 126 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

172
integration/test_state_inheritance.py
View File

@ -13,29 +13,39 @@ def StateInheritance():
import reflex as rx import reflex as rx
class ChildMixin: class ChildMixin:
child_mixin: str = "child_mixin"
@rx.var @rx.var
def child_mixin(self) -> str: def computed_child_mixin(self) -> str:
return "child_mixin" return "computed_child_mixin"
class Mixin(ChildMixin): class Mixin(ChildMixin):
mixin: str = "mixin"
@rx.var @rx.var
def mixin(self) -> str: def computed_mixin(self) -> str:
return "mixin" return "computed_mixin"
class OtherMixin(rx.Base): class OtherMixin(rx.Base):
other_mixin: str = "other_mixin"
@rx.var @rx.var
def other_mixin(self) -> str: def computed_other_mixin(self) -> str:
return "other_mixin" return "computed_other_mixin"
class Base1(rx.State, Mixin): class Base1(rx.State, Mixin):
base1: str = "base1"
@rx.var @rx.var
def basevar(self) -> str: def computed_basevar(self) -> str:
return "basevar1" return "computed_basevar1"
class Base2(rx.State): class Base2(rx.State):
base2: str = "base2"
@rx.var @rx.var
def basevar(self) -> str: def computed_basevar(self) -> str:
return "basevar2" return "computed_basevar2"
class Child1(Base1, OtherMixin): class Child1(Base1, OtherMixin):
pass pass
@ -44,32 +54,51 @@ def StateInheritance():
pass pass
class Child3(Child2): class Child3(Child2):
child3: str = "child3"
@rx.var @rx.var
def childvar(self) -> str: def computed_childvar(self) -> str:
return "childvar" return "computed_childvar"
def index() -> rx.Component: def index() -> rx.Component:
return rx.vstack( return rx.vstack(
rx.input( rx.input(
id="token", value=Base1.router.session.client_token, is_read_only=True id="token", value=Base1.router.session.client_token, is_read_only=True
), ),
rx.heading(Base1.computed_mixin, id="base1-computed_mixin"),
rx.heading(Base1.computed_basevar, id="base1-computed_basevar"),
rx.heading(Base1.computed_child_mixin, id="base1-child-mixin"),
rx.heading(Base1.base1, id="base1-base1"),
rx.heading(Base1.mixin, id="base1-mixin"), rx.heading(Base1.mixin, id="base1-mixin"),
rx.heading(Base1.basevar, id="base1-basevar"), rx.heading(Base1.child_mixin, id="base1-child_mixin"),
rx.heading(Base1.child_mixin, id="base1-child-mixin"), rx.heading(Base2.computed_basevar, id="base2-computed_basevar"),
rx.heading(Base2.basevar, id="base2-basevar"), rx.heading(Base2.base2, id="base2-base2"),
rx.heading(Child1.basevar, id="child1-basevar"), rx.heading(Child1.computed_basevar, id="child1-computed_basevar"),
rx.heading(Child1.computed_mixin, id="child1-computed_mixin"),
rx.heading(Child1.computed_other_mixin, id="child1-other-mixin"),
rx.heading(Child1.computed_child_mixin, id="child1-child-mixin"),
rx.heading(Child1.base1, id="child1-base1"),
rx.heading(Child1.mixin, id="child1-mixin"), rx.heading(Child1.mixin, id="child1-mixin"),
rx.heading(Child1.other_mixin, id="child1-other-mixin"), rx.heading(Child1.other_mixin, id="child1-other_mixin"),
rx.heading(Child1.child_mixin, id="child1-child-mixin"), rx.heading(Child1.child_mixin, id="child1-child_mixin"),
rx.heading(Child2.basevar, id="child2-basevar"), rx.heading(Child2.computed_basevar, id="child2-computed_basevar"),
rx.heading(Child2.computed_mixin, id="child2-computed_mixin"),
rx.heading(Child2.computed_other_mixin, id="child2-other-mixin"),
rx.heading(Child2.computed_child_mixin, id="child2-child-mixin"),
rx.heading(Child2.base2, id="child2-base2"),
rx.heading(Child2.mixin, id="child2-mixin"), rx.heading(Child2.mixin, id="child2-mixin"),
rx.heading(Child2.other_mixin, id="child2-other-mixin"), rx.heading(Child2.other_mixin, id="child2-other_mixin"),
rx.heading(Child2.child_mixin, id="child2-child-mixin"), rx.heading(Child2.child_mixin, id="child2-child_mixin"),
rx.heading(Child3.basevar, id="child3-basevar"), rx.heading(Child3.computed_basevar, id="child3-computed_basevar"),
rx.heading(Child3.computed_mixin, id="child3-computed_mixin"),
rx.heading(Child3.computed_other_mixin, id="child3-other-mixin"),
rx.heading(Child3.computed_childvar, id="child3-computed_childvar"),
rx.heading(Child3.computed_child_mixin, id="child3-child-mixin"),
rx.heading(Child3.child3, id="child3-child3"),
rx.heading(Child3.base2, id="child3-base2"),
rx.heading(Child3.mixin, id="child3-mixin"), rx.heading(Child3.mixin, id="child3-mixin"),
rx.heading(Child3.other_mixin, id="child3-other-mixin"), rx.heading(Child3.other_mixin, id="child3-other_mixin"),
rx.heading(Child3.childvar, id="child3-childvar"), rx.heading(Child3.child_mixin, id="child3-child_mixin"),
rx.heading(Child3.child_mixin, id="child3-child-mixin"),
) )
app = rx.App() app = rx.App()
@ -149,53 +178,104 @@ def test_state_inheritance(
""" """
assert state_inheritance.app_instance is not None assert state_inheritance.app_instance is not None
base1_mixin = driver.find_element(By.ID, "base1-computed_mixin")
assert base1_mixin.text == "computed_mixin"
base1_computed_basevar = driver.find_element(By.ID, "base1-computed_basevar")
assert base1_computed_basevar.text == "computed_basevar1"
base1_computed_child_mixin = driver.find_element(By.ID, "base1-child-mixin")
assert base1_computed_child_mixin.text == "computed_child_mixin"
base1_base1 = driver.find_element(By.ID, "base1-base1")
assert base1_base1.text == "base1"
base1_mixin = driver.find_element(By.ID, "base1-mixin") base1_mixin = driver.find_element(By.ID, "base1-mixin")
assert base1_mixin.text == "mixin" assert base1_mixin.text == "mixin"
base1_basevar = driver.find_element(By.ID, "base1-basevar") base1_child_mixin = driver.find_element(By.ID, "base1-child_mixin")
assert base1_basevar.text == "basevar1"
base1_child_mixin = driver.find_element(By.ID, "base1-child-mixin")
assert base1_child_mixin.text == "child_mixin" assert base1_child_mixin.text == "child_mixin"
base2_basevar = driver.find_element(By.ID, "base2-basevar") base2_computed_basevar = driver.find_element(By.ID, "base2-computed_basevar")
assert base2_basevar.text == "basevar2" assert base2_computed_basevar.text == "computed_basevar2"
child1_basevar = driver.find_element(By.ID, "child1-basevar") base2_base2 = driver.find_element(By.ID, "base2-base2")
assert child1_basevar.text == "basevar1" assert base2_base2.text == "base2"
child1_computed_basevar = driver.find_element(By.ID, "child1-computed_basevar")
assert child1_computed_basevar.text == "computed_basevar1"
child1_mixin = driver.find_element(By.ID, "child1-computed_mixin")
assert child1_mixin.text == "computed_mixin"
child1_computed_other_mixin = driver.find_element(By.ID, "child1-other-mixin")
assert child1_computed_other_mixin.text == "computed_other_mixin"
child1_computed_child_mixin = driver.find_element(By.ID, "child1-child-mixin")
assert child1_computed_child_mixin.text == "computed_child_mixin"
child1_base1 = driver.find_element(By.ID, "child1-base1")
assert child1_base1.text == "base1"
child1_mixin = driver.find_element(By.ID, "child1-mixin") child1_mixin = driver.find_element(By.ID, "child1-mixin")
assert child1_mixin.text == "mixin" assert child1_mixin.text == "mixin"
child1_other_mixin = driver.find_element(By.ID, "child1-other-mixin") child1_other_mixin = driver.find_element(By.ID, "child1-other_mixin")
assert child1_other_mixin.text == "other_mixin" assert child1_other_mixin.text == "other_mixin"
child1_child_mixin = driver.find_element(By.ID, "child1-child-mixin") child1_child_mixin = driver.find_element(By.ID, "child1-child_mixin")
assert child1_child_mixin.text == "child_mixin" assert child1_child_mixin.text == "child_mixin"
child2_basevar = driver.find_element(By.ID, "child2-basevar") child2_computed_basevar = driver.find_element(By.ID, "child2-computed_basevar")
assert child2_basevar.text == "basevar2" assert child2_computed_basevar.text == "computed_basevar2"
child2_mixin = driver.find_element(By.ID, "child2-computed_mixin")
assert child2_mixin.text == "computed_mixin"
child2_computed_other_mixin = driver.find_element(By.ID, "child2-other-mixin")
assert child2_computed_other_mixin.text == "computed_other_mixin"
child2_computed_child_mixin = driver.find_element(By.ID, "child2-child-mixin")
assert child2_computed_child_mixin.text == "computed_child_mixin"
child2_base2 = driver.find_element(By.ID, "child2-base2")
assert child2_base2.text == "base2"
child2_mixin = driver.find_element(By.ID, "child2-mixin") child2_mixin = driver.find_element(By.ID, "child2-mixin")
assert child2_mixin.text == "mixin" assert child2_mixin.text == "mixin"
child2_other_mixin = driver.find_element(By.ID, "child2-other-mixin") child2_other_mixin = driver.find_element(By.ID, "child2-other_mixin")
assert child2_other_mixin.text == "other_mixin" assert child2_other_mixin.text == "other_mixin"
child2_child_mixin = driver.find_element(By.ID, "child2-child-mixin") child2_child_mixin = driver.find_element(By.ID, "child2-child_mixin")
assert child2_child_mixin.text == "child_mixin" assert child2_child_mixin.text == "child_mixin"
child3_basevar = driver.find_element(By.ID, "child3-basevar") child3_computed_basevar = driver.find_element(By.ID, "child3-computed_basevar")
assert child3_basevar.text == "basevar2" assert child3_computed_basevar.text == "computed_basevar2"
child3_mixin = driver.find_element(By.ID, "child3-computed_mixin")
assert child3_mixin.text == "computed_mixin"
child3_computed_other_mixin = driver.find_element(By.ID, "child3-other-mixin")
assert child3_computed_other_mixin.text == "computed_other_mixin"
child3_computed_childvar = driver.find_element(By.ID, "child3-computed_childvar")
assert child3_computed_childvar.text == "computed_childvar"
child3_computed_child_mixin = driver.find_element(By.ID, "child3-child-mixin")
assert child3_computed_child_mixin.text == "computed_child_mixin"
child3_child3 = driver.find_element(By.ID, "child3-child3")
assert child3_child3.text == "child3"
child3_base2 = driver.find_element(By.ID, "child3-base2")
assert child3_base2.text == "base2"
child3_mixin = driver.find_element(By.ID, "child3-mixin") child3_mixin = driver.find_element(By.ID, "child3-mixin")
assert child3_mixin.text == "mixin" assert child3_mixin.text == "mixin"
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_childvar = driver.find_element(By.ID, "child3-childvar") child3_child_mixin = driver.find_element(By.ID, "child3-child_mixin")
assert child3_childvar.text == "childvar"
child3_child_mixin = driver.find_element(By.ID, "child3-child-mixin")
assert child3_child_mixin.text == "child_mixin" assert child3_child_mixin.text == "child_mixin"