1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
|
"""Tests for https://github.com/Textualize/textual/issues/3615"""
from asyncio import sleep
from textual import work
from textual.app import App
from textual.command import Hit, Hits, Provider
class SimpleSource(Provider):
async def search(self, query: str) -> Hits:
def goes_nowhere_does_nothing() -> None:
pass
for _ in range(100):
yield Hit(1, query, goes_nowhere_does_nothing, query)
class CommandPaletteNoWorkerApp(App[None]):
COMMANDS = {SimpleSource}
async def test_no_command_palette_worker_droppings() -> None:
"""The command palette should not leave any workers behind.."""
async with CommandPaletteNoWorkerApp().run_test() as pilot:
assert len(pilot.app.workers) == 0
pilot.app.action_command_palette()
await pilot.press("a", "enter")
assert len(pilot.app.workers) == 0
class CommandPaletteWithWorkerApp(App[None]):
COMMANDS = {SimpleSource}
def on_mount(self) -> None:
self.innocent_worker()
@work
async def innocent_worker(self) -> None:
while True:
await sleep(1)
async def test_innocent_worker_is_untouched() -> None:
"""Using the command palette should not halt other workers."""
async with CommandPaletteWithWorkerApp().run_test() as pilot:
assert len(pilot.app.workers) > 0
pilot.app.action_command_palette()
await pilot.press("a", "enter")
assert len(pilot.app.workers) > 0
|