/*
 * Copyright (c) 2008, Aaron Digulla
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 * 
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in
 *       the documentation and/or other materials provided with the
 *       distribution.
 *     * Neither the name of Aaron Digulla nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package de.pdark.decentxml;

import static org.junit.Assert.*;

import org.junit.Test;

public class TreeIteratorTest
{
    @Test
    public void testEmptyDocument () throws Exception
    {
        Document doc = new Document ();
        TreeIterator iter = new TreeIterator (doc);
        assertFalse (iter.hasNext ());
    }

    @Test
    public void testEmptyElement () throws Exception
    {
        Element e = new Element ("e");
        TreeIterator iter = new TreeIterator (e);
        assertFalse (iter.hasNext ());
    }
    
    @Test
    public void testSimpleDocument () throws Exception
    {
        Element root = new Element ("root");
        Document doc = new Document (root);
        
        TreeIterator iter = new TreeIterator (doc);
        assertTrue (iter.hasNext ());
        
        Node n = iter.next ();
        assertEquals (root, n);
        assertFalse (iter.hasNext ());
    }
    
    @Test
    public void testComplexDocument () throws Exception
    {
        Document doc = new Document (
            new Text ("\n"),
            new Element ("root").addNodes (
                new Text ("\n    "),
                new Comment (" comment "),
                new Text ("\n    "),
                new Element ("e").addNodes (
                    new Element ("e2")
                ),
                new Text ("\n")
            ),
            new Text ("\n")
        );
        doc.setEncoding (XMLInputStreamReader.ENCODING_UTF_8);
        
        StringBuilder result = toString (new TreeIterator (doc));
        
        assertEquals (
            "/ XMLDeclaration \"<?xml version=\"1.0\" encoding=\"UTF-8\"?>\"\n" + 
            "/ Text \"\\n\"\n" + 
            "/ Text \"\\n\"\n" + 
            "/root Element \"<root>\\n    <!-- comment -->\\n    <e><e2/></e>\\n</root>\"\n" + 
            "/root Text \"\\n    \"\n" + 
            "/root Comment \"<!-- comment -->\"\n" + 
            "/root Text \"\\n    \"\n" + 
            "/root/e Element \"<e><e2/></e>\"\n" + 
            "/root/e Element \"<e2/>\"\n" + 
            "/root Text \"\\n\"\n" + 
            "/ Text \"\\n\"",
            result.toString ());
    }

    public StringBuilder toString (TreeIterator iter)
    {
        StringBuilder result = new StringBuilder ();
        String delim = "";
        while (iter.hasNext ())
        {
            Node n = iter.next ();
            result.append (delim);
            delim = "\n";
            result.append (iter.getPath ());
            result.append (" ");
            result.append (n.getClass ().getSimpleName ());
            result.append (" \"");
            result.append (n.toXML ().replaceAll ("\n", "\\\\n"));
            result.append ("\"");
        }
        return result;
    }
}