/*
 *
 * ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is Rhino code, released
 * May 6, 1999.
 *
 * The Initial Developer of the Original Code is
 * Netscape Communications Corporation.
 * Portions created by the Initial Developer are Copyright (C) 1997-1999
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *   Bob Jervis
 *   Google Inc.
 *
 * Alternatively, the contents of this file may be used under the terms of
 * the GNU General Public License Version 2 or later (the "GPL"), in which
 * case the provisions of the GPL are applicable instead of those above. If
 * you wish to allow use of your version of this file only under the terms of
 * the GPL and not to allow others to use your version of this file under the
 * MPL, indicate your decision by deleting the provisions above and replacing
 * them with the notice and other provisions required by the GPL. If you do
 * not delete the provisions above, a recipient may use your version of this
 * file under either the MPL or the GPL.
 *
 * ***** END LICENSE BLOCK ***** */

package com.google.javascript.rhino.testing;

import static com.google.common.truth.Truth.assertThat;

import com.google.javascript.rhino.ErrorReporter;

import org.junit.Assert;

/**
 * <p>An error reporter for testing that verifies that messages reported to the
 * reporter are expected.</p>
 *
 * <p>Sample use</p>
 * <pre>
 * TestErrorReporter e =
 *   new TestErrorReporter(null, new String[] { "first warning" });
 * ...
 * assertTrue(e.hasEncounteredAllWarnings());
 * </pre>
 *
 */
public final class TestErrorReporter implements ErrorReporter {
  private String[] errors;
  private String[] warnings;
  private int errorsIndex = 0;
  private int warningsIndex = 0;

  public TestErrorReporter(String[] errors, String[] warnings) {
    this.errors = errors;
    this.warnings = warnings;
  }

  public static TestErrorReporter forNoExpectedReports() {
    return new TestErrorReporter(null, null);
  }

  public void setErrors(String[] errors) {
    this.errors = errors;
    errorsIndex = 0;
  }

  public void setWarnings(String[] warnings) {
    this.warnings = warnings;
    warningsIndex = 0;
  }

  @Override
  public void error(String message, String sourceName, int line,
      int lineOffset) {
    if (errors != null && errorsIndex < errors.length) {
      assertThat(message).isEqualTo(errors[errorsIndex++]);
    } else {
      Assert.fail("extra error: " + message);
    }
  }

  @Override
  public void warning(String message, String sourceName, int line,
      int lineOffset) {
    if (warnings != null && warningsIndex < warnings.length) {
      assertThat(message).isEqualTo(warnings[warningsIndex++]);
    } else {
      Assert.fail("extra warning: " + message);
    }
  }

  public void assertHasEncounteredAllWarnings() {
    if (warnings == null) {
      assertThat(warningsIndex).isEqualTo(0);
    } else {
      assertThat(warnings).hasLength(warningsIndex);
    }
  }

  public void assertHasEncounteredAllErrors() {
    if (errors == null) {
      assertThat(errorsIndex).isEqualTo(0);
    } else {
      assertThat(errors).hasLength(errorsIndex);
    }
  }

}

@@ -2344,29 +2344,101 @@
 rors
-;%0A    this.warnings =
+ == null ? new String%5B%5D %7B%7D : errors;%0A    this.warnings = warnings == null ? new String%5B%5D %7B%7D :
  war
@@ -2872,26 +2872,8 @@
 if (
-errors != null && 
 erro
@@ -3099,38 +3099,32 @@
 eName, int line,
-%0A     
  int lineOffset)
@@ -3138,28 +3138,8 @@
 if (
-warnings != null && 
 warn
@@ -3359,97 +3359,8 @@
 ) %7B%0A
-    if (warnings == null) %7B%0A      assertThat(warningsIndex).isEqualTo(0);%0A    %7D else %7B%0A  
     
@@ -3402,30 +3402,24 @@
 ingsIndex);%0A
-    %7D%0A
   %7D%0A%0A  publi
@@ -3463,93 +3463,8 @@
 ) %7B%0A
-    if (errors == null) %7B%0A      assertThat(errorsIndex).isEqualTo(0);%0A    %7D else %7B%0A  
     
@@ -3508,19 +3508,12 @@
 dex);%0A  
-  %7D%0A  %7D%0A
+%7D
 %0A%7D%0A

Simplify TestErrorReporter a little bit by getting rid of nulls right away.