extern crate rose_tree;

use rose_tree::RoseTree;

struct Weight;

#[test]
fn walk_children() {
    let (mut tree, root) = RoseTree::<Weight, u32>::new(Weight);
    let a = tree.add_child(root, Weight);
    let b = tree.add_child(root, Weight);
    let c = tree.add_child(root, Weight);

    let mut child_walker = tree.walk_children(root);
    assert_eq!(Some(c), child_walker.next(&tree));
    assert_eq!(Some(b), child_walker.next(&tree));
    assert_eq!(Some(a), child_walker.next(&tree));
    assert_eq!(None, child_walker.next(&tree));

    let d = tree.add_child(b, Weight);
    let e = tree.add_child(b, Weight);
    let f = tree.add_child(b, Weight);

    child_walker = tree.walk_children(b);
    assert_eq!(Some(f), child_walker.next(&tree));
    assert_eq!(Some(e), child_walker.next(&tree));
    assert_eq!(Some(d), child_walker.next(&tree));
    assert_eq!(None, child_walker.next(&tree));
}

#[test]
fn walk_siblings() {
    let (mut tree, root) = RoseTree::<Weight, u32>::new(Weight);
    let a = tree.add_child(root, Weight);
    let b = tree.add_child(root, Weight);
    let c = tree.add_child(root, Weight);
    let d = tree.add_child(root, Weight);

    let mut sibling_walker = tree.walk_siblings(a);
    assert_eq!(Some(d), sibling_walker.next(&tree));
    assert_eq!(Some(c), sibling_walker.next(&tree));
    assert_eq!(Some(b), sibling_walker.next(&tree));
    assert_eq!(None, sibling_walker.next(&tree));
}