infer_clone/infer/tests/utils/matchers/ResultContainsErrorInMethod.java

package utils.matchers;

import org.hamcrest.BaseMatcher;
import org.hamcrest.Description;
import org.hamcrest.Matcher;

import java.nio.file.Path;
import java.nio.file.Paths;

import utils.InferError;
import utils.InferResults;

public class ResultContainsErrorInMethod extends BaseMatcher<InferResults> {

  private String errorType;
  private Path errorFile;
  private String errorMethod;

  public ResultContainsErrorInMethod(String type, String file, String method) {
    this.errorType = type;
    this.errorFile = Paths.get(file);
    this.errorMethod = method;
  }

  @Override
  public boolean matches(Object o) {
    InferResults results = (InferResults) o;
    for (InferError foundError : results.getErrors()) {
      if (foundError.matchType(this.errorType)
          && foundError.matchMethod(this.errorMethod)
          && foundError.matchFile(this.errorFile)) {
        return true;
      }
    }
    return false;
  }

  @Override
  public void describeTo(Description description) {
    description.appendText(
        this.errorType + " error in file: " + this.errorFile
            + ", method: " + this.errorMethod);
  }

  @Override
  public void describeMismatch(Object item, Description description) {
    InferResults results = (InferResults) item;
    String resultsString = "";
    for (InferError error : results.getErrors()) {
      resultsString = resultsString + "\n\t" + error;
    }
    description.appendText(
        "Found errors: \n" + resultsString
            + "\nwith:\n" + results.inferCmdToString());
  }

  public static <T> Matcher<InferResults> contains(String type, String file, String method) {
    return new ResultContainsErrorInMethod(type, file, method);
  }

}