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 com.google.common.annotations.GwtCompatible;
20import com.google.common.collect.BiMap;
21import com.google.common.collect.ImmutableBiMap;
22import com.google.common.collect.Maps;
23
24import java.util.Map;
25import java.util.Map.Entry;
26
27/**
28 * Generators of various {@link com.google.common.collect.BiMap}s and derived
29 * collections.
30 *
31 * @author Jared Levy
32 * @author Hayward Chan
33 */
34@GwtCompatible
35public class BiMapGenerators {
36  public static class ImmutableBiMapGenerator extends TestStringBiMapGenerator {
37    @Override protected BiMap<String, String> create(Entry<String, String>[] entries) {
38      ImmutableBiMap.Builder<String, String> builder = ImmutableBiMap.builder();
39      for (Entry<String, String> entry : entries) {
40        builder.put(entry.getKey(), entry.getValue());
41      }
42      return builder.build();
43    }
44  }
45
46  public static class ImmutableBiMapCopyOfGenerator extends TestStringBiMapGenerator {
47    @Override protected BiMap<String, String> create(Entry<String, String>[] entries) {
48      Map<String, String> builder = Maps.newLinkedHashMap();
49      for (Entry<String, String> entry : entries) {
50        builder.put(entry.getKey(), entry.getValue());
51      }
52      return ImmutableBiMap.copyOf(builder);
53    }
54  }
55}
56