// Copyright (C) 2003-2009 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the CPL Common Public License version 1.0.
package util;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.PrintWriter;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;

import org.w3c.dom.Document;

public class XmlWriterTest extends RegexTestCase {
  private ByteArrayOutputStream output;
  private Document doc;

  static String sampleXml = null;

  static {
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    PrintWriter writer = new PrintWriter(out);
    writer.println("<?xml version=\"1.0\"?>");
    writer.println("<rootElement version=\"2.0\">");
    writer.println("\t<emptytElement dire=\"straights\" strawberry=\"alarmclock\"/>");
    writer.println("\t<fullElement>");
    writer.println("\t\t<childElement/>");
    writer.println("\t</fullElement>");
    writer.println("\t<text>some text</text>");
    writer.println("\t<cdata><![CDATA[<>&;]]></cdata>");
    writer.println("</rootElement>");
    writer.flush();
    sampleXml = new String(out.toByteArray());
  }

  public void setUp() throws Exception {
    DocumentBuilder builder = DocumentBuilderFactory.newInstance().newDocumentBuilder();
    doc = builder.parse(new ByteArrayInputStream(sampleXml.getBytes()));
    output = new ByteArrayOutputStream();
  }

  public void tearDown() throws Exception {
  }

  public void testAll() throws Exception {
    String results = writeXml(doc);

    assertEquals(sampleXml, results);
  }

  private String writeXml(Document doc) throws Exception {
    XmlWriter writer = new XmlWriter(output);
    writer.write(doc);
    writer.flush();
    writer.close();

    String results = new String(output.toByteArray());
    return results;
  }
}