| /* | |
| * Copyright (c) 2013, 2017, Oracle and/or its affiliates. All rights reserved. | |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | |
| * | |
| * This code is free software; you can redistribute it and/or modify it | |
| * under the terms of the GNU General Public License version 2 only, as | |
| * published by the Free Software Foundation. | |
| * | |
| * This code is distributed in the hope that it will be useful, but WITHOUT | |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| * version 2 for more details (a copy is included in the LICENSE file that | |
| * accompanied this code). | |
| * | |
| * You should have received a copy of the GNU General Public License version | |
| * 2 along with this work; if not, write to the Free Software Foundation, | |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| * | |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA | |
| * or visit www.oracle.com if you need additional information or have any | |
| * questions. | |
| */ | |
| /** | |
| * @test | |
| * @bug 8005698 | |
| * @library /lib/testlibrary/bootlib | |
| * @build java.base/java.util.SpliteratorTestHelper | |
| * @run testng SpliteratorCollisions | |
| * @summary Spliterator traversing and splitting hash maps containing colliding hashes | |
| */ | |
| import org.testng.annotations.DataProvider; | |
| import org.testng.annotations.Test; | |
| import java.util.ArrayList; | |
| import java.util.Arrays; | |
| import java.util.Collection; | |
| import java.util.Collections; | |
| import java.util.HashMap; | |
| import java.util.HashSet; | |
| import java.util.LinkedHashMap; | |
| import java.util.LinkedHashSet; | |
| import java.util.List; | |
| import java.util.Map; | |
| import java.util.Spliterator; | |
| import java.util.SpliteratorTestHelper; | |
| import java.util.TreeSet; | |
| import java.util.function.Function; | |
| import java.util.function.Supplier; | |
| import java.util.function.UnaryOperator; | |
| public class SpliteratorCollisions extends SpliteratorTestHelper { | |
| private static final List<Integer> SIZES = Arrays.asList(0, 1, 10, 100, 1000); | |
| private static class SpliteratorDataBuilder<T> { | |
| List<Object[]> data; | |
| List<T> exp; | |
| Map<T, T> mExp; | |
| SpliteratorDataBuilder(List<Object[]> data, List<T> exp) { | |
| this.data = data; | |
| this.exp = exp; | |
| this.mExp = createMap(exp); | |
| } | |
| Map<T, T> createMap(List<T> l) { | |
| Map<T, T> m = new LinkedHashMap<>(); | |
| for (T t : l) { | |
| m.put(t, t); | |
| } | |
| return m; | |
| } | |
| void add(String description, Collection<?> expected, Supplier<Spliterator<?>> s) { | |
| description = joiner(description).toString(); | |
| data.add(new Object[]{description, expected, s}); | |
| } | |
| void add(String description, Supplier<Spliterator<?>> s) { | |
| add(description, exp, s); | |
| } | |
| void addCollection(Function<Collection<T>, ? extends Collection<T>> c) { | |
| add("new " + c.apply(Collections.<T>emptyList()).getClass().getName() + ".spliterator()", | |
| () -> c.apply(exp).spliterator()); | |
| } | |
| void addList(Function<Collection<T>, ? extends List<T>> l) { | |
| // @@@ If collection is instance of List then add sub-list tests | |
| addCollection(l); | |
| } | |
| void addMap(Function<Map<T, T>, ? extends Map<T, T>> m) { | |
| String description = "new " + m.apply(Collections.<T, T>emptyMap()).getClass().getName(); | |
| add(description + ".keySet().spliterator()", () -> m.apply(mExp).keySet().spliterator()); | |
| add(description + ".values().spliterator()", () -> m.apply(mExp).values().spliterator()); | |
| add(description + ".entrySet().spliterator()", mExp.entrySet(), () -> m.apply(mExp).entrySet().spliterator()); | |
| } | |
| StringBuilder joiner(String description) { | |
| return new StringBuilder(description). | |
| append(" {"). | |
| append("size=").append(exp.size()). | |
| append("}"); | |
| } | |
| } | |
| static Object[][] spliteratorDataProvider; | |
| @DataProvider(name = "HashableIntSpliterator") | |
| public static Object[][] spliteratorDataProvider() { | |
| if (spliteratorDataProvider != null) { | |
| return spliteratorDataProvider; | |
| } | |
| List<Object[]> data = new ArrayList<>(); | |
| for (int size : SIZES) { | |
| List<HashableInteger> exp = listIntRange(size, false); | |
| SpliteratorDataBuilder<HashableInteger> db = new SpliteratorDataBuilder<>(data, exp); | |
| // Maps | |
| db.addMap(HashMap::new); | |
| db.addMap(LinkedHashMap::new); | |
| // Collections that use HashMap | |
| db.addCollection(HashSet::new); | |
| db.addCollection(LinkedHashSet::new); | |
| db.addCollection(TreeSet::new); | |
| } | |
| return spliteratorDataProvider = data.toArray(new Object[0][]); | |
| } | |
| static Object[][] spliteratorDataProviderWithNull; | |
| @DataProvider(name = "HashableIntSpliteratorWithNull") | |
| public static Object[][] spliteratorNullDataProvider() { | |
| if (spliteratorDataProviderWithNull != null) { | |
| return spliteratorDataProviderWithNull; | |
| } | |
| List<Object[]> data = new ArrayList<>(); | |
| for (int size : SIZES) { | |
| List<HashableInteger> exp = listIntRange(size, true); | |
| SpliteratorDataBuilder<HashableInteger> db = new SpliteratorDataBuilder<>(data, exp); | |
| // Maps | |
| db.addMap(HashMap::new); | |
| db.addMap(LinkedHashMap::new); | |
| // TODO: add this back in if we decide to keep TreeBin in WeakHashMap | |
| //db.addMap(WeakHashMap::new); | |
| // Collections that use HashMap | |
| db.addCollection(HashSet::new); | |
| db.addCollection(LinkedHashSet::new); | |
| // db.addCollection(TreeSet::new); | |
| } | |
| return spliteratorDataProviderWithNull = data.toArray(new Object[0][]); | |
| } | |
| static final class HashableInteger implements Comparable<HashableInteger> { | |
| final int value; | |
| final int hashmask; //yes duplication | |
| HashableInteger(int value, int hashmask) { | |
| this.value = value; | |
| this.hashmask = hashmask; | |
| } | |
| @Override | |
| public boolean equals(Object obj) { | |
| if (obj instanceof HashableInteger) { | |
| HashableInteger other = (HashableInteger) obj; | |
| return other.value == value; | |
| } | |
| return false; | |
| } | |
| @Override | |
| public int hashCode() { | |
| return value % hashmask; | |
| } | |
| @Override | |
| public int compareTo(HashableInteger o) { | |
| return value - o.value; | |
| } | |
| @Override | |
| public String toString() { | |
| return Integer.toString(value); | |
| } | |
| } | |
| private static List<HashableInteger> listIntRange(int upTo, boolean withNull) { | |
| List<HashableInteger> exp = new ArrayList<>(); | |
| if (withNull) { | |
| exp.add(null); | |
| } | |
| for (int i = 0; i < upTo; i++) { | |
| exp.add(new HashableInteger(i, 10)); | |
| } | |
| return Collections.unmodifiableList(exp); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testNullPointerException(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| assertThrowsNPE(() -> s.get().forEachRemaining(null)); | |
| assertThrowsNPE(() -> s.get().tryAdvance(null)); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testNullPointerExceptionWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| assertThrowsNPE(() -> s.get().forEachRemaining(null)); | |
| assertThrowsNPE(() -> s.get().tryAdvance(null)); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testForEach(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testForEach(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testForEachWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testForEach(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testTryAdvance(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testTryAdvance(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testTryAdvanceWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testTryAdvance(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testMixedTryAdvanceForEach(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testMixedTryAdvanceForEach(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testMixedTryAdvanceForEachWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testMixedTryAdvanceForEach(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testMixedTraverseAndSplit(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testMixedTraverseAndSplit(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testMixedTraverseAndSplitWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testMixedTraverseAndSplit(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testSplitAfterFullTraversal(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitAfterFullTraversal(s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testSplitAfterFullTraversalWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitAfterFullTraversal(s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testSplitOnce(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitOnce(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testSplitOnceWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitOnce(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testSplitSixDeep(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitSixDeep(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testSplitSixDeepWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitSixDeep(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliterator") | |
| void testSplitUntilNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitUntilNull(exp, s, UnaryOperator.identity()); | |
| } | |
| @Test(dataProvider = "HashableIntSpliteratorWithNull") | |
| void testSplitUntilNullWithNull(String description, | |
| Collection<HashableInteger> exp, | |
| Supplier<Spliterator<HashableInteger>> s) { | |
| testSplitUntilNull(exp, s, UnaryOperator.identity()); | |
| } | |
| } |