diff --git a/tests/tree/test_tree_messages.py b/tests/tree/test_tree_messages.py index 67620d70e..df7442d31 100644 --- a/tests/tree/test_tree_messages.py +++ b/tests/tree/test_tree_messages.py @@ -48,23 +48,26 @@ async def test_tree_node_selected_message() -> None: assert pilot.app.messages == ["NodeExpanded", "NodeSelected"] +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") + assert pilot.app.messages == ["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"] + await pilot.press("space") + assert pilot.app.messages == ["NodeExpanded"] 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", - ] + await pilot.press("space", "space") + assert pilot.app.messages == ["NodeExpanded", "NodeCollapsed"] async def test_tree_node_highlighted_message() -> None: