textual/tests/tree/test_tree_get_node_by_id.py

import pytest
from typing import cast
from textual.widgets import Tree
from textual.widgets._tree import NodeID


def test_get_tree_node_by_id() -> None:
    """It should be possible to get a TreeNode by its ID."""
    tree = Tree[None]("Anakin")
    child = tree.root.add("Leia")
    grandchild = child.add("Ben")
    assert tree.get_node_by_id(tree.root.id).id == tree.root.id
    assert tree.get_node_by_id(child.id).id == child.id
    assert tree.get_node_by_id(grandchild.id).id == grandchild.id
    with pytest.raises(Tree.UnknownNodeID):
        tree.get_node_by_id(cast(NodeID, grandchild.id + 1000))