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
15package com.google.common.collect.testing.google;
16
17import static com.google.common.collect.testing.features.MapFeature.SUPPORTS_REMOVE;
18
19import com.google.common.annotations.GwtCompatible;
20import com.google.common.collect.SortedSetMultimap;
21import com.google.common.collect.testing.features.MapFeature;
22
23import java.util.ArrayList;
24import java.util.Collection;
25import java.util.List;
26import java.util.SortedSet;
27
28/**
29 * Testers for {@link SortedSetMultimap#asMap}.
30 *
31 * @author Louis Wasserman
32 * @param <K> The key type of the tested multimap.
33 * @param <V> The value type of the tested multimap.
34 */
35@GwtCompatible
36public class SortedSetMultimapAsMapTester<K, V>
37    extends AbstractMultimapTester<K, V, SortedSetMultimap<K, V>> {
38  public void testAsMapValuesImplementSortedSet() {
39    for (Collection<V> valueCollection : multimap().asMap().values()) {
40      SortedSet<V> valueSet = (SortedSet<V>) valueCollection;
41      assertEquals(multimap().valueComparator(), valueSet.comparator());
42    }
43  }
44
45  public void testAsMapGetImplementsSortedSet() {
46    for (K key : multimap().keySet()) {
47      SortedSet<V> valueSet = (SortedSet<V>) multimap().asMap().get(key);
48      assertEquals(multimap().valueComparator(), valueSet.comparator());
49    }
50  }
51
52  @MapFeature.Require(SUPPORTS_REMOVE)
53  public void testAsMapRemoveImplementsSortedSet() {
54    List<K> keys = new ArrayList<K>(multimap().keySet());
55    for (K key : keys) {
56      resetCollection();
57      SortedSet<V> valueSet = (SortedSet<V>) multimap().asMap().remove(key);
58      assertEquals(multimap().valueComparator(), valueSet.comparator());
59    }
60  }
61}
62