34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
 
|
"""R2 · R5: rows appear while calls are still running; picking a body file starts the run."""
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from textual.widgets import DataTable, OptionList
|
||
|
|
|
||
|
|
from tbench.app import BenchApp
|
||
|
|
from tbench.cli import settings_from
|
||
|
|
|
||
|
|
|
||
|
|
def settings(url, n, body_file=None):
|
||
|
|
return settings_from(type("A", (), {"url": url, "n": n, "body_file": body_file})())
|
||
|
|
|
||
|
|
|
||
|
|
async def test_rows_arrive_one_by_one(server, body_file):
|
||
|
|
app = BenchApp(settings(server.base + "?delay=0.5", 6, str(body_file)), {})
|
||
|
|
async with app.run_test() as pilot:
|
||
|
|
await asyncio.sleep(1.3)
|
||
|
|
partial = app.query_one("#calls", DataTable).row_count
|
||
|
|
assert 1 <= partial < 6
|
||
|
|
await asyncio.sleep(2.5)
|
||
|
|
await pilot.pause()
|
||
|
|
assert app.query_one("#calls", DataTable).row_count == 6
|
||
|
|
assert "done 6/6" in str(app.query_one("#summary").render())
|
||
|
|
|
||
|
|
|
||
|
|
async def test_pick_a_body(server, body_file):
|
||
|
|
app = BenchApp(settings(server.base, 2), {}, pick=str(body_file.parent))
|
||
|
|
async with app.run_test() as pilot:
|
||
|
|
assert app.query_one(OptionList).option_count == 1
|
||
|
|
await pilot.press("enter")
|
||
|
|
await asyncio.sleep(0.5)
|
||
|
|
await pilot.pause()
|
||
|
|
assert app.query_one("#calls", DataTable).row_count == 2
|
||
|
|
assert server.bodies
|