test_lazytree.nim

import sequtils
import unittest
import xmlparser
import xmltree

import lazytree


suite "Recurse over tree":

  test "Only root node":
    let
      rootNode = parseXml("<root />")
      nodeSeq = toSeq(treeIter(rootNode))
    check nodeSeq.len() == 1
    check nodeSeq[0] of XMLNode

  test "Yield immediate child nodes":
    let
      rootNode = parseXml("<root><foo/><bar/><baz/></root>")
      nodeSeq = toSeq(treeIter(rootNode))
    check nodeSeq.len() == 4
    check nodeSeq[0].tag == "root"
    check nodeSeq[0].len() == 3
    check nodeSeq[1].tag == "foo"
    check nodeSeq[2].tag == "bar"
    check nodeSeq[3].tag == "baz"