com.atlassian.jira.rest.client.IterableMatcher.java Source code

Introduction

Here is the source code for com.atlassian.jira.rest.client.IterableMatcher.java

Source

/*
 * Copyright (C) 2010 Atlassian
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.atlassian.jira.rest.client;

import com.google.common.collect.Iterables;
import org.hamcrest.Description;
import org.junit.internal.matchers.TypeSafeMatcher;

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

public class IterableMatcher<T> extends TypeSafeMatcher<Iterable<T>> {
    private final Iterable<T> expected;

    public IterableMatcher(Iterable<T> expected) {
        this.expected = expected;
    }

    public static <T> IterableMatcher<T> hasOnlyElements(T... elements) {
        return new IterableMatcher<T>(Arrays.asList(elements));
    }

    public static <T> IterableMatcher<T> hasOnlyElements(Iterable<T> elements) {
        return new IterableMatcher<T>(elements);
    }

    public static <T> IterableMatcher<T> isEmpty() {
        return new IterableMatcher<T>(Collections.<T>emptyList());
    }

    public static <T> TypeSafeMatcher<Iterable<T>> contains(final T element) {
        return new TypeSafeMatcher<Iterable<T>>() {
            @Override
            public boolean matchesSafely(Iterable<T> given) {
                return Iterables.contains(given, element);
            }

            @Override
            public void describeTo(Description description) {
                description.appendText("an iterable containing element " + element);
            }

        };
    }

    @Override
    public boolean matchesSafely(Iterable<T> given) {
        final Set<T> s = asSet(expected);

        for (T t : given) {
            if (!s.remove(t)) {
                return false;
            }
        }
        return s.isEmpty();
    }

    private Set<T> asSet(Iterable<T> iterable) {
        final Set<T> s = new HashSet<T>();

        for (T t : iterable) {
            s.add(t);
        }
        return s;
    }

    @Override
    public void describeTo(Description description) {
        description.appendText("an iterable containing just elements " + asSet(expected));
    }
}