Modify Tree message unit tests to take new action in to account

See #1433.
This commit is contained in:
Dave Pearson
2023-01-31 13:32:46 +00:00
parent 2086b713db
commit 6db5217f3e

View File

@@ -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: