package junit.textui;
   
   import java.io.PrintStream;
   import java.text.NumberFormat;
   import java.util.Enumeration;
   
   import junit.framework.AssertionFailedError;
   import junit.framework.Test;
   import junit.framework.TestFailure;
  import junit.framework.TestListener;
  import junit.framework.TestResult;
  import junit.runner.BaseTestRunner;
  
  public class ResultPrinter implements TestListener {
      PrintStream fWriter;
      int fColumn = 0;
  
      public ResultPrinter(PrintStream writer) {
          fWriter = writer;
      }
  
      /* API for use by textui.TestRunner */
  
      synchronized void print(TestResult result, long runTime) {
          printHeader(runTime);
          printErrors(result);
          printFailures(result);
          printFooter(result);
      }
  
      void printWaitPrompt() {
          getWriter().println();
          getWriter().println("<RETURN> to continue");
      }
  
      /* Internal methods */
  
      protected void printHeader(long runTime) {
          getWriter().println();
          getWriter().println("Time: " + elapsedTimeAsString(runTime));
      }
  
      protected void printErrors(TestResult result) {
          printDefects(result.errors(), result.errorCount(), "error");
      }
  
      protected void printFailures(TestResult result) {
          printDefects(result.failures(), result.failureCount(), "failure");
      }
  
      protected void printDefects(Enumeration<TestFailure> booBoos, int count, String type) {
          if (count == 0) return;
          if (count == 1) {
              getWriter().println("There was " + count + " " + type + ":");
          } else {
              getWriter().println("There were " + count + " " + type + "s:");
          }
          for (int i = 1; booBoos.hasMoreElements(); i++) {
              printDefect(booBoos.nextElement(), i);
          }
      }
  
      public void printDefect(TestFailure booBoo, int count) { // only public for testing purposes
          printDefectHeader(booBoo, count);
          printDefectTrace(booBoo);
      }
  
      protected void printDefectHeader(TestFailure booBoo, int count) {
          // I feel like making this a println, then adding a line giving the throwable a chance to print something
          // before we get to the stack trace.
          getWriter().print(count + ") " + booBoo.failedTest());
      }
  
      protected void printDefectTrace(TestFailure booBoo) {
          getWriter().print(BaseTestRunner.getFilteredTrace(booBoo.trace()));
      }
  
      protected void printFooter(TestResult result) {
          if (result.wasSuccessful()) {
              getWriter().println();
              getWriter().print("OK");
              getWriter().println(" (" + result.runCount() + " test" + (result.runCount() == 1 ? "" : "s") + ")");
  
          } else {
              getWriter().println();
              getWriter().println("FAILURES!!!");
              getWriter().println("Tests run: " + result.runCount() +
                      ",  Failures: " + result.failureCount() +
                      ",  Errors: " + result.errorCount());
          }
          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);
     }
 
     public PrintStream getWriter() {
         return fWriter;
     }
 
     /**
      * @see junit.framework.TestListener#addError(Test, Throwable)
      */
     public void addError(Test test, Throwable e) {
         getWriter().print("E");
     }
 
     /**
      * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError)
      */
     public void addFailure(Test test, AssertionFailedError t) {
         getWriter().print("F");
     }
 
     /**
      * @see junit.framework.TestListener#endTest(Test)
      */
     public void endTest(Test test) {
     }
 
     /**
      * @see junit.framework.TestListener#startTest(Test)
      */
     public void startTest(Test test) {
         getWriter().print(".");
         if (fColumn++ >= 40) {
             getWriter().println();
             fColumn = 0;
         }
     }
 
 }