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

add a slightly more complicated test

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-10-15 17:12:38 -07:00
parent 233033cb7d
commit b17ed24155
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/units/components/test_component.py
View File

@ -54,6 +54,12 @@ def test_state():
def do_something_with_int(self, arg: int): def do_something_with_int(self, arg: int):
pass pass
def do_something_with_list_int(self, arg: List[int]):
pass
def do_something_with_list_str(self, arg: List[str]):
pass
return TestState return TestState
@ -109,6 +115,7 @@ def component2() -> Type[Component]:
"on_open": identity_event(bool), "on_open": identity_event(bool),
"on_close": identity_event(bool), "on_close": identity_event(bool),
"on_user_visited_count_changed": identity_event(int), "on_user_visited_count_changed": identity_event(int),
"on_user_list_changed": identity_event(List[str]),
} }
def _get_imports(self) -> ParsedImportDict: def _get_imports(self) -> ParsedImportDict:
@ -594,7 +601,13 @@ def test_get_event_triggers(component1, component2):
assert component1().get_event_triggers().keys() == default_triggers assert component1().get_event_triggers().keys() == default_triggers
assert ( assert (
component2().get_event_triggers().keys() component2().get_event_triggers().keys()
== {"on_open", "on_close", "on_prop_event", "on_user_visited_count_changed"} == {
"on_open",
"on_close",
"on_prop_event",
"on_user_visited_count_changed",
"on_user_list_changed",
}
| default_triggers | default_triggers
) )
@ -936,10 +949,15 @@ def test_invalid_event_handler_args(component2, test_state):
component2.create( component2.create(
on_user_visited_count_changed=test_state.do_something_with_bool on_user_visited_count_changed=test_state.do_something_with_bool
) )
with pytest.raises(EventHandlerArgTypeMismatch):
component2.create(on_user_list_changed=test_state.do_something_with_int)
with pytest.raises(EventHandlerArgTypeMismatch):
component2.create(on_user_list_changed=test_state.do_something_with_list_int)
component2.create(on_open=test_state.do_something_with_int) component2.create(on_open=test_state.do_something_with_int)
component2.create(on_open=test_state.do_something_with_bool) component2.create(on_open=test_state.do_something_with_bool)
component2.create(on_user_visited_count_changed=test_state.do_something_with_int) component2.create(on_user_visited_count_changed=test_state.do_something_with_int)
component2.create(on_user_list_changed=test_state.do_something_with_list_str)
# lambda cannot return weird values. # lambda cannot return weird values.
with pytest.raises(ValueError): with pytest.raises(ValueError):