Home | History | Annotate | Download | only in collect
      1 /*
      2  * Copyright (C) 2007 The Guava Authors
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  * http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package com.google.common.collect;
     18 
     19 import static com.google.common.truth.Truth.assertThat;
     20 
     21 import com.google.common.annotations.GwtCompatible;
     22 import com.google.common.collect.ImmutableSet.Builder;
     23 import com.google.common.testing.EqualsTester;
     24 
     25 import java.util.Collection;
     26 import java.util.Collections;
     27 import java.util.Iterator;
     28 import java.util.Set;
     29 
     30 /**
     31  * Unit test for {@link ImmutableSet}.
     32  *
     33  * @author Kevin Bourrillion
     34  * @author Jared Levy
     35  * @author Nick Kralevich
     36  */
     37 @GwtCompatible(emulated = true)
     38 public class ImmutableSetTest extends AbstractImmutableSetTest {
     39 
     40   @Override protected Set<String> of() {
     41     return ImmutableSet.of();
     42   }
     43 
     44   @Override protected Set<String> of(String e) {
     45     return ImmutableSet.of(e);
     46   }
     47 
     48   @Override protected Set<String> of(String e1, String e2) {
     49     return ImmutableSet.of(e1, e2);
     50   }
     51 
     52   @Override protected Set<String> of(String e1, String e2, String e3) {
     53     return ImmutableSet.of(e1, e2, e3);
     54   }
     55 
     56   @Override protected Set<String> of(
     57       String e1, String e2, String e3, String e4) {
     58     return ImmutableSet.of(e1, e2, e3, e4);
     59   }
     60 
     61   @Override protected Set<String> of(
     62       String e1, String e2, String e3, String e4, String e5) {
     63     return ImmutableSet.of(e1, e2, e3, e4, e5);
     64   }
     65 
     66   @Override protected Set<String> of(String e1, String e2, String e3,
     67       String e4, String e5, String e6, String... rest) {
     68     return ImmutableSet.of(e1, e2, e3, e4, e5, e6, rest);
     69   }
     70 
     71   @Override protected Set<String> copyOf(String[] elements) {
     72     return ImmutableSet.copyOf(elements);
     73   }
     74 
     75   @Override protected Set<String> copyOf(Collection<String> elements) {
     76     return ImmutableSet.copyOf(elements);
     77   }
     78 
     79   @Override protected Set<String> copyOf(Iterable<String> elements) {
     80     return ImmutableSet.copyOf(elements);
     81   }
     82 
     83   @Override protected Set<String> copyOf(Iterator<String> elements) {
     84     return ImmutableSet.copyOf(elements);
     85   }
     86 
     87   public void testCreation_allDuplicates() {
     88     ImmutableSet<String> set = ImmutableSet.copyOf(Lists.newArrayList("a", "a"));
     89     assertTrue(set instanceof SingletonImmutableSet);
     90     assertEquals(Lists.newArrayList("a"), Lists.newArrayList(set));
     91   }
     92 
     93   public void testCreation_oneDuplicate() {
     94     // now we'll get the varargs overload
     95     ImmutableSet<String> set = ImmutableSet.of(
     96         "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "a");
     97     assertEquals(Lists.newArrayList(
     98         "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m"),
     99         Lists.newArrayList(set));
    100   }
    101 
    102   public void testCreation_manyDuplicates() {
    103     // now we'll get the varargs overload
    104     ImmutableSet<String> set = ImmutableSet.of(
    105         "a", "b", "c", "c", "c", "c", "b", "b", "a", "a", "c", "c", "c", "a");
    106     assertThat(set).has().exactly("a", "b", "c").inOrder();
    107   }
    108 
    109   public void testCreation_arrayOfArray() {
    110     String[] array = new String[] { "a" };
    111     Set<String[]> set = ImmutableSet.<String[]>of(array);
    112     assertEquals(Collections.singleton(array), set);
    113   }
    114 
    115   public void testCopyOf_copiesImmutableSortedSet() {
    116     ImmutableSortedSet<String> sortedSet = ImmutableSortedSet.of("a");
    117     ImmutableSet<String> copy = ImmutableSet.copyOf(sortedSet);
    118     assertNotSame(sortedSet, copy);
    119   }
    120 
    121   @Override <E extends Comparable<E>> Builder<E> builder() {
    122     return ImmutableSet.builder();
    123   }
    124 
    125   @Override int getComplexBuilderSetLastElement() {
    126     return LAST_COLOR_ADDED;
    127   }
    128 
    129   public void testEquals() {
    130     new EqualsTester()
    131         .addEqualityGroup(ImmutableSet.of(), ImmutableSet.of())
    132         .addEqualityGroup(ImmutableSet.of(1), ImmutableSet.of(1), ImmutableSet.of(1, 1))
    133         .addEqualityGroup(ImmutableSet.of(1, 2, 1), ImmutableSet.of(2, 1, 1))
    134         .testEquals();
    135   }
    136 }
    137 
    138