View Javadoc
1   /*
2    * Copyright (C) 2013 The Guava Authors
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
5    * in compliance with the License. You may obtain a copy of the License at
6    *
7    * http://www.apache.org/licenses/LICENSE-2.0
8    *
9    * Unless required by applicable law or agreed to in writing, software distributed under the License
10   * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
11   * or implied. See the License for the specific language governing permissions and limitations under
12   * the License.
13   */
14  
15  package com.google.common.collect.testing.google;
16  
17  import static com.google.common.collect.testing.features.CollectionFeature.SUPPORTS_ITERATOR_REMOVE;
18  import static com.google.common.collect.testing.features.CollectionSize.ZERO;
19  import static com.google.common.collect.testing.features.MapFeature.ALLOWS_NULL_KEYS;
20  import static com.google.common.collect.testing.features.MapFeature.ALLOWS_NULL_KEY_QUERIES;
21  import static com.google.common.collect.testing.features.MapFeature.SUPPORTS_REMOVE;
22  
23  import com.google.common.annotations.GwtCompatible;
24  import com.google.common.collect.Multimap;
25  import com.google.common.collect.testing.features.CollectionFeature;
26  import com.google.common.collect.testing.features.CollectionSize;
27  import com.google.common.collect.testing.features.MapFeature;
28  import java.util.Iterator;
29  import java.util.Map;
30  
31  /**
32   * Tester for {@code Multimap.keySet}.
33   *
34   * @author Louis Wasserman
35   */
36  @GwtCompatible
37  public class MultimapKeySetTester<K, V> extends AbstractMultimapTester<K, V, Multimap<K, V>> {
38    public void testKeySet() {
39      for (Map.Entry<K, V> entry : getSampleElements()) {
40        assertTrue(multimap().keySet().contains(entry.getKey()));
41      }
42    }
43  
44    @CollectionSize.Require(absent = ZERO)
45    @MapFeature.Require(ALLOWS_NULL_KEYS)
46    public void testKeySetContainsNullKeyPresent() {
47      initMultimapWithNullKey();
48      assertTrue(multimap().keySet().contains(null));
49    }
50  
51    @MapFeature.Require(ALLOWS_NULL_KEY_QUERIES)
52    public void testKeySetContainsNullKeyAbsent() {
53      assertFalse(multimap().keySet().contains(null));
54    }
55  
56    @MapFeature.Require(SUPPORTS_REMOVE)
57    public void testKeySetRemovePropagatesToMultimap() {
58      int key0Count = multimap().get(k0()).size();
59      assertEquals(key0Count > 0, multimap().keySet().remove(k0()));
60      assertEquals(getNumElements() - key0Count, multimap().size());
61      assertGet(k0());
62    }
63  
64    @CollectionSize.Require(absent = ZERO)
65    @CollectionFeature.Require(SUPPORTS_ITERATOR_REMOVE)
66    public void testKeySetIteratorRemove() {
67      int key0Count = multimap().get(k0()).size();
68      Iterator<K> keyItr = multimap().keySet().iterator();
69      while (keyItr.hasNext()) {
70        if (keyItr.next().equals(k0())) {
71          keyItr.remove();
72        }
73      }
74      assertEquals(getNumElements() - key0Count, multimap().size());
75      assertGet(k0());
76    }
77  }