package org.andromda.maven.plugin.bootstrap;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import junit.framework.AssertionFailedError;
import junit.framework.Test;
import junit.framework.TestListener;
/**
* Formats the translation-library test results into the correct format.
*
* @author Chad Brandon
*/
public class TranslationLibraryTestFormatter
implements TestListener
{
/**
* Stores the report file location.
*/
private File reportFile;
/**
* Stores the contents of the report.
*/
private StringWriter report;
/**
* The print writer for the report.
*/
private PrintWriter reportWriter;
public TranslationLibraryTestFormatter()
{
this.report = new StringWriter();
this.reportWriter = new PrintWriter(this.report);
}
/**
* Sets the file that will contain the report results (i.e.
* the results of the cartridge test run).
*
* @param reportFile the file to which the report output will be written.
*/
public void setReportFile(final File reportFile)
{
this.reportFile = reportFile;
}
/**
* Keeps track of the number of errors.
*/
private int numberOfErrors = 0;
/**
* @see junit.framework.TestListener#addError(junit.framework.Test, java.lang.Throwable)
*/
public void addError(
Test test,
Throwable throwable)
{
this.numberOfErrors++;
this.collectFailure(
test,
throwable);
}
/**
* Keeps track of the number of failures.
*/
private int numberOfFailures = 0;
/**
* @see junit.framework.TestListener#addFailure(junit.framework.Test, junit.framework.AssertionFailedError)
*/
public void addFailure(
Test test,
AssertionFailedError failure)
{
this.numberOfFailures++;
this.collectFailure(
test,
failure);
}
public void endTest(Test test)
{
}
/**
* Keeps track of the number of tests run.
*/
private int numberOfTests = 0;
public void startTest(Test test)
{
this.numberOfTests++;
}
/**
* Stores the start time of the test suite
*/
private long startTime = 0;
/**
* Stores the new line separator.
*/
private static final String newLine = System.getProperty("line.separator");
/**
* The testsuite started.
*/
public void startTestSuite(final String name)
{
startTime = System.currentTimeMillis();
this.reportWriter.println("-------------------------------------------------------------------------------");
this.reportWriter.println(name + " Test Suite");
this.reportWriter.println("-------------------------------------------------------------------------------");
}
/**
* Adds a failure to the current <code>failures</code>
* collection (these are rendered at the end of test suite
* execution).
*
* @param type the failure type (error or failure).
* @param test the actual test.
* @param throwable the failure information.
*/
private void collectFailure(
Test test,
Throwable throwable)
{
this.failures.add(new Failure(
test,
throwable));
}
private Collection failures = new ArrayList();
/**
* Signifies the test suite ended and returns the summary of the
* test.
*
* @param test the test suite being run.
* @return the test summary.
*/
String endTestSuite()
{
final double elapsed = ((System.currentTimeMillis() - this.startTime) / 1000.0);
final StringBuffer summary = new StringBuffer("Tests: " + String.valueOf(this.numberOfTests) + ", ");
summary.append("Failures: " + String.valueOf(this.numberOfFailures) + ", ");
summary.append("Errors: " + String.valueOf(this.numberOfErrors) + ", ");
summary.append("Time elapsed: " + elapsed).append(" sec");
summary.append(newLine);
summary.append(newLine);
this.reportWriter.print(summary);
for (final Iterator iterator = this.failures.iterator(); iterator.hasNext();)
{
final Failure failure = (Failure)iterator.next();
final Throwable information = failure.information;
if (information instanceof AssertionFailedError)
{
this.reportWriter.println(failure.information.getMessage());
}
else
{
this.reportWriter.println("ERROR:");
}
information.printStackTrace(this.reportWriter);
this.reportWriter.println();
}
if (this.reportFile != null)
{
try
{
final File parent = this.reportFile.getParentFile();
if (parent != null && !parent.exists())
{
parent.mkdirs();
}
final FileWriter writer = new FileWriter(this.reportFile);
writer.write(report.toString());
writer.flush();
writer.close();
}
catch (IOException exception)
{
throw new RuntimeException(exception);
}
}
return summary.toString();
}
/**
* Stores the information about a test failure.
*/
private static final class Failure
{
Test test;
Throwable information;
Failure(
final Test test,
final Throwable information)
{
this.test = test;
this.information = information;
}
}
}