1/*
2 * Copyright (C) 2008 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
17package com.google.common.collect.testing.google;
18
19import static com.google.common.base.Preconditions.checkNotNull;
20
21import com.google.common.annotations.GwtCompatible;
22import com.google.common.collect.ImmutableBiMap;
23import com.google.common.collect.Maps;
24import com.google.common.collect.testing.SampleElements;
25import com.google.common.collect.testing.TestMapEntrySetGenerator;
26import com.google.common.collect.testing.TestStringSetGenerator;
27
28import java.util.Map;
29import java.util.Map.Entry;
30import java.util.Set;
31
32/**
33 * Generators of various {@link com.google.common.collect.BiMap}s and derived
34 * collections.
35 *
36 * @author Jared Levy
37 * @author Hayward Chan
38 */
39@GwtCompatible
40public class BiMapGenerators {
41
42  public static class ImmutableBiMapKeySetGenerator
43      extends TestStringSetGenerator {
44    @Override protected Set<String> create(String[] elements) {
45      Map<String, Integer> map = Maps.newLinkedHashMap();
46      for (int i = 0; i < elements.length; i++) {
47        map.put(elements[i], i);
48      }
49      return ImmutableBiMap.copyOf(map).keySet();
50    }
51  }
52
53  public static class ImmutableBiMapValuesGenerator
54      extends TestStringSetGenerator {
55    @Override protected Set<String> create(String[] elements) {
56      Map<Integer, String> map = Maps.newLinkedHashMap();
57      for (int i = 0; i < elements.length; i++) {
58        map.put(i, elements[i]);
59      }
60      return ImmutableBiMap.copyOf(map).values();
61    }
62  }
63
64  public static class ImmutableBiMapInverseEntrySetGenerator
65      extends TestMapEntrySetGenerator<String, String> {
66
67    public ImmutableBiMapInverseEntrySetGenerator() {
68      super(new SampleElements.Strings(), new SampleElements.Strings());
69    }
70    @Override public Set<Entry<String, String>> createFromEntries(
71        Entry<String, String>[] entries) {
72      Map<String, String> map = Maps.newLinkedHashMap();
73      for (Entry<String, String> entry : entries) {
74        checkNotNull(entry);
75        map.put(entry.getValue(), entry.getKey());
76      }
77      return ImmutableBiMap.copyOf(map).inverse().entrySet();
78    }
79  }
80
81  public static class ImmutableBiMapInverseKeySetGenerator
82      extends TestStringSetGenerator {
83    @Override protected Set<String> create(String[] elements) {
84      Map<Integer, String> map = Maps.newLinkedHashMap();
85      for (int i = 0; i < elements.length; i++) {
86        map.put(i, elements[i]);
87      }
88      return ImmutableBiMap.copyOf(map).inverse().keySet();
89    }
90  }
91
92  public static class ImmutableBiMapInverseValuesGenerator
93      extends TestStringSetGenerator {
94    @Override protected Set<String> create(String[] elements) {
95      Map<String, Integer> map = Maps.newLinkedHashMap();
96      for (int i = 0; i < elements.length; i++) {
97        map.put(elements[i], i);
98      }
99      return ImmutableBiMap.copyOf(map).inverse().values();
100    }
101  }
102
103  public static class ImmutableBiMapEntrySetGenerator
104      extends TestMapEntrySetGenerator<String, String> {
105
106    public ImmutableBiMapEntrySetGenerator() {
107      super(new SampleElements.Strings(), new SampleElements.Strings());
108    }
109    @Override public Set<Entry<String, String>> createFromEntries(
110        Entry<String, String>[] entries) {
111      Map<String, String> map = Maps.newLinkedHashMap();
112      for (Entry<String, String> entry : entries) {
113        checkNotNull(entry);
114        map.put(entry.getKey(), entry.getValue());
115      }
116      return ImmutableBiMap.copyOf(map).entrySet();
117    }
118  }
119}
120