ru.yandex.qatools.camelot.test.Matchers Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camelot-test Show documentation
The newest version!
package ru.yandex.qatools.camelot.test;

import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.TypeSafeMatcher;

/**
 * @author Ilya Sadykov (mailto: [email protected])
 * @author Innokenty Shuvalov (mailto: [email protected])
 */
public class Matchers {

    private static class NotNullMatcher
            extends TypeSafeMatcher {

        private final String key;

        private NotNullMatcher(String key) {
            this.key = key;
        }

        @Override
        protected boolean matchesSafely(AggregatorStateStorage stateStorage) {
            return stateStorage.getActual(key) != null;
        }

        @Override
        public void describeTo(Description description) {
            description.appendText(String.format(
                    "storage containing aggregator state by key '%s'", key));
        }
    }

    private static class InstanceOfMatcher
            extends TypeSafeMatcher {

        private final String key;

        private final Class stateClass;

        private InstanceOfMatcher(String key, Class stateClass) {
            this.key = key;
            this.stateClass = stateClass;
        }

        @Override
        protected boolean matchesSafely(AggregatorStateStorage stateStorage) {
            Object actual = stateStorage.getActual(key);
            return actual != null && stateClass.isInstance(actual);
        }

        @Override
        public void describeTo(Description description) {
            description.appendText(String.format(
                    "Storage containing instance of %s by key '%s'",
                    stateClass.getSimpleName(), key));
        }
    }

    public static Matcher containStateFor(String key) {
        return new NotNullMatcher(key);
    }

    public static Matcher containStateFor(String key, Class stateClass) {
        return new InstanceOfMatcher(key, stateClass);
    }
}