/* * Copyright (C) 2007 The Guava Authors * * 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.google.common.collect; import static com.google.common.collect.Iterators.get; import static com.google.common.collect.Iterators.getLast; import static com.google.common.collect.Iterators.skip; import static com.google.common.collect.Lists.newArrayList; import static com.google.common.collect.testing.IteratorFeature.MODIFIABLE; import static com.google.common.collect.testing.IteratorFeature.UNMODIFIABLE; import static java.util.Arrays.asList; import static java.util.Collections.singleton; import static org.junit.contrib.truth.Truth.ASSERT; import com.google.common.annotations.GwtCompatible; import com.google.common.annotations.GwtIncompatible; import com.google.common.base.Function; import com.google.common.base.Predicate; import com.google.common.base.Predicates; import com.google.common.collect.testing.IteratorFeature; import com.google.common.collect.testing.IteratorTester; import com.google.common.collect.testing.ListTestSuiteBuilder; import com.google.common.collect.testing.TestStringListGenerator; import com.google.common.collect.testing.features.CollectionFeature; import com.google.common.collect.testing.features.CollectionSize; import com.google.common.collect.testing.features.ListFeature; import com.google.common.testing.NullPointerTester; import junit.framework.Test; import junit.framework.TestCase; import junit.framework.TestSuite; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.ConcurrentModificationException; import java.util.Enumeration; import java.util.Iterator; import java.util.List; import java.util.NoSuchElementException; import java.util.RandomAccess; import java.util.Set; import java.util.Vector; /** * Unit test for {@code Iterators}. * * @author Kevin Bourrillion */ @GwtCompatible(emulated = true) public class IteratorsTest extends TestCase { @GwtIncompatible("suite") public static Test suite() { TestSuite suite = new TestSuite(IteratorsTest.class.getSimpleName()); suite.addTest(testsForRemoveAllAndRetainAll()); suite.addTestSuite(IteratorsTest.class); return suite; } public void testEmptyIterator() { Iterator iterator = Iterators.emptyIterator(); assertFalse(iterator.hasNext()); try { iterator.next(); fail("no exception thrown"); } catch (NoSuchElementException expected) { } try { iterator.remove(); fail("no exception thrown"); } catch (UnsupportedOperationException expected) { } } public void testSize0() { Iterator iterator = Iterators.emptyIterator(); assertEquals(0, Iterators.size(iterator)); } public void testSize1() { Iterator iterator = Collections.singleton(0).iterator(); assertEquals(1, Iterators.size(iterator)); } public void testSize_partiallyConsumed() { Iterator iterator = asList(1, 2, 3, 4, 5).iterator(); iterator.next(); iterator.next(); assertEquals(3, Iterators.size(iterator)); } public void test_contains_nonnull_yes() { Iterator set = asList("a", null, "b").iterator(); assertTrue(Iterators.contains(set, "b")); } public void test_contains_nonnull_no() { Iterator set = asList("a", "b").iterator(); assertFalse(Iterators.contains(set, "c")); } public void test_contains_null_yes() { Iterator set = asList("a", null, "b").iterator(); assertTrue(Iterators.contains(set, null)); } public void test_contains_null_no() { Iterator set = asList("a", "b").iterator(); assertFalse(Iterators.contains(set, null)); } public void testGetOnlyElement_noDefault_valid() { Iterator iterator = Collections.singletonList("foo").iterator(); assertEquals("foo", Iterators.getOnlyElement(iterator)); } public void testGetOnlyElement_noDefault_empty() { Iterator iterator = Iterators.emptyIterator(); try { Iterators.getOnlyElement(iterator); fail(); } catch (NoSuchElementException expected) { } } public void testGetOnlyElement_noDefault_moreThanOneLessThanFiveElements() { Iterator iterator = asList("one", "two").iterator(); try { Iterators.getOnlyElement(iterator); fail(); } catch (IllegalArgumentException expected) { assertEquals("expected one element but was: ", expected.getMessage()); } } public void testGetOnlyElement_noDefault_fiveElements() { Iterator iterator = asList("one", "two", "three", "four", "five").iterator(); try { Iterators.getOnlyElement(iterator); fail(); } catch (IllegalArgumentException expected) { assertEquals("expected one element but was: " + "", expected.getMessage()); } } public void testGetOnlyElement_noDefault_moreThanFiveElements() { Iterator iterator = asList("one", "two", "three", "four", "five", "six").iterator(); try { Iterators.getOnlyElement(iterator); fail(); } catch (IllegalArgumentException expected) { assertEquals("expected one element but was: " + "", expected.getMessage()); } } public void testGetOnlyElement_withDefault_singleton() { Iterator iterator = Collections.singletonList("foo").iterator(); assertEquals("foo", Iterators.getOnlyElement(iterator, "bar")); } public void testGetOnlyElement_withDefault_empty() { Iterator iterator = Iterators.emptyIterator(); assertEquals("bar", Iterators.getOnlyElement(iterator, "bar")); } public void testGetOnlyElement_withDefault_empty_null() { Iterator iterator = Iterators.emptyIterator(); assertNull(Iterators.getOnlyElement(iterator, null)); } public void testGetOnlyElement_withDefault_two() { Iterator iterator = asList("foo", "bar").iterator(); try { Iterators.getOnlyElement(iterator, "x"); fail(); } catch (IllegalArgumentException expected) { assertEquals("expected one element but was: ", expected.getMessage()); } } @GwtIncompatible("Iterators.toArray(Iterator, Class)") public void testToArrayEmpty() { Iterator iterator = Collections.emptyList().iterator(); String[] array = Iterators.toArray(iterator, String.class); assertTrue(Arrays.equals(new String[0], array)); } @GwtIncompatible("Iterators.toArray(Iterator, Class)") public void testToArraySingleton() { Iterator iterator = Collections.singletonList("a").iterator(); String[] array = Iterators.toArray(iterator, String.class); assertTrue(Arrays.equals(new String[] { "a" }, array)); } @GwtIncompatible("Iterators.toArray(Iterator, Class)") public void testToArray() { String[] sourceArray = new String[] {"a", "b", "c"}; Iterator iterator = asList(sourceArray).iterator(); String[] newArray = Iterators.toArray(iterator, String.class); assertTrue(Arrays.equals(sourceArray, newArray)); } public void testFilterSimple() { Iterator unfiltered = Lists.newArrayList("foo", "bar").iterator(); Iterator filtered = Iterators.filter(unfiltered, Predicates.equalTo("foo")); List expected = Collections.singletonList("foo"); List actual = Lists.newArrayList(filtered); assertEquals(expected, actual); } public void testFilterNoMatch() { Iterator unfiltered = Lists.newArrayList("foo", "bar").iterator(); Iterator filtered = Iterators.filter(unfiltered, Predicates.alwaysFalse()); List expected = Collections.emptyList(); List actual = Lists.newArrayList(filtered); assertEquals(expected, actual); } public void testFilterMatchAll() { Iterator unfiltered = Lists.newArrayList("foo", "bar").iterator(); Iterator filtered = Iterators.filter(unfiltered, Predicates.alwaysTrue()); List expected = Lists.newArrayList("foo", "bar"); List actual = Lists.newArrayList(filtered); assertEquals(expected, actual); } public void testFilterNothing() { Iterator unfiltered = Collections.emptyList().iterator(); Iterator filtered = Iterators.filter(unfiltered, new Predicate() { @Override public boolean apply(String s) { fail("Should never be evaluated"); return false; } }); List expected = Collections.emptyList(); List actual = Lists.newArrayList(filtered); assertEquals(expected, actual); } @GwtIncompatible("unreasonable slow") public void testFilterUsingIteratorTester() { final List list = asList(1, 2, 3, 4, 5); final Predicate isEven = new Predicate() { @Override public boolean apply(Integer integer) { return integer % 2 == 0; } }; new IteratorTester(5, UNMODIFIABLE, asList(2, 4), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.filter(list.iterator(), isEven); } }.test(); } public void testAny() { List list = Lists.newArrayList(); Predicate predicate = Predicates.equalTo("pants"); assertFalse(Iterators.any(list.iterator(), predicate)); list.add("cool"); assertFalse(Iterators.any(list.iterator(), predicate)); list.add("pants"); assertTrue(Iterators.any(list.iterator(), predicate)); } public void testAll() { List list = Lists.newArrayList(); Predicate predicate = Predicates.equalTo("cool"); assertTrue(Iterators.all(list.iterator(), predicate)); list.add("cool"); assertTrue(Iterators.all(list.iterator(), predicate)); list.add("pants"); assertFalse(Iterators.all(list.iterator(), predicate)); } public void testFind_firstElement() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.find(iterator, Predicates.equalTo("cool"))); assertEquals("pants", iterator.next()); } public void testFind_lastElement() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("pants", Iterators.find(iterator, Predicates.equalTo("pants"))); assertFalse(iterator.hasNext()); } public void testFind_notPresent() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); try { Iterators.find(iterator, Predicates.alwaysFalse()); fail(); } catch (NoSuchElementException e) { } assertFalse(iterator.hasNext()); } public void testFind_matchAlways() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.find(iterator, Predicates.alwaysTrue())); } public void testFind_withDefault_first() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.find(iterator, Predicates.equalTo("cool"), "woot")); assertEquals("pants", iterator.next()); } public void testFind_withDefault_last() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("pants", Iterators.find(iterator, Predicates.equalTo("pants"), "woot")); assertFalse(iterator.hasNext()); } public void testFind_withDefault_notPresent() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("woot", Iterators.find(iterator, Predicates.alwaysFalse(), "woot")); assertFalse(iterator.hasNext()); } public void testFind_withDefault_notPresent_nullReturn() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertNull( Iterators.find(iterator, Predicates.alwaysFalse(), null)); assertFalse(iterator.hasNext()); } public void testFind_withDefault_matchAlways() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.find(iterator, Predicates.alwaysTrue(), "woot")); assertEquals("pants", iterator.next()); } public void testTryFind_firstElement() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.tryFind(iterator, Predicates.equalTo("cool")).get()); } public void testTryFind_lastElement() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("pants", Iterators.tryFind(iterator, Predicates.equalTo("pants")).get()); } public void testTryFind_alwaysTrue() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("cool", Iterators.tryFind(iterator, Predicates.alwaysTrue()).get()); } public void testTryFind_alwaysFalse_orDefault() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertEquals("woot", Iterators.tryFind(iterator, Predicates.alwaysFalse()).or("woot")); assertFalse(iterator.hasNext()); } public void testTryFind_alwaysFalse_isPresent() { Iterable list = Lists.newArrayList("cool", "pants"); Iterator iterator = list.iterator(); assertFalse( Iterators.tryFind(iterator, Predicates.alwaysFalse()).isPresent()); assertFalse(iterator.hasNext()); } public void testTransform() { Iterator input = asList("1", "2", "3").iterator(); Iterator result = Iterators.transform(input, new Function() { @Override public Integer apply(String from) { return Integer.valueOf(from); } }); List actual = Lists.newArrayList(result); List expected = asList(1, 2, 3); assertEquals(expected, actual); } public void testTransformRemove() { List list = Lists.newArrayList("1", "2", "3"); Iterator input = list.iterator(); Iterator iterator = Iterators.transform(input, new Function() { @Override public Integer apply(String from) { return Integer.valueOf(from); } }); assertEquals(Integer.valueOf(1), iterator.next()); assertEquals(Integer.valueOf(2), iterator.next()); iterator.remove(); assertEquals(asList("1", "3"), list); } public void testPoorlyBehavedTransform() { Iterator input = asList("1", null, "3").iterator(); Iterator result = Iterators.transform(input, new Function() { @Override public Integer apply(String from) { return Integer.valueOf(from); } }); result.next(); try { result.next(); fail("Expected NFE"); } catch (NumberFormatException nfe) { // Expected to fail. } } public void testNullFriendlyTransform() { Iterator input = asList(1, 2, null, 3).iterator(); Iterator result = Iterators.transform(input, new Function() { @Override public String apply(Integer from) { return String.valueOf(from); } }); List actual = Lists.newArrayList(result); List expected = asList("1", "2", "null", "3"); assertEquals(expected, actual); } public void testCycleOfEmpty() { // "" for javac 1.5. Iterator cycle = Iterators.cycle(); assertFalse(cycle.hasNext()); } public void testCycleOfOne() { Iterator cycle = Iterators.cycle("a"); for (int i = 0; i < 3; i++) { assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); } } public void testCycleOfOneWithRemove() { Iterable iterable = Lists.newArrayList("a"); Iterator cycle = Iterators.cycle(iterable); assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); cycle.remove(); assertEquals(Collections.emptyList(), iterable); assertFalse(cycle.hasNext()); } public void testCycleOfTwo() { Iterator cycle = Iterators.cycle("a", "b"); for (int i = 0; i < 3; i++) { assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); assertTrue(cycle.hasNext()); assertEquals("b", cycle.next()); } } public void testCycleOfTwoWithRemove() { Iterable iterable = Lists.newArrayList("a", "b"); Iterator cycle = Iterators.cycle(iterable); assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); assertTrue(cycle.hasNext()); assertEquals("b", cycle.next()); assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); cycle.remove(); assertEquals(Collections.singletonList("b"), iterable); assertTrue(cycle.hasNext()); assertEquals("b", cycle.next()); assertTrue(cycle.hasNext()); assertEquals("b", cycle.next()); cycle.remove(); assertEquals(Collections.emptyList(), iterable); assertFalse(cycle.hasNext()); } public void testCycleRemoveWithoutNext() { Iterator cycle = Iterators.cycle("a", "b"); assertTrue(cycle.hasNext()); try { cycle.remove(); fail("no exception thrown"); } catch (IllegalStateException expected) { } } public void testCycleRemoveSameElementTwice() { Iterator cycle = Iterators.cycle("a", "b"); cycle.next(); cycle.remove(); try { cycle.remove(); fail("no exception thrown"); } catch (IllegalStateException expected) { } } public void testCycleWhenRemoveIsNotSupported() { Iterable iterable = asList("a", "b"); Iterator cycle = Iterators.cycle(iterable); cycle.next(); try { cycle.remove(); fail("no exception thrown"); } catch (UnsupportedOperationException expected) { } } public void testCycleRemoveAfterHasNext() { Iterable iterable = Lists.newArrayList("a"); Iterator cycle = Iterators.cycle(iterable); assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); assertTrue(cycle.hasNext()); cycle.remove(); assertEquals(Collections.emptyList(), iterable); assertFalse(cycle.hasNext()); } public void testCycleNoSuchElementException() { Iterable iterable = Lists.newArrayList("a"); Iterator cycle = Iterators.cycle(iterable); assertTrue(cycle.hasNext()); assertEquals("a", cycle.next()); cycle.remove(); assertFalse(cycle.hasNext()); try { cycle.next(); fail(); } catch (NoSuchElementException expected) {} } @GwtIncompatible("unreasonable slow") public void testCycleUsingIteratorTester() { new IteratorTester(5, UNMODIFIABLE, asList(1, 2, 1, 2, 1, 2, 1, 2, 1, 2, 1, 2), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.cycle(asList(1, 2)); } }.test(); } @GwtIncompatible("slow (~5s)") public void testConcatNoIteratorsYieldsEmpty() { new EmptyIteratorTester() { @SuppressWarnings("unchecked") @Override protected Iterator newTargetIterator() { return Iterators.concat(); } }.test(); } @GwtIncompatible("slow (~5s)") public void testConcatOneEmptyIteratorYieldsEmpty() { new EmptyIteratorTester() { @SuppressWarnings("unchecked") @Override protected Iterator newTargetIterator() { return Iterators.concat(iterateOver()); } }.test(); } @GwtIncompatible("slow (~5s)") public void testConcatMultipleEmptyIteratorsYieldsEmpty() { new EmptyIteratorTester() { @Override protected Iterator newTargetIterator() { return Iterators.concat(iterateOver(), iterateOver()); } }.test(); } @GwtIncompatible("slow (~3s)") public void testConcatSingletonYieldsSingleton() { new SingletonIteratorTester() { @SuppressWarnings("unchecked") @Override protected Iterator newTargetIterator() { return Iterators.concat(iterateOver(1)); } }.test(); } @GwtIncompatible("slow (~5s)") public void testConcatEmptyAndSingletonAndEmptyYieldsSingleton() { new SingletonIteratorTester() { @Override protected Iterator newTargetIterator() { return Iterators.concat(iterateOver(), iterateOver(1), iterateOver()); } }.test(); } @GwtIncompatible("fairly slow (~40s)") public void testConcatSingletonAndSingletonYieldsDoubleton() { new DoubletonIteratorTester() { @Override protected Iterator newTargetIterator() { return Iterators.concat(iterateOver(1), iterateOver(2)); } }.test(); } @GwtIncompatible("fairly slow (~40s)") public void testConcatSingletonAndSingletonWithEmptiesYieldsDoubleton() { new DoubletonIteratorTester() { @Override protected Iterator newTargetIterator() { return Iterators.concat( iterateOver(1), iterateOver(), iterateOver(), iterateOver(2)); } }.test(); } @GwtIncompatible("fairly slow (~50s)") public void testConcatUnmodifiable() { new IteratorTester(5, UNMODIFIABLE, asList(1, 2), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.concat(asList(1).iterator(), Arrays.asList().iterator(), asList(2).iterator()); } }.test(); } /** * Illustrates the somewhat bizarre behavior when a null is passed in. */ public void testConcatContainingNull() { @SuppressWarnings("unchecked") Iterator> input = asList(iterateOver(1, 2), null, iterateOver(3)).iterator(); Iterator result = Iterators.concat(input); assertEquals(1, (int) result.next()); assertEquals(2, (int) result.next()); try { result.hasNext(); fail("no exception thrown"); } catch (NullPointerException e) { } try { result.next(); fail("no exception thrown"); } catch (NullPointerException e) { } // There is no way to get "through" to the 3. Buh-bye } @SuppressWarnings("unchecked") public void testConcatVarArgsContainingNull() { try { Iterators.concat(iterateOver(1, 2), null, iterateOver(3), iterateOver(4), iterateOver(5)); fail("no exception thrown"); } catch (NullPointerException e) { } } public void testAddAllWithEmptyIterator() { List alreadyThere = Lists.newArrayList("already", "there"); boolean changed = Iterators.addAll(alreadyThere, Iterators.emptyIterator()); ASSERT.that(alreadyThere).hasContentsInOrder("already", "there"); assertFalse(changed); } public void testAddAllToList() { List alreadyThere = Lists.newArrayList("already", "there"); List freshlyAdded = Lists.newArrayList("freshly", "added"); boolean changed = Iterators.addAll(alreadyThere, freshlyAdded.iterator()); ASSERT.that(alreadyThere).hasContentsInOrder("already", "there", "freshly", "added"); assertTrue(changed); } public void testAddAllToSet() { Set alreadyThere = Sets.newLinkedHashSet(asList("already", "there")); List oneMore = Lists.newArrayList("there"); boolean changed = Iterators.addAll(alreadyThere, oneMore.iterator()); ASSERT.that(alreadyThere).hasContentsInOrder("already", "there"); assertFalse(changed); } @GwtIncompatible("NullPointerTester") public void testNullPointerExceptions() throws Exception { NullPointerTester tester = new NullPointerTester(); tester.testAllPublicStaticMethods(Iterators.class); } @GwtIncompatible("Only used by @GwtIncompatible code") private static abstract class EmptyIteratorTester extends IteratorTester { protected EmptyIteratorTester() { super(3, MODIFIABLE, Collections.emptySet(), IteratorTester.KnownOrder.KNOWN_ORDER); } } @GwtIncompatible("Only used by @GwtIncompatible code") private static abstract class SingletonIteratorTester extends IteratorTester { protected SingletonIteratorTester() { super(3, MODIFIABLE, singleton(1), IteratorTester.KnownOrder.KNOWN_ORDER); } } @GwtIncompatible("Only used by @GwtIncompatible code") private static abstract class DoubletonIteratorTester extends IteratorTester { protected DoubletonIteratorTester() { super(5, MODIFIABLE, newArrayList(1, 2), IteratorTester.KnownOrder.KNOWN_ORDER); } } private static Iterator iterateOver(final Integer... values) { return newArrayList(values).iterator(); } public void testElementsEqual() { Iterable a; Iterable b; // Base case. a = Lists.newArrayList(); b = Collections.emptySet(); assertTrue(Iterators.elementsEqual(a.iterator(), b.iterator())); // A few elements. a = asList(4, 8, 15, 16, 23, 42); b = asList(4, 8, 15, 16, 23, 42); assertTrue(Iterators.elementsEqual(a.iterator(), b.iterator())); // The same, but with nulls. a = asList(4, 8, null, 16, 23, 42); b = asList(4, 8, null, 16, 23, 42); assertTrue(Iterators.elementsEqual(a.iterator(), b.iterator())); // Different Iterable types (still equal elements, though). a = ImmutableList.of(4, 8, 15, 16, 23, 42); b = asList(4, 8, 15, 16, 23, 42); assertTrue(Iterators.elementsEqual(a.iterator(), b.iterator())); // An element differs. a = asList(4, 8, 15, 12, 23, 42); b = asList(4, 8, 15, 16, 23, 42); assertFalse(Iterators.elementsEqual(a.iterator(), b.iterator())); // null versus non-null. a = asList(4, 8, 15, null, 23, 42); b = asList(4, 8, 15, 16, 23, 42); assertFalse(Iterators.elementsEqual(a.iterator(), b.iterator())); assertFalse(Iterators.elementsEqual(b.iterator(), a.iterator())); // Different lengths. a = asList(4, 8, 15, 16, 23); b = asList(4, 8, 15, 16, 23, 42); assertFalse(Iterators.elementsEqual(a.iterator(), b.iterator())); assertFalse(Iterators.elementsEqual(b.iterator(), a.iterator())); // Different lengths, one is empty. a = Collections.emptySet(); b = asList(4, 8, 15, 16, 23, 42); assertFalse(Iterators.elementsEqual(a.iterator(), b.iterator())); assertFalse(Iterators.elementsEqual(b.iterator(), a.iterator())); } public void testPartition_badSize() { Iterator source = Iterators.singletonIterator(1); try { Iterators.partition(source, 0); fail(); } catch (IllegalArgumentException expected) { } } public void testPartition_empty() { Iterator source = Iterators.emptyIterator(); Iterator> partitions = Iterators.partition(source, 1); assertFalse(partitions.hasNext()); } public void testPartition_singleton1() { Iterator source = Iterators.singletonIterator(1); Iterator> partitions = Iterators.partition(source, 1); assertTrue(partitions.hasNext()); assertTrue(partitions.hasNext()); assertEquals(ImmutableList.of(1), partitions.next()); assertFalse(partitions.hasNext()); } public void testPartition_singleton2() { Iterator source = Iterators.singletonIterator(1); Iterator> partitions = Iterators.partition(source, 2); assertTrue(partitions.hasNext()); assertTrue(partitions.hasNext()); assertEquals(ImmutableList.of(1), partitions.next()); assertFalse(partitions.hasNext()); } @GwtIncompatible("fairly slow (~50s)") public void testPartition_general() { new IteratorTester>(5, IteratorFeature.UNMODIFIABLE, ImmutableList.of( asList(1, 2, 3), asList(4, 5, 6), asList(7)), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator> newTargetIterator() { Iterator source = Iterators.forArray(1, 2, 3, 4, 5, 6, 7); return Iterators.partition(source, 3); } }.test(); } public void testPartition_view() { List list = asList(1, 2); Iterator> partitions = Iterators.partition(list.iterator(), 1); // Changes before the partition is retrieved are reflected list.set(0, 3); List first = partitions.next(); // Changes after are not list.set(0, 4); assertEquals(ImmutableList.of(3), first); } @GwtIncompatible("?") // TODO: Figure out why this is failing in GWT. public void testPartitionRandomAccess() { Iterator source = asList(1, 2, 3).iterator(); Iterator> partitions = Iterators.partition(source, 2); assertTrue(partitions.next() instanceof RandomAccess); assertTrue(partitions.next() instanceof RandomAccess); } public void testPaddedPartition_badSize() { Iterator source = Iterators.singletonIterator(1); try { Iterators.paddedPartition(source, 0); fail(); } catch (IllegalArgumentException expected) { } } public void testPaddedPartition_empty() { Iterator source = Iterators.emptyIterator(); Iterator> partitions = Iterators.paddedPartition(source, 1); assertFalse(partitions.hasNext()); } public void testPaddedPartition_singleton1() { Iterator source = Iterators.singletonIterator(1); Iterator> partitions = Iterators.paddedPartition(source, 1); assertTrue(partitions.hasNext()); assertTrue(partitions.hasNext()); assertEquals(ImmutableList.of(1), partitions.next()); assertFalse(partitions.hasNext()); } public void testPaddedPartition_singleton2() { Iterator source = Iterators.singletonIterator(1); Iterator> partitions = Iterators.paddedPartition(source, 2); assertTrue(partitions.hasNext()); assertTrue(partitions.hasNext()); assertEquals(asList(1, null), partitions.next()); assertFalse(partitions.hasNext()); } @GwtIncompatible("fairly slow (~50s)") public void testPaddedPartition_general() { new IteratorTester>(5, IteratorFeature.UNMODIFIABLE, ImmutableList.of( asList(1, 2, 3), asList(4, 5, 6), asList(7, null, null)), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator> newTargetIterator() { Iterator source = Iterators.forArray(1, 2, 3, 4, 5, 6, 7); return Iterators.paddedPartition(source, 3); } }.test(); } public void testPaddedPartition_view() { List list = asList(1, 2); Iterator> partitions = Iterators.paddedPartition(list.iterator(), 1); // Changes before the PaddedPartition is retrieved are reflected list.set(0, 3); List first = partitions.next(); // Changes after are not list.set(0, 4); assertEquals(ImmutableList.of(3), first); } public void testPaddedPartitionRandomAccess() { Iterator source = asList(1, 2, 3).iterator(); Iterator> partitions = Iterators.paddedPartition(source, 2); assertTrue(partitions.next() instanceof RandomAccess); assertTrue(partitions.next() instanceof RandomAccess); } public void testForArrayEmpty() { String[] array = new String[0]; Iterator iterator = Iterators.forArray(array); assertFalse(iterator.hasNext()); try { iterator.next(); fail(); } catch (NoSuchElementException expected) {} } public void testForArrayTypical() { String[] array = {"foo", "bar"}; Iterator iterator = Iterators.forArray(array); assertTrue(iterator.hasNext()); assertEquals("foo", iterator.next()); assertTrue(iterator.hasNext()); try { iterator.remove(); fail(); } catch (UnsupportedOperationException expected) {} assertEquals("bar", iterator.next()); assertFalse(iterator.hasNext()); try { iterator.next(); fail(); } catch (NoSuchElementException expected) {} } public void testForArrayOffset() { String[] array = {"foo", "bar", "cat", "dog"}; Iterator iterator = Iterators.forArray(array, 1, 2); assertTrue(iterator.hasNext()); assertEquals("bar", iterator.next()); assertTrue(iterator.hasNext()); assertEquals("cat", iterator.next()); assertFalse(iterator.hasNext()); try { Iterators.forArray(array, 2, 3); fail(); } catch (IndexOutOfBoundsException expected) {} } public void testForArrayLength0() { String[] array = {"foo", "bar"}; assertFalse(Iterators.forArray(array, 0, 0).hasNext()); assertFalse(Iterators.forArray(array, 1, 0).hasNext()); assertFalse(Iterators.forArray(array, 2, 0).hasNext()); try { Iterators.forArray(array, -1, 0); fail(); } catch (IndexOutOfBoundsException expected) {} try { Iterators.forArray(array, 3, 0); fail(); } catch (IndexOutOfBoundsException expected) {} } @GwtIncompatible("unreasonable slow") public void testForArrayUsingTester() { new IteratorTester(6, UNMODIFIABLE, asList(1, 2, 3), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.forArray(1, 2, 3); } }.test(); } @GwtIncompatible("unreasonable slow") public void testForArrayWithOffsetUsingTester() { new IteratorTester(6, UNMODIFIABLE, asList(1, 2, 3), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.forArray(new Integer[] { 0, 1, 2, 3, 4 }, 1, 3); } }.test(); } public void testForEnumerationEmpty() { Enumeration enumer = enumerate(); Iterator iter = Iterators.forEnumeration(enumer); assertFalse(iter.hasNext()); try { iter.next(); fail(); } catch (NoSuchElementException expected) { } } public void testForEnumerationSingleton() { Enumeration enumer = enumerate(1); Iterator iter = Iterators.forEnumeration(enumer); assertTrue(iter.hasNext()); assertTrue(iter.hasNext()); assertEquals(1, (int) iter.next()); try { iter.remove(); fail(); } catch (UnsupportedOperationException expected) { } assertFalse(iter.hasNext()); try { iter.next(); fail(); } catch (NoSuchElementException expected) { } } public void testForEnumerationTypical() { Enumeration enumer = enumerate(1, 2, 3); Iterator iter = Iterators.forEnumeration(enumer); assertTrue(iter.hasNext()); assertEquals(1, (int) iter.next()); assertTrue(iter.hasNext()); assertEquals(2, (int) iter.next()); assertTrue(iter.hasNext()); assertEquals(3, (int) iter.next()); assertFalse(iter.hasNext()); } public void testAsEnumerationEmpty() { Iterator iter = Iterators.emptyIterator(); Enumeration enumer = Iterators.asEnumeration(iter); assertFalse(enumer.hasMoreElements()); try { enumer.nextElement(); fail(); } catch (NoSuchElementException expected) { } } public void testAsEnumerationSingleton() { Iterator iter = ImmutableList.of(1).iterator(); Enumeration enumer = Iterators.asEnumeration(iter); assertTrue(enumer.hasMoreElements()); assertTrue(enumer.hasMoreElements()); assertEquals(1, (int) enumer.nextElement()); assertFalse(enumer.hasMoreElements()); try { enumer.nextElement(); fail(); } catch (NoSuchElementException expected) { } } public void testAsEnumerationTypical() { Iterator iter = ImmutableList.of(1, 2, 3).iterator(); Enumeration enumer = Iterators.asEnumeration(iter); assertTrue(enumer.hasMoreElements()); assertEquals(1, (int) enumer.nextElement()); assertTrue(enumer.hasMoreElements()); assertEquals(2, (int) enumer.nextElement()); assertTrue(enumer.hasMoreElements()); assertEquals(3, (int) enumer.nextElement()); assertFalse(enumer.hasMoreElements()); } private static Enumeration enumerate(Integer... ints) { Vector vector = new Vector(); vector.addAll(asList(ints)); return vector.elements(); } public void testToString() { List list = Collections.emptyList(); assertEquals("[]", Iterators.toString(list.iterator())); list = Lists.newArrayList("yam", "bam", "jam", "ham"); assertEquals("[yam, bam, jam, ham]", Iterators.toString(list.iterator())); } public void testLimit() { List list = newArrayList(); try { Iterators.limit(list.iterator(), -1); fail("expected exception"); } catch (IllegalArgumentException expected) { // expected } assertFalse(Iterators.limit(list.iterator(), 0).hasNext()); assertFalse(Iterators.limit(list.iterator(), 1).hasNext()); list.add("cool"); assertFalse(Iterators.limit(list.iterator(), 0).hasNext()); assertEquals(list, newArrayList(Iterators.limit(list.iterator(), 1))); assertEquals(list, newArrayList(Iterators.limit(list.iterator(), 2))); list.add("pants"); assertFalse(Iterators.limit(list.iterator(), 0).hasNext()); assertEquals(ImmutableList.of("cool"), newArrayList(Iterators.limit(list.iterator(), 1))); assertEquals(list, newArrayList(Iterators.limit(list.iterator(), 2))); assertEquals(list, newArrayList(Iterators.limit(list.iterator(), 3))); } public void testLimitRemove() { List list = newArrayList(); list.add("cool"); list.add("pants"); Iterator iterator = Iterators.limit(list.iterator(), 1); iterator.next(); iterator.remove(); assertFalse(iterator.hasNext()); assertEquals(1, list.size()); assertEquals("pants", list.get(0)); } @GwtIncompatible("fairly slow (~30s)") public void testLimitUsingIteratorTester() { final List list = Lists.newArrayList(1, 2, 3, 4, 5); new IteratorTester(5, MODIFIABLE, newArrayList(1, 2, 3), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.limit(Lists.newArrayList(list).iterator(), 3); } }.test(); } public void testGetNext_withDefault_singleton() { Iterator iterator = Collections.singletonList("foo").iterator(); assertEquals("foo", Iterators.getNext(iterator, "bar")); } public void testGetNext_withDefault_empty() { Iterator iterator = Iterators.emptyIterator(); assertEquals("bar", Iterators.getNext(iterator, "bar")); } public void testGetNext_withDefault_empty_null() { Iterator iterator = Iterators.emptyIterator(); assertNull(Iterators.getNext(iterator, null)); } public void testGetNext_withDefault_two() { Iterator iterator = asList("foo", "bar").iterator(); assertEquals("foo", Iterators.getNext(iterator, "x")); } public void testGetLast_basic() { List list = newArrayList(); list.add("a"); list.add("b"); assertEquals("b", getLast(list.iterator())); } public void testGetLast_exception() { List list = newArrayList(); try { getLast(list.iterator()); fail(); } catch (NoSuchElementException expected) { } } public void testGetLast_withDefault_singleton() { Iterator iterator = Collections.singletonList("foo").iterator(); assertEquals("foo", Iterators.getLast(iterator, "bar")); } public void testGetLast_withDefault_empty() { Iterator iterator = Iterators.emptyIterator(); assertEquals("bar", Iterators.getLast(iterator, "bar")); } public void testGetLast_withDefault_empty_null() { Iterator iterator = Iterators.emptyIterator(); assertNull(Iterators.getLast(iterator, null)); } public void testGetLast_withDefault_two() { Iterator iterator = asList("foo", "bar").iterator(); assertEquals("bar", Iterators.getLast(iterator, "x")); } public void testGet_basic() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); assertEquals("b", get(iterator, 1)); assertFalse(iterator.hasNext()); } public void testGet_atSize() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); try { get(iterator, 2); fail(); } catch (IndexOutOfBoundsException expected) {} assertFalse(iterator.hasNext()); } public void testGet_pastEnd() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); try { get(iterator, 5); fail(); } catch (IndexOutOfBoundsException expected) {} assertFalse(iterator.hasNext()); } public void testGet_empty() { List list = newArrayList(); Iterator iterator = list.iterator(); try { get(iterator, 0); fail(); } catch (IndexOutOfBoundsException expected) {} assertFalse(iterator.hasNext()); } public void testGet_negativeIndex() { List list = newArrayList("a", "b", "c"); Iterator iterator = list.iterator(); try { get(iterator, -1); fail(); } catch (IndexOutOfBoundsException expected) {} } public void testGet_withDefault_basic() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); assertEquals("a", get(iterator, 0, "c")); assertTrue(iterator.hasNext()); } public void testGet_withDefault_atSize() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); assertEquals("c", get(iterator, 2, "c")); assertFalse(iterator.hasNext()); } public void testGet_withDefault_pastEnd() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); assertEquals("c", get(iterator, 3, "c")); assertFalse(iterator.hasNext()); } public void testGet_withDefault_negativeIndex() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); try { get(iterator, -1, "c"); fail(); } catch (IndexOutOfBoundsException expected) { // pass } assertTrue(iterator.hasNext()); } public void testSkip_basic() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); skip(iterator, 1); assertEquals("b", iterator.next()); } public void testSkip_pastEnd() { List list = newArrayList(); list.add("a"); list.add("b"); Iterator iterator = list.iterator(); skip(iterator, 5); assertFalse(iterator.hasNext()); } public void testSkip_illegalArgument() { List list = newArrayList("a", "b", "c"); Iterator iterator = list.iterator(); try { skip(iterator, -1); fail(); } catch (IllegalArgumentException expected) {} } public void testFrequency() { List list = newArrayList("a", null, "b", null, "a", null); assertEquals(2, Iterators.frequency(list.iterator(), "a")); assertEquals(1, Iterators.frequency(list.iterator(), "b")); assertEquals(0, Iterators.frequency(list.iterator(), "c")); assertEquals(0, Iterators.frequency(list.iterator(), 4.2)); assertEquals(3, Iterators.frequency(list.iterator(), null)); } @GwtIncompatible("slow (~4s)") public void testSingletonIterator() { new IteratorTester( 3, UNMODIFIABLE, singleton(1), IteratorTester.KnownOrder.KNOWN_ORDER) { @Override protected Iterator newTargetIterator() { return Iterators.singletonIterator(1); } }.test(); } public void testRemoveAll() { List list = newArrayList("a", "b", "c", "d", "e"); assertTrue(Iterators.removeAll( list.iterator(), newArrayList("b", "d", "f"))); assertEquals(newArrayList("a", "c", "e"), list); assertFalse(Iterators.removeAll( list.iterator(), newArrayList("x", "y", "z"))); assertEquals(newArrayList("a", "c", "e"), list); } public void testRemoveIf() { List list = newArrayList("a", "b", "c", "d", "e"); assertTrue(Iterators.removeIf( list.iterator(), new Predicate() { @Override public boolean apply(String s) { return s.equals("b") || s.equals("d") || s.equals("f"); } })); assertEquals(newArrayList("a", "c", "e"), list); assertFalse(Iterators.removeIf( list.iterator(), new Predicate() { @Override public boolean apply(String s) { return s.equals("x") || s.equals("y") || s.equals("z"); } })); assertEquals(newArrayList("a", "c", "e"), list); } public void testRetainAll() { List list = newArrayList("a", "b", "c", "d", "e"); assertTrue(Iterators.retainAll( list.iterator(), newArrayList("b", "d", "f"))); assertEquals(newArrayList("b", "d"), list); assertFalse(Iterators.retainAll( list.iterator(), newArrayList("b", "e", "d"))); assertEquals(newArrayList("b", "d"), list); } @GwtIncompatible("ListTestSuiteBuilder") private static Test testsForRemoveAllAndRetainAll() { return ListTestSuiteBuilder.using(new TestStringListGenerator() { @Override public List create(final String[] elements) { final List delegate = newArrayList(elements); return new ForwardingList() { @Override protected List delegate() { return delegate; } @Override public boolean removeAll(Collection c) { return Iterators.removeAll(iterator(), c); } @Override public boolean retainAll(Collection c) { return Iterators.retainAll(iterator(), c); } }; } }) .named("ArrayList with Iterators.removeAll and retainAll") .withFeatures( ListFeature.GENERAL_PURPOSE, CollectionFeature.ALLOWS_NULL_VALUES, CollectionSize.ANY) .createTestSuite(); } public void testConsumingIterator() { // Test data List list = Lists.newArrayList("a", "b"); // Test & Verify Iterator consumingIterator = Iterators.consumingIterator(list.iterator()); ASSERT.that(list).hasContentsInOrder("a", "b"); assertTrue(consumingIterator.hasNext()); ASSERT.that(list).hasContentsInOrder("a", "b"); assertEquals("a", consumingIterator.next()); ASSERT.that(list).hasContentsInOrder("b"); assertTrue(consumingIterator.hasNext()); assertEquals("b", consumingIterator.next()); ASSERT.that(list).isEmpty(); assertFalse(consumingIterator.hasNext()); } @GwtIncompatible("?") // TODO: Figure out why this is failing in GWT. public void testConsumingIterator_duelingIterators() { // Test data List list = Lists.newArrayList("a", "b"); // Test & Verify Iterator i1 = Iterators.consumingIterator(list.iterator()); Iterator i2 = Iterators.consumingIterator(list.iterator()); i1.next(); try { i2.next(); fail("Concurrent modification should throw an exception."); } catch (ConcurrentModificationException cme) { // Pass } } public void testIndexOf_consumedData() { Iterator iterator = Lists.newArrayList("manny", "mo", "jack").iterator(); assertEquals(1, Iterators.indexOf(iterator, Predicates.equalTo("mo"))); assertEquals("jack", iterator.next()); assertFalse(iterator.hasNext()); } public void testIndexOf_consumedDataWithDuplicates() { Iterator iterator = Lists.newArrayList("manny", "mo", "mo", "jack").iterator(); assertEquals(1, Iterators.indexOf(iterator, Predicates.equalTo("mo"))); assertEquals("mo", iterator.next()); assertEquals("jack", iterator.next()); assertFalse(iterator.hasNext()); } public void testIndexOf_consumedDataNoMatch() { Iterator iterator = Lists.newArrayList("manny", "mo", "mo", "jack").iterator(); assertEquals(-1, Iterators.indexOf(iterator, Predicates.equalTo("bob"))); assertFalse(iterator.hasNext()); } @SuppressWarnings("deprecation") public void testUnmodifiableIteratorShortCircuit() { Iterator mod = Lists.newArrayList("a", "b", "c").iterator(); UnmodifiableIterator unmod = Iterators.unmodifiableIterator(mod); assertNotSame(mod, unmod); assertSame(unmod, Iterators.unmodifiableIterator(unmod)); assertSame(unmod, Iterators.unmodifiableIterator((Iterator) unmod)); } @SuppressWarnings("deprecation") public void testPeekingIteratorShortCircuit() { Iterator nonpeek = Lists.newArrayList("a", "b", "c").iterator(); PeekingIterator peek = Iterators.peekingIterator(nonpeek); assertNotSame(peek, nonpeek); assertSame(peek, Iterators.peekingIterator(peek)); assertSame(peek, Iterators.peekingIterator((Iterator) peek)); } }