mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
02ed90c633
Currently, in the various TreeNode messages, and the handlers you'd write to handle them, there's no way to easily know *which* tree sent the message and so which tree the node belongs to. This commit adds public access to the tree reference to the nodes, so that in an event handler the developer can check the tree involved in the event. See #2413.
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from textual.app import App, ComposeResult
|
|
from textual.message import Message
|
|
from textual.widgets import Tree
|
|
|
|
|
|
class MyTree(Tree[None]):
|
|
pass
|
|
|
|
|
|
class TreeApp(App[None]):
|
|
"""Test tree app."""
|
|
|
|
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
self.messages: list[tuple[str, str]] = []
|
|
|
|
def compose(self) -> ComposeResult:
|
|
"""Compose the child widgets."""
|
|
yield MyTree("Root", id="test-tree")
|
|
|
|
def on_mount(self) -> None:
|
|
self.query_one(MyTree).root.add("Child")
|
|
self.query_one(MyTree).focus()
|
|
|
|
def record(
|
|
self,
|
|
event: Tree.NodeSelected[None]
|
|
| Tree.NodeExpanded[None]
|
|
| Tree.NodeCollapsed[None]
|
|
| Tree.NodeHighlighted[None],
|
|
) -> None:
|
|
self.messages.append(
|
|
(event.__class__.__name__, event.node.tree.id or "Unknown")
|
|
)
|
|
|
|
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")
|
|
await pilot.pause()
|
|
assert pilot.app.messages == [
|
|
("NodeExpanded", "test-tree"),
|
|
("NodeSelected", "test-tree"),
|
|
]
|
|
|
|
|
|
async def test_tree_node_selected_message_no_auto() -> None:
|
|
"""Selecting a node should result in only a selected message being emitted."""
|
|
async with TreeApp().run_test() as pilot:
|
|
pilot.app.query_one(MyTree).auto_expand = False
|
|
await pilot.press("enter")
|
|
await pilot.pause()
|
|
assert pilot.app.messages == [("NodeSelected", "test-tree")]
|
|
|
|
|
|
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("space")
|
|
await pilot.pause()
|
|
assert pilot.app.messages == [("NodeExpanded", "test-tree")]
|
|
|
|
|
|
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("space", "space")
|
|
await pilot.pause()
|
|
assert pilot.app.messages == [
|
|
("NodeExpanded", "test-tree"),
|
|
("NodeCollapsed", "test-tree"),
|
|
]
|
|
|
|
|
|
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")
|
|
await pilot.pause()
|
|
assert pilot.app.messages == [
|
|
("NodeExpanded", "test-tree"),
|
|
("NodeSelected", "test-tree"),
|
|
("NodeHighlighted", "test-tree"),
|
|
]
|