XmlResultFormatter.java :  » Wiki-Engine » fitnesse » fitnesse » runner » Java Open Source

// Copyright (C) 2003,2004,2005 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the GNU General Public License version 2 or later.
package fitnesse.runner;

import java.io.*;
import fit.Counts;
import fitnesse.components.*;
import fitnesse.util.*;
import fitnesse.components.XmlWriter;
import org.w3c.dom.*;

public class XmlResultFormatter implements ResultFormatter
{
  private ContentBuffer buffer;
  private Document document;
  private boolean closed = false;
  private byte[] tailBytes;

  public XmlResultFormatter(String host, String rootPath) throws Exception
  {
    buffer = new ContentBuffer(".xml");
    createDocument(host, rootPath);
    writeDocumentHeader();
  }

  private void writeDocumentHeader() throws Exception
  {
    ByteArrayOutputStream output = new ByteArrayOutputStream();
    XmlWriter writer = new XmlWriter(output);
    writer.write(document);
    writer.close();
    String xmlText = output.toString();
    int endIndex = xmlText.indexOf("</testResults>");
    String head = xmlText.substring(0, endIndex);
    String tail = xmlText.substring(endIndex);
    tailBytes = tail.getBytes();
    buffer.append(head);
  }

  private void createDocument(String host, String rootPath) throws Exception
  {
    document = XmlUtil.newDocument();
    Element root = document.createElement("testResults");
    document.appendChild(root);
    XmlUtil.addTextNode(document, root, "host", host);
    XmlUtil.addTextNode(document, root, "rootPath", rootPath);
  }

  public void acceptResult(PageResult result) throws Exception
  {
    Element resultElement = document.createElement("result");
    XmlUtil.addTextNode(document, resultElement, "relativePageName", result.title());
    XmlUtil.addCdataNode(document, resultElement, "content", result.content());
    resultElement.appendChild(makeCountsElement("counts", result.counts()));
    writeElement(resultElement);
  }

  public void acceptFinalCount(Counts count) throws Exception
  {
    Element countsElement = makeCountsElement("finalCounts", count);
    writeElement(countsElement);
  }

  public int getByteCount() throws Exception
  {
    close();
    return buffer.getSize();
  }

  public InputStream getResultStream() throws Exception
  {
    close();
    return buffer.getInputStream();
  }

  private void close() throws Exception
  {
    if(!closed)
    {
      buffer.append(tailBytes);
      closed = true;
    }
  }

  private void writeElement(Element resultElement) throws Exception
  {
    ByteArrayOutputStream output = new ByteArrayOutputStream();
    XmlWriter writer = new XmlWriter(output);
    writer.write(resultElement, 1);
    writer.close();
    buffer.append(output.toByteArray());
  }

  private Element makeCountsElement(String name, Counts counts)
  {
    Element countsElement = document.createElement(name);
    XmlUtil.addTextNode(document, countsElement, "right", counts.right + "");
    XmlUtil.addTextNode(document, countsElement, "wrong", counts.wrong + "");
    XmlUtil.addTextNode(document, countsElement, "ignores", counts.ignores + "");
    XmlUtil.addTextNode(document, countsElement, "exceptions", counts.exceptions + "");
    return countsElement;
  }
}