Files
textual/tests/command_palette/test_worker_interference.py

51 lines
1.4 KiB
Python

"""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