org.javafunk.matchbox.implementations.IsBeanWithSameAttributesAsMatcher.java Source code

Introduction

Here is the source code for org.javafunk.matchbox.implementations.IsBeanWithSameAttributesAsMatcher.java

Source

/*
 * Copyright (C) 2012 Matchbox committers.
 * All rights reserved.
 *
 * The software in this package is published under the terms of the BSD
 * style license a copy of which has been included with this distribution in
 * the LICENSE.txt file.
 */
package org.javafunk.matchbox.implementations;

import org.apache.commons.beanutils.BeanMap;
import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.TypeSafeDiagnosingMatcher;

import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import static java.lang.String.format;
import static java.util.Arrays.asList;
import static org.apache.commons.lang.StringUtils.join;

public class IsBeanWithSameAttributesAsMatcher<T> extends TypeSafeDiagnosingMatcher<T> {
    private final T expectedObject;
    private final Set<String> ignoreProperties;

    public IsBeanWithSameAttributesAsMatcher(T expectedObject, Set<String> ignoreProperties) {
        this.expectedObject = expectedObject;
        this.ignoreProperties = ignoreProperties;
    }

    public static <T> Matcher<T> isBeanWithSameAttributesAs(T expectedObject) {
        return isBeanWithSameAttributesAs(expectedObject, Collections.<String>emptySet());
    }

    public static <T> Matcher<T> isBeanWithSameAttributesAs(T expectedObject, Set<String> ignoreProperties) {
        return new IsBeanWithSameAttributesAsMatcher<T>(expectedObject, ignoreProperties);
    }

    public static Set<String> ignoring(String... ignoreProperties) {
        return new HashSet<String>(asList(ignoreProperties));
    }

    @Override
    protected boolean matchesSafely(T actualObject, Description description) {
        @SuppressWarnings("unchecked")
        Map<String, Object> actualProperties = new BeanMap(actualObject);
        @SuppressWarnings("unchecked")
        Map<String, Object> expectedProperties = new BeanMap(expectedObject);
        for (String propertyName : actualProperties.keySet()) {
            if (ignoreProperties.contains(propertyName)) {
                continue;
            }
            Object actualValue = actualProperties.get(propertyName);
            Object expectedValue = expectedProperties.get(propertyName);
            if ((actualValue == null && expectedValue != null)
                    || (actualValue != null && !actualValue.equals(expectedValue))) {
                description.appendText(format("got a %s ", actualObject.getClass().getSimpleName()))
                        .appendValue(expectedObject)
                        .appendText(format("\nMismatch: expected property \"%s\" = ", propertyName))
                        .appendValue(expectedValue)
                        .appendText(format("\n            actual property \"%s\" = ", propertyName))
                        .appendValue(actualValue);
                return false;
            }
        }
        return true;
    }

    @Override
    public void describeTo(Description description) {
        description.appendText(format("a %s matching ", expectedObject.getClass().getSimpleName()))
                .appendValue(expectedObject);
        if (ignoreProperties.size() > 0) {
            description.appendText(" ignoring properties ").appendText(join(ignoreProperties, ", "));
        }
    }

    @Override
    public boolean equals(Object object) {
        return EqualsBuilder.reflectionEquals(this, object);
    }

    @Override
    public int hashCode() {
        return HashCodeBuilder.reflectionHashCode(this);
    }
}