Added IsCollectionOnlyContaining to close issue 5

neildunn · neildunn · commit 07fda5d07f69 · 2007-08-02T14:44:38.000Z
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -4,6 +4,10 @@
   generics bug in hasKey and hasValue static factories previously declared
   in IsMapContaining (ngd)
 
+* Added IsCollectionOnlyContaining which matches collections with
+  items only matching a specified item. E.g onlyContains(3,4,5) or
+  onlyContains(lessThan(9))
+
 == Version 1.1: Released Jun 30 2007 ==
 
 * Hamcrest Generator now includes JavaDoc and parameter names in generated code
diff --git a/hamcrest-library/src/main/java/org/hamcrest/collection/IsCollectionOnlyContaining.java b/hamcrest-library/src/main/java/org/hamcrest/collection/IsCollectionOnlyContaining.java
@@ -0,0 +1,58 @@
+package org.hamcrest.collection;
+
+import static org.hamcrest.core.AnyOf.anyOf;
+import static org.hamcrest.core.IsEqual.equalTo;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+import org.hamcrest.Description;
+import org.hamcrest.Factory;
+import org.hamcrest.Matcher;
+import org.hamcrest.TypeSafeMatcher;
+
+/**
+ * Matches collections that only contain elements that satisfy a given matcher.
+ * </p>
+ * For example <pre>[1,2,3]</pre> satisfies onlyContains(lessThan(4))
+ * 
+ * @author ngd
+ */
+public class IsCollectionOnlyContaining<T> extends TypeSafeMatcher<Collection<T>> {
+
+	private final Matcher<T> matcher;
+
+	public IsCollectionOnlyContaining(Matcher<T> matcher) {
+		this.matcher = matcher;
+	}
+	
+	@Override
+	public boolean matchesSafely(Collection<T> items) {		
+		for (T item : items) {
+			if (!matcher.matches(item)) {
+				return false;
+			}
+		}		
+		return true;
+	}
+	
+	public void describeTo(Description description) {
+		description
+    	   	.appendText("a collection containing items matching ")
+    	   	.appendDescriptionOf(matcher);
+	}
+	
+	@Factory
+	public static <T> Matcher<Collection<T>> onlyContains(T... items) {
+		Collection<Matcher<T>> matchers = new ArrayList<Matcher<T>>();
+		for (T item : items) {
+			matchers.add(equalTo(item));
+		}
+		return new IsCollectionOnlyContaining<T>(anyOf(matchers));
+	}
+
+	@Factory 
+	public static <T> Matcher<Collection<T>> onlyContains(Matcher<T>... matchers) {		
+		return new IsCollectionOnlyContaining<T>(anyOf(matchers));
+	}
+}
