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

accidentally pushed unfinished changes

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-11-23 11:44:58 +01:00
parent 9ec9ab7929
commit 03eda2e90e
No known key found for this signature in database
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration/test_background_task.py
View File

@ -44,7 +44,7 @@ def BackgroundTask():
@rx.event(background=True) @rx.event(background=True)
async def fast_yielding(self): async def fast_yielding(self):
for _ in range(100): for _ in range(1000):
yield State.increment() yield State.increment()
@rx.event @rx.event
@ -174,6 +174,11 @@ def BackgroundTask():
on_click=State.yield_in_async_with_self, on_click=State.yield_in_async_with_self,
id="yield-in-async-with-self", id="yield-in-async-with-self",
), ),
rx.button(
"Fast Yielding",
on_click=State.fast_yielding,
id="fast-yielding",
),
rx.button("Reset", on_click=State.reset_counter, id="reset"), rx.button("Reset", on_click=State.reset_counter, id="reset"),
) )
@ -397,11 +402,11 @@ def test_fast_yielding(
assert background_task.app_instance is not None assert background_task.app_instance is not None
# get a reference to all buttons # get a reference to all buttons
fast_yielding_button = driver.find_element(By.ID, "yield-increment") fast_yielding_button = driver.find_element(By.ID, "fast-yielding")
# get a reference to the counter # get a reference to the counter
counter = driver.find_element(By.ID, "counter") counter = driver.find_element(By.ID, "counter")
assert background_task._poll_for(lambda: counter.text == "0", timeout=5) assert background_task._poll_for(lambda: counter.text == "0", timeout=5)
fast_yielding_button.click() fast_yielding_button.click()
assert background_task._poll_for(lambda: counter.text == "100", timeout=5) assert background_task._poll_for(lambda: counter.text == "1000", timeout=50)