net.amygdalum.extensions.hamcrest.exceptions.ExceptionMatcher Maven / Gradle / Ivy

Go to download
package net.amygdalum.extensions.hamcrest.exceptions;

import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.core.IsEqual.equalTo;

import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.TypeSafeDiagnosingMatcher;

import net.amygdalum.extensions.hamcrest.util.SimpleClass;

public class ExceptionMatcher extends TypeSafeDiagnosingMatcher {

	private Class clazz;
	private Matcher message;
	private Matcher cause;

	public ExceptionMatcher(Class clazz) {
		this.clazz = clazz;
	}

	public static  ExceptionMatcher matchesException(Class clazz) {
		return new ExceptionMatcher(clazz);
	}

	public ExceptionMatcher withMessage(String message) {
		return withMessage(equalTo(message));
	}

	public ExceptionMatcher withMessage(Matcher message) {
		this.message = message;
		return this;
	}

	public ExceptionMatcher withCause(Class cause) {
		Matcher causeMatcher = instanceOf(cause);
		return withCause(causeMatcher);
	}

	public ExceptionMatcher withCause(Matcher cause) {
		this.cause = cause;
		return this;
	}

	@Override
	public void describeTo(Description description) {
			description.appendText("of type ").appendValue(new SimpleClass(clazz));
		if (message != null) {
			description.appendText(" with message ").appendValue(message);
		}
		if (cause != null) {
			description.appendText(" with cause ").appendValue(cause);
		}
	}

	@Override
	protected boolean matchesSafely(Throwable item, Description mismatchDescription) {
		if (!clazz.isInstance(item)) {
			mismatchDescription.appendText("found class ").appendValue(new SimpleClass(item.getClass()));
			return false;
		}
		if (message != null && !message.matches(item.getMessage())) {
			mismatchDescription.appendText("found message ").appendValue(item.getMessage());
			return false;
		}
		if (cause != null && !cause.matches(item.getCause())) {
			mismatchDescription.appendText("found cause ").appendValue(item.getCause());
			return false;
		}
		return true;
	}

}