package org.junit.internal.runners;

import java.io.PrintStream;

import java.text.NumberFormat;

import org.junit.runner.Description;

import org.junit.runner.Result;

import org.junit.runner.notification.Failure;

import org.junit.runner.notification.RunListener;

public class TextListener extends RunListener {

	private final PrintStream fWriter;

	public TextListener() {

		this(System.out);

	}

	public TextListener(PrintStream writer) {

		this.fWriter= writer;

	}

	@Override

	public void testRunFinished(Result result) {

		printHeader(result.getRunTime());

		printFailures(result);

		printFooter(result);

	}

	@Override

	public void testStarted(Description description) {

		fWriter.append('.');

	}

	@Override

	public void testFailure(Failure failure) {

		fWriter.append('E');

	}

	@Override

	public void testIgnored(Description description) {

		fWriter.append('I');

	}

	/*

	 * Internal methods

	 */

	private PrintStream getWriter() {

		return fWriter;

	}

	protected void printHeader(long runTime) {

		getWriter().println();

		getWriter().println("Time: " + elapsedTimeAsString(runTime));

	}

	protected void printFailures(Result result) {

		if (result.getFailureCount() == 0)

			return;

		if (result.getFailureCount() == 1)

			getWriter().println("There was " + result.getFailureCount() + " failure:");

		else

			getWriter().println("There were " + result.getFailureCount() + " failures:");

		int i= 1;

		for (Failure each : result.getFailures())

			printFailure(each, i++);

	}

	protected void printFailure(Failure failure, int count) {

		printFailureHeader(failure, count);

		printFailureTrace(failure);

	}

	protected void printFailureHeader(Failure failure, int count) {

		getWriter().println(count + ") " + failure.getTestHeader());

	}

	protected void printFailureTrace(Failure failure) {

		getWriter().print(failure.getTrace());

	}

	protected void printFooter(Result result) {

		if (result.wasSuccessful()) {

			getWriter().println();

			getWriter().print("OK");

			getWriter().println(" (" + result.getRunCount() + " test" + (result.getRunCount() == 1 ? "" : "s") + ")");

		} else {

			getWriter().println();

			getWriter().println("FAILURES!!!");

			getWriter().println("Tests run: " + result.getRunCount() + ",  Failures: " + result.getFailureCount());

		}

		getWriter().println();

	}

	/**

	 * Returns the formatted string of the elapsed time. Duplicated from

	 * BaseTestRunner. Fix it.

	 */

	protected String elapsedTimeAsString(long runTime) {

		return NumberFormat.getInstance().format((double) runTime / 1000);

	}

}