mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
df37a9b90a
* Add get_child_by_id and get_widget_by_id * Remove redundant code * Add unit tests for app-level get_child_by_id and get_widget_by_id * Remove redundant test fixture injection * Update CHANGELOG * Enforce uniqueness of ID amongst widget children * Enforce unique widget IDs amongst widgets mounted together * Update CHANGELOG.md * Ensuring unique IDs in a more logical place * Add docstring to NodeList._get_by_id * Dont use duplicate IDs in tests, dont mount 2000 widgets * Mounting less widgets in a unit test * Reword error message * Use lower-level depth first search in get_widget_by_id to break out early
147 lines
5.5 KiB
Python
147 lines
5.5 KiB
Python
import asyncio
|
|
from textual.app import App
|
|
from textual.widget import Widget
|
|
from textual.widgets import Static, Button
|
|
from textual.containers import Container
|
|
|
|
async def test_remove_single_widget():
|
|
"""It should be possible to the only widget on a screen."""
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(Static())
|
|
assert len(pilot.app.screen.children) == 1
|
|
await pilot.app.query_one(Static).remove()
|
|
assert len(pilot.app.screen.children) == 0
|
|
|
|
async def test_many_remove_all_widgets():
|
|
"""It should be possible to remove all widgets on a multi-widget screen."""
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(*[Static() for _ in range(10)])
|
|
assert len(pilot.app.screen.children) == 10
|
|
await pilot.app.query(Static).remove()
|
|
assert len(pilot.app.screen.children) == 0
|
|
|
|
async def test_many_remove_some_widgets():
|
|
"""It should be possible to remove some widgets on a multi-widget screen."""
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(*[Static(classes=f"is-{n%2}") for n in range(10)])
|
|
assert len(pilot.app.screen.children) == 10
|
|
await pilot.app.query(".is-0").remove()
|
|
assert len(pilot.app.screen.children) == 5
|
|
|
|
async def test_remove_branch():
|
|
"""It should be possible to remove a whole branch in the DOM."""
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Static()
|
|
)
|
|
)
|
|
)
|
|
)
|
|
),
|
|
Static(),
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Static()
|
|
)
|
|
)
|
|
)
|
|
)
|
|
),
|
|
)
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 13
|
|
await pilot.app.screen.children[0].remove()
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 7
|
|
|
|
async def test_remove_overlap():
|
|
"""It should be possible to remove an overlapping collection of widgets."""
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Static()
|
|
)
|
|
)
|
|
)
|
|
)
|
|
),
|
|
Static(),
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Container(
|
|
Static()
|
|
)
|
|
)
|
|
)
|
|
)
|
|
),
|
|
)
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 13
|
|
await pilot.app.query(Container).remove()
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 1
|
|
|
|
async def test_remove_move_focus():
|
|
"""Removing a focused widget should settle focus elsewhere."""
|
|
async with App().run_test() as pilot:
|
|
buttons = [ Button(str(n)) for n in range(10)]
|
|
await pilot.app.mount(Container(*buttons[:5]), Container(*buttons[5:]))
|
|
assert len(pilot.app.screen.children) == 2
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 12
|
|
assert pilot.app.focused is None
|
|
await pilot.press( "tab" )
|
|
assert pilot.app.focused is not None
|
|
assert pilot.app.focused == buttons[0]
|
|
await pilot.app.screen.children[0].remove()
|
|
assert len(pilot.app.screen.children) == 1
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 6
|
|
assert pilot.app.focused is not None
|
|
assert pilot.app.focused == buttons[9]
|
|
|
|
async def test_widget_remove_order():
|
|
"""A Widget.remove of a top-level widget should cause bottom-first removal."""
|
|
|
|
removals: list[str] = []
|
|
|
|
class Removable(Container):
|
|
def on_unmount( self, _ ):
|
|
removals.append(self.id if self.id is not None else "unknown")
|
|
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(
|
|
Removable(Removable(Removable(id="grandchild"), id="child"), id="parent")
|
|
)
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 3
|
|
await pilot.app.screen.children[0].remove()
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 0
|
|
assert removals == ["grandchild", "child", "parent"]
|
|
|
|
async def test_query_remove_order():
|
|
"""A DOMQuery.remove of a top-level widget should cause bottom-first removal."""
|
|
|
|
removals: list[str] = []
|
|
|
|
class Removable(Container):
|
|
def on_unmount( self, _ ):
|
|
removals.append(self.id if self.id is not None else "unknown")
|
|
|
|
async with App().run_test() as pilot:
|
|
await pilot.app.mount(
|
|
Removable(Removable(Removable(id="grandchild"), id="child"), id="parent")
|
|
)
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 3
|
|
await pilot.app.query(Removable).remove()
|
|
assert len(pilot.app.screen.walk_children(with_self=False)) == 0
|
|
assert removals == ["grandchild", "child", "parent"]
|