ReporterChainingTest.java

package org.approvaltests.reporters.tests;

import junit.framework.TestCase;

import org.approvaltests.Approvals;
import org.approvaltests.reporters.EnvironmentAwareReporter;
import org.approvaltests.reporters.FirstWorkingReporter;
import org.approvaltests.reporters.MultiReporter;

public class ReporterChainingTest extends TestCase
{
  public static class ExceptionThrowingReporter implements EnvironmentAwareReporter
  {
    public boolean run = false;
    @Override
    public void report(String received, String approved) throws Exception
    {
      run = true;
      throw new Error("Error");
    }
    @Override
    public boolean isWorkingInThisEnvironment(String forFile)
    {
      return true;
    }
  }
  public static class NonWorkingReporter implements EnvironmentAwareReporter
  {
    @Override
    public void report(String received, String approved) throws Exception
    {
    }
    @Override
    public boolean isWorkingInThisEnvironment(String forFile)
    {
      return false;
    }
  }
  public static class WorkingReporter implements EnvironmentAwareReporter
  {
    public String received;
    @Override
    public void report(String received, String approved) throws Exception
    {
      this.received = received;
    }
    @Override
    public boolean isWorkingInThisEnvironment(String forFile)
    {
      return true;
    }
  }
  public void testFirstWorkingReporter() throws Exception
  {
    WorkingReporter workingReporter = new WorkingReporter();
    WorkingReporter workingReporter2 = new WorkingReporter();
    FirstWorkingReporter reporter = new FirstWorkingReporter(new NonWorkingReporter(), workingReporter,
        workingReporter2);
    reporter.report("Hello", "world");
    assertEquals("Hello", workingReporter.received);
    assertNull(workingReporter2.received);
    assertTrue(reporter.isWorkingInThisEnvironment(""));
  }
  public void testMultiReporter() throws Exception
  {
    WorkingReporter workingReporter = new WorkingReporter();
    WorkingReporter workingReporter2 = new WorkingReporter();
    MultiReporter reporter = new MultiReporter(workingReporter, workingReporter2);
    reporter.report("Hello", "world");
    assertEquals("Hello", workingReporter.received);
    assertEquals("Hello", workingReporter2.received);
  }
  public void testMultiReporterWithExceptions() throws Exception
  {
    ExceptionThrowingReporter exception1 = new ExceptionThrowingReporter();
    ExceptionThrowingReporter exception2 = new ExceptionThrowingReporter();
    MultiReporter reporter = new MultiReporter(exception1, exception2);
    try
    {
      reporter.report("Hello", "world");
    }
    catch (Throwable t)
    {
      assertEquals(true, exception1.run);
      assertEquals(true, exception2.run);
      Approvals.verify(t.getMessage());
    }
  }
}