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

Introduction

Here is the source code for org.javafunk.matchbox.implementations.HasOnlyItemsInAnyOrderMatcher.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 com.google.common.collect.HashMultiset;
import com.google.common.collect.Multiset;
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 static java.util.Arrays.asList;

public class HasOnlyItemsInAnyOrderMatcher<E> extends TypeSafeDiagnosingMatcher<Iterable<E>> {
    private final Multiset<E> expectedMultiset;

    public HasOnlyItemsInAnyOrderMatcher(Iterable<E> expectedMultiset) {
        this.expectedMultiset = HashMultiset.create(expectedMultiset);
    }

    public static <T> Matcher<Iterable<T>> hasOnlyItemsInAnyOrder(T... items) {
        return hasOnlyItemsInAnyOrder(asList(items));
    }

    public static <T> Matcher<Iterable<T>> hasOnlyItemsInAnyOrder(Iterable<T> expectedItems) {
        return new HasOnlyItemsInAnyOrderMatcher<T>(expectedItems);
    }

    @Override
    protected boolean matchesSafely(final Iterable<E> actualItems, Description description) {
        boolean matches = true;
        final Multiset<E> actualMultiset = HashMultiset.create(actualItems);

        description.appendText("got ");

        if (actualMultiset.size() > 0) {
            description.appendValueList("", ", ", "", actualItems);
        } else {
            description.appendText("empty collection");
            if (expectedMultiset.size() != 0) {
                return false;
            }
        }

        if (actualMultiset.size() != expectedMultiset.size()) {
            description.appendText("\n").appendText("got collection with size ").appendValue(actualMultiset.size())
                    .appendText(" rather than ").appendValue(expectedMultiset.size());
            matches = false;
        }

        if (checkForDifferences(expectedMultiset, actualMultiset, description, "expected but didn't get ")) {
            matches = false;
        }
        if (checkForDifferences(actualMultiset, expectedMultiset, description, "got but didn't expect ")) {
            matches = false;
        }

        return matches;
    }

    private boolean checkForDifferences(Multiset<E> expectedSet, Multiset<E> actualSet, Description description,
            String message) {
        Multiset<E> differences = HashMultiset.create(expectedSet);
        for (E item : actualSet) {
            differences.remove(item);
        }
        if (differences.size() > 0) {
            description.appendText("\n").appendText(message).appendValueList("", ", ", "", differences);
            return true;
        }
        return false;
    }

    @Override
    public void describeTo(Description description) {
        if (expectedMultiset.size() > 0) {
            description.appendText("Collection with size ").appendValue(expectedMultiset.size())
                    .appendText(" containing exactly items ").appendValueList("", ", ", "", expectedMultiset)
                    .appendText(" in any order.");
        } else {
            description.appendText("empty collection");
        }
    }

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

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