mockit.internal.expectations.argumentMatching.HamcrestAdapter Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.expectations.argumentMatching;

import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;

import mockit.internal.reflection.FieldReflection;

import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.StringDescription;
import org.hamcrest.core.Is;
import org.hamcrest.core.IsEqual;
import org.hamcrest.core.IsNot;
import org.hamcrest.core.IsSame;

/**
 * Adapts the org.hamcrest.Matcher interface to {@link ArgumentMatcher}.
 */
public final class HamcrestAdapter implements ArgumentMatcher {
    @NonNull
    private final Matcher hamcrestMatcher;

    public HamcrestAdapter(@NonNull Matcher matcher) {
        hamcrestMatcher = matcher;
    }

    @Override
    public boolean same(@NonNull HamcrestAdapter other) {
        return hamcrestMatcher == other.hamcrestMatcher;
    }

    @Override
    public boolean matches(@Nullable Object argValue) {
        return hamcrestMatcher.matches(argValue);
    }

    @Override
    public void writeMismatchPhrase(@NonNull ArgumentMismatch argumentMismatch) {
        Description strDescription = new StringDescription();
        hamcrestMatcher.describeTo(strDescription);
        argumentMismatch.append(strDescription.toString());
    }

    @Nullable
    public Object getInnerValue() {
        Object innermostMatcher = getInnermostMatcher();
        return getArgumentValueFromMatcherIfAvailable(innermostMatcher);
    }

    @NonNull
    private Object getInnermostMatcher() {
        Matcher innerMatcher = hamcrestMatcher;

        while (innerMatcher instanceof Is || innerMatcher instanceof IsNot) {
            innerMatcher = FieldReflection.getField(innerMatcher.getClass(), Matcher.class, innerMatcher);
        }

        assert innerMatcher != null;
        return innerMatcher;
    }

    @Nullable
    private static Object getArgumentValueFromMatcherIfAvailable(@NonNull Object argMatcher) {
        if (argMatcher instanceof IsEqual || argMatcher instanceof IsSame
                || "org.hamcrest.number.OrderingComparison".equals(argMatcher.getClass().getName())) {
            return FieldReflection.getField(argMatcher.getClass(), Object.class, argMatcher);
        }

        return null;
    }
}