mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from typing import Any
|
|
from textual.app import App, ComposeResult
|
|
from textual.widgets import Tree
|
|
from textual.message import Message
|
|
|
|
|
|
class TreeApp(App[None]):
|
|
"""Test tree app."""
|
|
|
|
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
self.messages: list[str] = []
|
|
|
|
def compose(self) -> ComposeResult:
|
|
"""Compose the child widgets."""
|
|
yield Tree[None]("Root")
|
|
|
|
def on_mount(self) -> None:
|
|
self.query_one(Tree[None]).root.add("Child")
|
|
self.query_one(Tree[None]).focus()
|
|
|
|
def record(self, event: Message) -> None:
|
|
self.messages.append(event.__class__.__name__)
|
|
|
|
def on_tree_node_selected(self, event: Tree.NodeSelected[None]) -> None:
|
|
self.record(event)
|
|
|
|
def on_tree_node_expanded(self, event: Tree.NodeExpanded[None]) -> None:
|
|
self.record(event)
|
|
|
|
def on_tree_node_collapsed(self, event: Tree.NodeCollapsed[None]) -> None:
|
|
self.record(event)
|
|
|
|
def on_tree_node_highlighted(self, event: Tree.NodeHighlighted[None]) -> None:
|
|
self.record(event)
|
|
|
|
|
|
async def test_tree_node_selected_message() -> None:
|
|
"""Selecting a node should result in a selected message being emitted."""
|
|
async with TreeApp().run_test() as pilot:
|
|
await pilot.press("enter")
|
|
assert pilot.app.messages == ["NodeExpanded", "NodeSelected"]
|
|
|
|
|
|
async def test_tree_node_expanded_message() -> None:
|
|
"""Expanding a node should result in an expanded message being emitted."""
|
|
async with TreeApp().run_test() as pilot:
|
|
await pilot.press("enter")
|
|
assert pilot.app.messages == ["NodeExpanded", "NodeSelected"]
|
|
|
|
|
|
async def test_tree_node_collapsed_message() -> None:
|
|
"""Collapsing a node should result in a collapsed message being emitted."""
|
|
async with TreeApp().run_test() as pilot:
|
|
await pilot.press("enter", "enter")
|
|
assert pilot.app.messages == [
|
|
"NodeExpanded",
|
|
"NodeSelected",
|
|
"NodeCollapsed",
|
|
"NodeSelected",
|
|
]
|
|
|
|
|
|
async def test_tree_node_highlighted_message() -> None:
|
|
"""Highlighting a node should result in a highlighted message being emitted."""
|
|
async with TreeApp().run_test() as pilot:
|
|
await pilot.press("enter", "down")
|
|
assert pilot.app.messages == ["NodeExpanded", "NodeSelected", "NodeHighlighted"]
|