blob: 6a38fcdb5896558be3a07ed62fb5c31c76ffac75 [file] [log] [blame]
/*
* 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.truth.Truth.assertThat;
import com.google.common.annotations.GwtCompatible;
import com.google.common.annotations.GwtIncompatible;
import com.google.common.base.Equivalence;
import com.google.common.collect.ImmutableSet.Builder;
import com.google.common.collect.testing.ListTestSuiteBuilder;
import com.google.common.collect.testing.SetTestSuiteBuilder;
import com.google.common.collect.testing.TestStringSetGenerator;
import com.google.common.collect.testing.features.CollectionFeature;
import com.google.common.collect.testing.features.CollectionSize;
import com.google.common.collect.testing.google.SetGenerators.DegeneratedImmutableSetGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetAsListGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetCopyOfGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetSizedBuilderGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetTooBigBuilderGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetTooSmallBuilderGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetUnsizedBuilderGenerator;
import com.google.common.collect.testing.google.SetGenerators.ImmutableSetWithBadHashesGenerator;
import com.google.common.testing.CollectorTester;
import com.google.common.testing.EqualsTester;
import com.google.errorprone.annotations.CanIgnoreReturnValue;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import java.util.function.BiPredicate;
import java.util.stream.Collector;
import junit.framework.Test;
import junit.framework.TestSuite;
import org.checkerframework.checker.nullness.qual.Nullable;
/**
* Unit test for {@link ImmutableSet}.
*
* @author Kevin Bourrillion
* @author Jared Levy
* @author Nick Kralevich
*/
@GwtCompatible(emulated = true)
public class ImmutableSetTest extends AbstractImmutableSetTest {
@GwtIncompatible // suite
public static Test suite() {
TestSuite suite = new TestSuite();
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetCopyOfGenerator())
.named(ImmutableSetTest.class.getName())
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetUnsizedBuilderGenerator())
.named(ImmutableSetTest.class.getName() + ", with unsized builder")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(
new TestStringSetGenerator() {
@Override
protected Set<String> create(String[] elements) {
ImmutableSet.Builder<String> builder = ImmutableSet.builder();
builder.forceJdk();
builder.add(elements);
return builder.build();
}
})
.named(ImmutableSetTest.class.getName() + ", with JDK builder")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetSizedBuilderGenerator())
.named(ImmutableSetTest.class.getName() + ", with exactly sized builder")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetTooBigBuilderGenerator())
.named(ImmutableSetTest.class.getName() + ", with oversized builder")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetTooSmallBuilderGenerator())
.named(ImmutableSetTest.class.getName() + ", with undersized builder")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new ImmutableSetWithBadHashesGenerator())
.named(ImmutableSetTest.class.getName() + ", with bad hashes")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
SetTestSuiteBuilder.using(new DegeneratedImmutableSetGenerator())
.named(ImmutableSetTest.class.getName() + ", degenerate")
.withFeatures(
CollectionSize.ONE,
CollectionFeature.KNOWN_ORDER,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTest(
ListTestSuiteBuilder.using(new ImmutableSetAsListGenerator())
.named("ImmutableSet.asList")
.withFeatures(
CollectionSize.ANY,
CollectionFeature.REJECTS_DUPLICATES_AT_CREATION,
CollectionFeature.SERIALIZABLE,
CollectionFeature.ALLOWS_NULL_QUERIES)
.createTestSuite());
suite.addTestSuite(ImmutableSetTest.class);
return suite;
}
@Override
protected <E extends Comparable<? super E>> Set<E> of() {
return ImmutableSet.of();
}
@Override
protected <E extends Comparable<? super E>> Set<E> of(E e) {
return ImmutableSet.of(e);
}
@Override
protected <E extends Comparable<? super E>> Set<E> of(E e1, E e2) {
return ImmutableSet.of(e1, e2);
}
@Override
protected <E extends Comparable<? super E>> Set<E> of(E e1, E e2, E e3) {
return ImmutableSet.of(e1, e2, e3);
}
@Override
protected <E extends Comparable<? super E>> Set<E> of(E e1, E e2, E e3, E e4) {
return ImmutableSet.of(e1, e2, e3, e4);
}
@Override
protected <E extends Comparable<? super E>> Set<E> of(E e1, E e2, E e3, E e4, E e5) {
return ImmutableSet.of(e1, e2, e3, e4, e5);
}
@SuppressWarnings("unchecked")
@Override
protected <E extends Comparable<? super E>> Set<E> of(
E e1, E e2, E e3, E e4, E e5, E e6, E... rest) {
return ImmutableSet.of(e1, e2, e3, e4, e5, e6, rest);
}
@Override
protected <E extends Comparable<? super E>> Set<E> copyOf(E[] elements) {
return ImmutableSet.copyOf(elements);
}
@Override
protected <E extends Comparable<? super E>> Set<E> copyOf(Collection<? extends E> elements) {
return ImmutableSet.copyOf(elements);
}
@Override
protected <E extends Comparable<? super E>> Set<E> copyOf(Iterable<? extends E> elements) {
return ImmutableSet.copyOf(elements);
}
@Override
protected <E extends Comparable<? super E>> Set<E> copyOf(Iterator<? extends E> elements) {
return ImmutableSet.copyOf(elements);
}
public void testCreation_allDuplicates() {
ImmutableSet<String> set = ImmutableSet.copyOf(Lists.newArrayList("a", "a"));
assertTrue(set instanceof SingletonImmutableSet);
assertEquals(Lists.newArrayList("a"), Lists.newArrayList(set));
}
public void testCreation_oneDuplicate() {
// now we'll get the varargs overload
ImmutableSet<String> set =
ImmutableSet.of("a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "a");
assertEquals(
Lists.newArrayList("a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m"),
Lists.newArrayList(set));
}
public void testCreation_manyDuplicates() {
// now we'll get the varargs overload
ImmutableSet<String> set =
ImmutableSet.of("a", "b", "c", "c", "c", "c", "b", "b", "a", "a", "c", "c", "c", "a");
assertThat(set).containsExactly("a", "b", "c").inOrder();
}
public void testCreation_arrayOfArray() {
String[] array = new String[] {"a"};
Set<String[]> set = ImmutableSet.<String[]>of(array);
assertEquals(Collections.singleton(array), set);
}
@GwtIncompatible // ImmutableSet.chooseTableSize
public void testChooseTableSize() {
assertEquals(8, ImmutableSet.chooseTableSize(3));
assertEquals(8, ImmutableSet.chooseTableSize(4));
assertEquals(1 << 29, ImmutableSet.chooseTableSize(1 << 28));
assertEquals(1 << 29, ImmutableSet.chooseTableSize((1 << 29) * 3 / 5));
// Now we hit the cap
assertEquals(1 << 30, ImmutableSet.chooseTableSize(1 << 29));
assertEquals(1 << 30, ImmutableSet.chooseTableSize((1 << 30) - 1));
// Now we've gone too far
try {
ImmutableSet.chooseTableSize(1 << 30);
fail();
} catch (IllegalArgumentException expected) {
}
}
@GwtIncompatible // RegularImmutableSet.table not in emulation
public void testResizeTable() {
verifyTableSize(100, 2, 8);
verifyTableSize(100, 5, 8);
verifyTableSize(100, 33, 64);
verifyTableSize(17, 17, 32);
verifyTableSize(17, 16, 32);
verifyTableSize(17, 15, 32);
}
@GwtIncompatible // RegularImmutableSet.table not in emulation
private void verifyTableSize(int inputSize, int setSize, int tableSize) {
Builder<Integer> builder = ImmutableSet.builder();
for (int i = 0; i < inputSize; i++) {
builder.add(i % setSize);
}
ImmutableSet<Integer> set = builder.build();
assertTrue(set instanceof RegularImmutableSet);
assertEquals(
"Input size " + inputSize + " and set size " + setSize,
tableSize,
((RegularImmutableSet<Integer>) set).table.length);
}
public void testCopyOf_copiesImmutableSortedSet() {
ImmutableSortedSet<String> sortedSet = ImmutableSortedSet.of("a");
ImmutableSet<String> copy = ImmutableSet.copyOf(sortedSet);
assertNotSame(sortedSet, copy);
}
public void testToImmutableSet() {
Collector<String, ?, ImmutableSet<String>> collector = ImmutableSet.toImmutableSet();
Equivalence<ImmutableSet<String>> equivalence =
Equivalence.equals().onResultOf(ImmutableSet::asList);
CollectorTester.of(collector, equivalence)
.expectCollects(ImmutableSet.of("a", "b", "c", "d"), "a", "b", "a", "c", "b", "b", "d");
}
public void testToImmutableSet_duplicates() {
class TypeWithDuplicates {
final int a;
final int b;
TypeWithDuplicates(int a, int b) {
this.a = a;
this.b = b;
}
@Override
public int hashCode() {
return a;
}
@Override
public boolean equals(Object obj) {
return obj instanceof TypeWithDuplicates && ((TypeWithDuplicates) obj).a == a;
}
public boolean fullEquals(TypeWithDuplicates other) {
return other != null && a == other.a && b == other.b;
}
}
Collector<TypeWithDuplicates, ?, ImmutableSet<TypeWithDuplicates>> collector =
ImmutableSet.toImmutableSet();
BiPredicate<ImmutableSet<TypeWithDuplicates>, ImmutableSet<TypeWithDuplicates>> equivalence =
(set1, set2) -> {
if (!set1.equals(set2)) {
return false;
}
for (int i = 0; i < set1.size(); i++) {
if (!set1.asList().get(i).fullEquals(set2.asList().get(i))) {
return false;
}
}
return true;
};
TypeWithDuplicates a = new TypeWithDuplicates(1, 1);
TypeWithDuplicates b1 = new TypeWithDuplicates(2, 1);
TypeWithDuplicates b2 = new TypeWithDuplicates(2, 2);
TypeWithDuplicates c = new TypeWithDuplicates(3, 1);
CollectorTester.of(collector, equivalence)
.expectCollects(ImmutableSet.of(a, b1, c), a, b1, c, b2);
}
@GwtIncompatible // GWT is single threaded
public void testCopyOf_threadSafe() {
verifyThreadSafe();
}
@Override
<E extends Comparable<E>> Builder<E> builder() {
return ImmutableSet.builder();
}
@Override
int getComplexBuilderSetLastElement() {
return LAST_COLOR_ADDED;
}
public void testEquals() {
new EqualsTester()
.addEqualityGroup(ImmutableSet.of(), ImmutableSet.of())
.addEqualityGroup(ImmutableSet.of(1), ImmutableSet.of(1), ImmutableSet.of(1, 1))
.addEqualityGroup(ImmutableSet.of(1, 2, 1), ImmutableSet.of(2, 1, 1))
.testEquals();
}
/**
* A Comparable wrapper around a String which executes callbacks on calls to hashCode, equals, and
* compareTo.
*/
private static class CountsHashCodeAndEquals implements Comparable<CountsHashCodeAndEquals> {
private final String delegateString;
private final Runnable onHashCode;
private final Runnable onEquals;
private final Runnable onCompareTo;
CountsHashCodeAndEquals(
String delegateString, Runnable onHashCode, Runnable onEquals, Runnable onCompareTo) {
this.delegateString = delegateString;
this.onHashCode = onHashCode;
this.onEquals = onEquals;
this.onCompareTo = onCompareTo;
}
@Override
public int hashCode() {
onHashCode.run();
return delegateString.hashCode();
}
@Override
public boolean equals(@Nullable Object other) {
onEquals.run();
return other instanceof CountsHashCodeAndEquals
&& delegateString.equals(((CountsHashCodeAndEquals) other).delegateString);
}
@Override
public int compareTo(CountsHashCodeAndEquals o) {
onCompareTo.run();
return delegateString.compareTo(o.delegateString);
}
}
/** A holder of counters for calls to hashCode, equals, and compareTo. */
private static final class CallsCounter {
long hashCode;
long equals;
long compareTo;
long total() {
return hashCode + equals + compareTo;
}
void zero() {
hashCode = 0;
equals = 0;
compareTo = 0;
}
}
/** All the ways to construct an ImmutableSet. */
enum ConstructionPathway {
OF {
@Override
ImmutableSet<?> create(List<?> list) {
Object o1 = list.get(0);
Object o2 = list.get(1);
Object o3 = list.get(2);
Object o4 = list.get(3);
Object o5 = list.get(4);
Object o6 = list.get(5);
Object[] rest = list.subList(6, list.size()).toArray();
return ImmutableSet.of(o1, o2, o3, o4, o5, o6, rest);
}
},
COPY_OF_ARRAY {
@Override
ImmutableSet<?> create(List<?> list) {
return ImmutableSet.copyOf(list.toArray());
}
},
COPY_OF_LIST {
@Override
ImmutableSet<?> create(List<?> list) {
return ImmutableSet.copyOf(list);
}
},
BUILDER_ADD_ONE_BY_ONE {
@Override
ImmutableSet<?> create(List<?> list) {
ImmutableSet.Builder<Object> builder = ImmutableSet.builder();
for (Object o : list) {
builder.add(o);
}
return builder.build();
}
},
BUILDER_ADD_ARRAY {
@Override
ImmutableSet<?> create(List<?> list) {
ImmutableSet.Builder<Object> builder = ImmutableSet.builder();
builder.add(list.toArray());
return builder.build();
}
},
BUILDER_ADD_LIST {
@Override
ImmutableSet<?> create(List<?> list) {
ImmutableSet.Builder<Object> builder = ImmutableSet.builder();
builder.addAll(list);
return builder.build();
}
};
@CanIgnoreReturnValue
abstract ImmutableSet<?> create(List<?> list);
}
/**
* Returns a list of objects with the same hash code, of size 2^power, counting calls to equals,
* hashCode, and compareTo in counter.
*/
static List<CountsHashCodeAndEquals> createAdversarialInput(int power, CallsCounter counter) {
String str1 = "Aa";
String str2 = "BB";
assertEquals(str1.hashCode(), str2.hashCode());
List<String> haveSameHashes2 = Arrays.asList(str1, str2);
List<CountsHashCodeAndEquals> result =
Lists.newArrayList(
Lists.transform(
Lists.cartesianProduct(Collections.nCopies(power, haveSameHashes2)),
strs ->
new CountsHashCodeAndEquals(
String.join("", strs),
() -> counter.hashCode++,
() -> counter.equals++,
() -> counter.compareTo++)));
assertEquals(
result.get(0).delegateString.hashCode(),
result.get(result.size() - 1).delegateString.hashCode());
return result;
}
@GwtIncompatible
public void testResistsHashFloodingInConstruction() {
CallsCounter smallCounter = new CallsCounter();
List<CountsHashCodeAndEquals> haveSameHashesSmall = createAdversarialInput(10, smallCounter);
int smallSize = haveSameHashesSmall.size();
CallsCounter largeCounter = new CallsCounter();
List<CountsHashCodeAndEquals> haveSameHashesLarge = createAdversarialInput(15, largeCounter);
int largeSize = haveSameHashesLarge.size();
for (ConstructionPathway pathway : ConstructionPathway.values()) {
smallCounter.zero();
pathway.create(haveSameHashesSmall);
largeCounter.zero();
pathway.create(haveSameHashesLarge);
double ratio = (double) largeCounter.total() / smallCounter.total();
assertThat(ratio)
.named(
"ratio of equals/hashCode/compareTo operations to build an ImmutableSet via pathway "
+ "%s of size %s versus size %s",
pathway, haveSameHashesLarge.size(), haveSameHashesSmall.size())
.isAtMost(2.0 * (largeSize * Math.log(largeSize)) / (smallSize * Math.log(smallSize)));
// We allow up to 2x wobble in the constant factors.
}
}
@GwtIncompatible
public void testResistsHashFloodingOnContains() {
CallsCounter smallCounter = new CallsCounter();
List<CountsHashCodeAndEquals> haveSameHashesSmall = createAdversarialInput(10, smallCounter);
ImmutableSet<?> smallSet = ConstructionPathway.COPY_OF_LIST.create(haveSameHashesSmall);
long worstCaseOpsSmall = worstCaseQueryOperations(smallSet, smallCounter);
CallsCounter largeCounter = new CallsCounter();
List<CountsHashCodeAndEquals> haveSameHashesLarge = createAdversarialInput(15, largeCounter);
ImmutableSet<?> largeSet = ConstructionPathway.COPY_OF_LIST.create(haveSameHashesLarge);
long worstCaseOpsLarge = worstCaseQueryOperations(largeSet, largeCounter);
double ratio = (double) worstCaseOpsLarge / worstCaseOpsSmall;
int smallSize = haveSameHashesSmall.size();
int largeSize = haveSameHashesLarge.size();
assertThat(ratio)
.named(
"ratio of equals/hashCode/compareTo operations to worst-case query an ImmutableSet "
+ "of size %s versus size %s",
haveSameHashesLarge.size(), haveSameHashesSmall.size())
.isAtMost(2 * Math.log(largeSize) / Math.log(smallSize));
// We allow up to 2x wobble in the constant factors.
}
private static long worstCaseQueryOperations(Set<?> set, CallsCounter counter) {
long worstCalls = 0;
for (Object k : set) {
counter.zero();
if (set.contains(k)) {
worstCalls = Math.max(worstCalls, counter.total());
}
}
return worstCalls;
}
}