Harness

package com.goldencode.harness;

import com.goldencode.harness.test.Test;

import java.io.File;

import java.util.List;

import javax.xml.parsers.DocumentBuilder;

import javax.xml.parsers.DocumentBuilderFactory;

import javax.xml.transform.Transformer;

import javax.xml.transform.TransformerFactory;

import javax.xml.transform.dom.DOMSource;

import javax.xml.transform.stream.StreamResult;

import org.w3c.dom.Attr;

import org.w3c.dom.Document;

import org.w3c.dom.Element;

public class JunitGenerator {

   /** Test plan for which to generate a report. */

   private final TestPlan testPlan;

   /** Object in which to store the details of failures. */

   private final FailureNotification rc;

   /**

    * Create an instance.

    *

    * @param    testPlan

    *           The test plan for which to generate a report.

    * @param    rc

    *           Object in which to store the details of failures.

    */

   public JunitGenerator(TestPlan testPlan, FailureNotification rc)

   {

      this.testPlan = testPlan;

      this.rc       = rc;

   }

   public void generate() {

      String outputFileName = "results_" + testPlan.getSourceFileName();

      try {

         DocumentBuilderFactory documentFactory = DocumentBuilderFactory.newInstance();

         DocumentBuilder documentBuilder = documentFactory.newDocumentBuilder();

         Document document = documentBuilder.newDocument();

         // root element

         Element testSuite = document.createElement("testsuite");

         document.appendChild(testSuite);

         int total = 0, passed = 0, failed = 0, skipped = 0;

         double totalDuration = 0;

         List<TestSet> sets = testPlan.getSets();

         for (TestSet set : sets) {

            List<Test> tests = set.getTests();

            for (Test test : tests) {

               total++;

               Element testCase = document.createElement("testcase");

               Attr cName = document.createAttribute("classname");

               cName.setValue(test.getName());

               testCase.setAttributeNode(cName);

               Attr nName = document.createAttribute("name");

               nName.setValue(test.getDescription());

               testCase.setAttributeNode(nName);

               Attr tName = document.createAttribute("time");

               totalDuration += test.getElapsed();

               tName.setValue(String.valueOf(((double) test.getElapsed()) / 1000));

               testCase.setAttributeNode(tName);

               switch (test.getStatus()) {

                  case FAILED:

                     if(!test.getFailureMode().equals(FailureMode.SILENT)) {

                        Element failure = document.createElement("failure");

                        Attr failType = document.createAttribute("type");

                        failType.setValue(test.getReason().substring(0, 10));

                        failure.setAttributeNode(failType);

                        failure.appendChild(document.createTextNode(test.getReason()));

                        testCase.appendChild(failure);

                        failed++;

                     }

                  case NOT_RUN:

                     skipped++;

                     break;

               }

               testSuite.appendChild(testCase);

            }

         }

         Attr testsTotal = document.createAttribute("tests");

         testsTotal.setValue(String.valueOf(total));

         testSuite.setAttributeNode(testsTotal);

         Attr testsFail = document.createAttribute("failures");

         testsFail.setValue(String.valueOf(failed));

         testSuite.setAttributeNode(testsFail);

         Attr testsSkip = document.createAttribute("skipped");

         testsSkip.setValue(String.valueOf(skipped));

         testSuite.setAttributeNode(testsSkip);

         Attr testsDuration = document.createAttribute("time");

         testsDuration.setValue(String.valueOf(totalDuration / 1000));

         testSuite.setAttributeNode(testsDuration);

         TransformerFactory transformerFactory = TransformerFactory.newInstance();

         Transformer transformer = transformerFactory.newTransformer();

         DOMSource domSource = new DOMSource(document);

         StreamResult streamResult = new StreamResult(new File(outputFileName));

         transformer.transform(domSource, streamResult);

      }catch (Exception ex) {

         rc.setResult(ExitCode.UNKNOWN_FAILURE);

         rc.setDetails(outputFileName);

         return;

      }

   }

}