common/tests/org.eclipse.jpt.common.utility.tests/src/org/eclipse/jpt/common/utility/tests/internal/iterables/TransformationIterableTests.java

/*******************************************************************************
 * 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);
	}

}