1 files changed, 104 insertions, 0 deletions
diff --git a/common/tests/org.eclipse.jpt.common.utility.tests/src/org/eclipse/jpt/common/utility/tests/internal/iterables/TransformationIterableTests.java b/common/tests/org.eclipse.jpt.common.utility.tests/src/org/eclipse/jpt/common/utility/tests/internal/iterables/TransformationIterableTests.java
new file mode 100644
index 0000000000..5a16de67c8
--- /dev/null
+++ b/common/tests/org.eclipse.jpt.common.utility.tests/src/org/eclipse/jpt/common/utility/tests/internal/iterables/TransformationIterableTests.java
@@ -0,0 +1,104 @@
+/*******************************************************************************
+ * Copyright (c) 2009 Oracle. All rights reserved.
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0, which accompanies this distribution
+ * and is available at http://www.eclipse.org/legal/epl-v10.html.
+ * 
+ * Contributors:
+ *     Oracle - initial API and implementation
+ ******************************************************************************/
+package org.eclipse.jpt.common.utility.tests.internal.iterables;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+import junit.framework.TestCase;
+
+import org.eclipse.jpt.common.utility.internal.Transformer;
+import org.eclipse.jpt.common.utility.internal.iterables.TransformationIterable;
+
+@SuppressWarnings("nls")
+public class TransformationIterableTests extends TestCase {
+
+	public TransformationIterableTests(String name) {
+		super(name);
+	}
+
+	public void testTransform1() {
+		int i = 1;
+		for (Integer integer : this.buildIterable1()) {
+			assertEquals(i++, integer.intValue());
+		}
+	}
+
+	private Iterable<Integer> buildIterable1() {
+		return this.buildTransformationIterable1(this.buildNestedIterable());
+	}
+
+	private Iterable<Integer> buildTransformationIterable1(Iterable<String> nestedIterable) {
+		// transform each string into an integer with a value of the string's length
+		return new TransformationIterable<String, Integer>(nestedIterable) {
+			@Override
+			protected Integer transform(String next) {
+				return new Integer(next.length());
+			}
+		};
+	}
+
+	public void testTransform2() {
+		int i = 1;
+		for (Integer integer : this.buildIterable2()) {
+			assertEquals(i++, integer.intValue());
+		}
+	}
+
+	private Iterable<Integer> buildIterable2() {
+		return this.buildTransformationIterable2(this.buildNestedIterable());
+	}
+
+	private Iterable<Integer> buildTransformationIterable2(Iterable<String> nestedIterable) {
+		// transform each string into an integer with a value of the string's length
+		return new TransformationIterable<String, Integer>(nestedIterable, this.buildTransformer());
+	}
+
+	private Transformer<String, Integer> buildTransformer() {
+		// transform each string into an integer with a value of the string's length
+		return new Transformer<String, Integer>() {
+			public Integer transform(String next) {
+				return new Integer(next.length());
+			}
+		};
+	}
+
+	private Iterable<String> buildNestedIterable() {
+		Collection<String> c = new ArrayList<String>();
+		c.add("1");
+		c.add("22");
+		c.add("333");
+		c.add("4444");
+		c.add("55555");
+		c.add("666666");
+		c.add("7777777");
+		c.add("88888888");
+		return c;
+	}
+
+	public void testToString() {
+		assertNotNull(this.buildIterable1().toString());
+	}
+
+	public void testMissingTransformer() {
+		Iterable<Integer> iterable = new TransformationIterable<String, Integer>(this.buildNestedIterable());
+		boolean exCaught = false;
+		try {
+			int i = 1;
+			for (Integer integer : iterable) {
+				assertEquals(i++, integer.intValue());
+			}
+		} catch (RuntimeException ex) {
+			exCaught = true;
+		}
+		assertTrue(exCaught);
+	}
+
+}