Merge pull request #1528 from davep/tree-node-highlighted-message

Add `Tree.NodeHighlighted` message

tests/tree/test_tree_messages.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+from typing import Any
+from textual.app import App, ComposeResult
+from textual.widgets import Tree
+from textual.message import Message
+
+
+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[str] = []
+
+    def compose(self) -> ComposeResult:
+        """Compose the child widgets."""
+        yield MyTree("Root")
+
+    def on_mount(self) -> None:
+        self.query_one(MyTree).root.add("Child")
+        self.query_one(MyTree).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")
+        await pilot.pause(2 / 100)
+        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")
+        await pilot.pause(2 / 100)
+        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")
+        await pilot.pause(2 / 100)
+        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")
+        await pilot.pause(2 / 100)
+        assert pilot.app.messages == ["NodeExpanded", "NodeSelected", "NodeHighlighted"]