DefaultTestResultListener.java

  1. /**
  2.  * Powerunit - A JDK1.8 test framework
  3.  * Copyright (C) 2014 Mathieu Boretti.
  4.  *
  5.  * This file is part of Powerunit
  6.  *
  7.  * Powerunit is free software: you can redistribute it and/or modify
  8.  * it under the terms of the GNU General Public License as published by
  9.  * the Free Software Foundation, either version 3 of the License, or
  10.  * (at your option) any later version.
  11.  *
  12.  * Powerunit is distributed in the hope that it will be useful,
  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15.  * GNU General Public License for more details.
  16.  *
  17.  * You should have received a copy of the GNU General Public License
  18.  * along with Powerunit. If not, see <http://www.gnu.org/licenses/>.
  19.  */
  20. package ch.powerunit.impl;

  22. import java.io.ByteArrayOutputStream;
  23. import java.io.File;
  24. import java.io.PrintStream;
  25. import java.util.HashMap;
  26. import java.util.Map;

  28. import javax.xml.bind.JAXBContext;
  29. import javax.xml.bind.JAXBException;

  31. import ch.powerunit.TestContext;
  32. import ch.powerunit.TestResultListener;
  33. import ch.powerunit.report.Error;
  34. import ch.powerunit.report.Failure;
  35. import ch.powerunit.report.Testcase;
  36. import ch.powerunit.report.Testsuite;
  37. import ch.powerunit.report.Testsuites;

  39. /**
  40.  * @author borettim
  41.  *
  42.  */
  43. public class DefaultTestResultListener<T> implements TestResultListener<T> {

  45.     public DefaultTestResultListener(String targetFolder) {
  46.         this(targetFolder, null);
  47.     }

  49.     public DefaultTestResultListener(String targetFolder,
  50.             PrintStream outputConsole) {
  51.         this(targetFolder, outputConsole, null);
  52.     }

  54.     public DefaultTestResultListener(String targetFolder,
  55.             PrintStream outputConsole, PowerUnit mbean) {
  56.         this.outputConsole = outputConsole;
  57.         this.targetFolder = new File(targetFolder);
  58.         this.targetFolder.mkdirs();
  59.         this.mbean = mbean;
  60.     }

  62.     private static final JAXBContext JAXB_CONTEXT;

  64.     private final StringBuilder resumedSucess = new StringBuilder();

  66.     private final StringBuilder resumedFailure = new StringBuilder();

  68.     private final StringBuilder resumedSkipped = new StringBuilder();

  70.     /**
  71.      * @return the resumed
  72.      */
  73.     public String getResumed() {
  74.         return "Success tests:\n" + resumedSucess.toString()
  75.                 + "\n\nSkipped tests:\n" + resumedSkipped.toString()
  76.                 + "\n\nFailed tests:\n" + resumedFailure.toString() + "\n";
  77.     }

  79.     /**
  80.      * @return the resumedSucess
  81.      */
  82.     public String getResumedSucess() {
  83.         return resumedSucess.toString();
  84.     }

  86.     /**
  87.      * @return the resumedFailure
  88.      */
  89.     public String getResumedFailure() {
  90.         return resumedFailure.toString();
  91.     }

  93.     /**
  94.      * @return the resumedSkipped
  95.      */
  96.     public String getResumedSkipped() {
  97.         return resumedSkipped.toString();
  98.     }

  100.     static {
  101.         try {
  102.             JAXB_CONTEXT = JAXBContext.newInstance(Testsuites.class);
  103.         } catch (JAXBException e) {
  104.             throw new IllegalArgumentException("Unable to setup jaxb "
  105.                     + e.getMessage(), e);
  106.         }
  107.     }

  109.     private boolean error = false;

  111.     private final PrintStream outputConsole;

  113.     private final Map<String, Testsuite> result = new HashMap<>();

  115.     private final Map<String, Testsuites> results = new HashMap<>();

  117.     private final Map<String, Map<String, Testcase>> resultcase = new HashMap<>();

  119.     private final File targetFolder;

  121.     private final PowerUnit mbean;

  123.     private void printf(String format, Object... args) {
  124.         if (outputConsole != null) {
  125.             outputConsole.printf(format, args);
  126.         }
  127.     }

  129.     @Override
  130.     public void notifySetStart(String setName, String parameters) {
  131.         Testsuites tss = new Testsuites();
  132.         tss.setDisabled(0);
  133.         tss.setErrors(0);
  134.         tss.setFailures(0);
  135.         tss.setName(setName);
  136.         tss.setTests(0);
  137.         tss.setTime(0L);
  138.         results.put(setName, tss);

  140.         Testsuite ts = new Testsuite();
  141.         result.put(setName, ts);
  142.         ts.setName(setName.replace('$', '.'));
  143.         ts.setDisabled(0);
  144.         ts.setErrors(0);
  145.         ts.setFailures(0);
  146.         ts.setTests(0);
  147.         ts.setTime(0L);
  148.         resultcase.put(setName, new HashMap<>());

  150.     }

  152.     @Override
  153.     public void notifySetEnd(String setName, String parameters) {
  154.         try {
  155.             File target = new File(targetFolder, setName + ".xml");
  156.             printf("Pushing test results into %1$s%n", target.getAbsolutePath());
  157.             Object o = results.get(setName);
  158.             if (results.get(setName).getTestsuite().isEmpty()) {
  159.                 o = result.get(setName);
  160.             }
  161.             if (mbean != null) {
  162.                 ByteArrayOutputStream str = new ByteArrayOutputStream();
  163.                 JAXB_CONTEXT.createMarshaller().marshal(o, str);
  164.                 mbean.addResult(new String(str.toByteArray()));
  165.             }
  166.             JAXB_CONTEXT.createMarshaller().marshal(o, target);
  167.         } catch (JAXBException e) {
  168.             throw new IllegalArgumentException("Unable to setup jaxb "
  169.                     + e.getMessage(), e);
  170.         }
  171.     }

  173.     @Override
  174.     public void notifyStart(TestContext<T> context) {
  175.         String setName = context.getSetName()
  176.                 + (context.getParameterName() == null ? "" : context
  177.                         .getParameterName());
  178.         Testsuite ts = result.get(setName);
  179.         Testcase tc = new Testcase();
  180.         tc.setName(context.getLocalTestName()
  181.                 + (context.getParameterName() == null ? "" : "["
  182.                         + context.getParameterName() + "]"));
  183.         tc.setClassname(context.getTestSuiteObject().getClass()
  184.                 .getCanonicalName());
  185.         resultcase.get(setName).put(context.getFullTestName(), tc);
  186.         ts.getTestcase().add(tc);
  187.         tc.setTime(System.currentTimeMillis());
  188.         printf("Start of test %1$s%n", context.getFullTestName());
  189.     }

  191.     @Override
  192.     public void notifySuccess(TestContext<T> context) {
  193.         String setName = context.getSetName()
  194.                 + (context.getParameterName() == null ? "" : context
  195.                         .getParameterName());
  196.         long end = System.currentTimeMillis();
  197.         Testsuite ts = result.get(setName);
  198.         Testcase tc = resultcase.get(setName).get(context.getFullTestName());
  199.         tc.setTime((end - tc.getTime()) / 1000);
  200.         ts.setTime(ts.getTime() + tc.getTime());
  201.         ts.setTests(ts.getTests() + 1);
  202.         printf("Success of test %1$s%n", context.getFullTestName());
  203.         resumedSucess.append("\t").append(context.getLocalTestName())
  204.                 .append(" of ").append(context.getSetName()).append("\n");
  205.     }

  207.     @Override
  208.     public void notifyFailure(TestContext<T> context, Throwable cause) {
  209.         String setName = context.getSetName()
  210.                 + (context.getParameterName() == null ? "" : context
  211.                         .getParameterName());
  212.         long end = System.currentTimeMillis();
  213.         Testsuite ts = result.get(setName);
  214.         Testcase tc = resultcase.get(setName).get(context.getFullTestName());
  215.         tc.setTime((end - tc.getTime()) / 1000);
  216.         ts.setTime(ts.getTime() + tc.getTime());
  217.         error = true;
  218.         ts.setFailures(ts.getFailures() + 1);

  220.         Failure f = new Failure();
  221.         tc.getFailure().add(f);
  222.         f.setType(cause.getClass().getCanonicalName());
  223.         f.setMessage(cause.getMessage());
  224.         StringBuilder stack = new StringBuilder("" + cause.getMessage())
  225.                 .append('\n');
  226.         completeStack(cause, stack);
  227.         f.setContent(stack.toString());
  228.         printf("Failure of test %1$s because of %2$s%n",
  229.                 context.getFullTestName(), cause.getMessage());
  230.         resumedFailure.append("\t").append(context.getLocalTestName())
  231.                 .append(" of ").append(context.getSetName())
  232.                 .append(" caused by ").append(cause.getMessage()).append("\n");
  233.     }

  235.     @Override
  236.     public void notifySkipped(TestContext<T> context) {
  237.         String setName = context.getSetName()
  238.                 + (context.getParameterName() == null ? "" : context
  239.                         .getParameterName());
  240.         long end = System.currentTimeMillis();
  241.         Testsuite ts = result.get(setName);
  242.         Testcase tc = resultcase.get(setName).get(context.getFullTestName());
  243.         tc.setTime((end - tc.getTime()) / 1000);
  244.         ts.setTime(ts.getTime() + tc.getTime());
  245.         ts.setDisabled(ts.getDisabled() + 1);

  247.         tc.setSkipped("Skipped");
  248.         printf("Skip of test %1$s%n", context.getFullTestName());
  249.         resumedSkipped.append("\t").append(context.getLocalTestName())
  250.                 .append(" of ").append(context.getSetName()).append("\n");
  251.     }

  253.     @Override
  254.     public void notifyError(TestContext<T> context, Throwable cause) {
  255.         String setName = context.getSetName()
  256.                 + (context.getParameterName() == null ? "" : context
  257.                         .getParameterName());
  258.         long end = System.currentTimeMillis();
  259.         Testsuite ts = result.get(setName);
  260.         Testcase tc = resultcase.get(setName).get(context.getFullTestName());
  261.         tc.setTime((end - tc.getTime()) / 1000);
  262.         ts.setTime(ts.getTime() + tc.getTime());
  263.         error = true;
  264.         ts.setErrors(ts.getErrors() + 1);

  266.         Error e = new Error();
  267.         tc.getError().add(e);
  268.         e.setType(cause.getClass().getCanonicalName());
  269.         e.setMessage(cause.getMessage());
  270.         StringBuilder stack = new StringBuilder("" + cause.getMessage())
  271.                 .append('\n');
  272.         completeStack(cause, stack);
  273.         e.setContent(stack.toString());
  274.         printf("Error of test %1$s because of %2$s%n",
  275.                 context.getFullTestName(), cause.getMessage());
  276.         resumedFailure.append("\t").append(context.getLocalTestName())
  277.                 .append(" of ").append(context.getSetName())
  278.                 .append(" caused by ").append(cause.getMessage()).append("\n");
  279.     }

  281.     /**
  282.      * @param cause
  283.      * @param stack
  284.      */
  285.     private void completeStack(Throwable cause, StringBuilder stack) {
  286.         for (StackTraceElement ste : cause.getStackTrace()) {
  287.             stack.append(ste.toString()).append('\n');
  288.         }
  289.         Throwable by = cause.getCause();
  290.         if (by != null) {
  291.             stack.append("caused by ").append(by.getMessage() + "\n");
  292.             completeStack(by, stack);
  293.         }
  294.     }

  296.     @Override
  297.     public void notifyParameterStart(String setName, String parameterName) {
  298.         Testsuites tss = results.get(setName);
  299.         Testsuite ts = new Testsuite();
  300.         result.put(setName + parameterName, ts);
  301.         ts.setName(setName + "[" + parameterName + "]");
  302.         ts.setDisabled(0);
  303.         ts.setErrors(0);
  304.         ts.setFailures(0);
  305.         ts.setTests(0);
  306.         ts.setTime(0L);
  307.         resultcase.put(setName + parameterName, new HashMap<>());
  308.         tss.getTestsuite().add(ts);
  309.     }

  311.     @Override
  312.     public void notifyParameterEnd(String setName, String parameterName) {
  313.         Testsuites tss = results.get(setName);
  314.         Testsuite ts = result.get(setName + parameterName);
  315.         tss.setDisabled(tss.getDisabled() + ts.getDisabled());
  316.         tss.setErrors(tss.getErrors() + ts.getErrors());
  317.         tss.setFailures(tss.getFailures() + ts.getFailures());
  318.         tss.setTests(tss.getTests() + ts.getTests());
  319.         tss.setTime(tss.getTime() + ts.getTime());
  320.     }

  322.     /**
  323.      * @return the error
  324.      */
  325.     public boolean isError() {
  326.         return error;
  327.     }

  329. }
Created with JaCoCo 0.7.5.201505241946