Home | History | Annotate | Download | only in collect 
      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 
     17 package com.google.common.collect;
     18 
     19 import static com.google.common.base.Preconditions.checkArgument;
     20 
     21 import com.google.common.annotations.GwtCompatible;
     22 import com.google.common.annotations.GwtIncompatible;
     23 import com.google.common.collect.MapConstraintsTest.TestKeyException;
     24 import com.google.common.collect.MapConstraintsTest.TestValueException;
     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 com.google.common.collect.testing.google.BiMapTestSuiteBuilder;
     29 import com.google.common.collect.testing.google.TestStringBiMapGenerator;
     30 
     31 import junit.framework.Test;
     32 import junit.framework.TestCase;
     33 import junit.framework.TestSuite;
     34 
     35 import java.util.Map.Entry;
     36 
     37 /**
     38  * Tests for {@link MapConstraints#constrainedBiMap}.
     39  *
     40  * @author Jared Levy
     41  * @author Louis Wasserman
     42  */
     43 @GwtCompatible(emulated = true)
     44 public class ConstrainedBiMapTest extends TestCase {
     45 
     46   private static final String TEST_KEY = "42";
     47   private static final String TEST_VALUE = "test";
     48   private static final MapConstraint<String, String> TEST_CONSTRAINT = new TestConstraint();
     49 
     50   @GwtIncompatible("suite")
     51   public static Test suite() {
     52     TestSuite suite = new TestSuite();
     53     suite.addTest(BiMapTestSuiteBuilder
     54         .using(new ConstrainedBiMapGenerator())
     55         .named("Maps.constrainedBiMap[HashBiMap]")
     56         .withFeatures(
     57             CollectionSize.ANY,
     58             CollectionFeature.SUPPORTS_ITERATOR_REMOVE,
     59             MapFeature.ALLOWS_NULL_KEYS,
     60             MapFeature.ALLOWS_NULL_VALUES,
     61             MapFeature.ALLOWS_ANY_NULL_QUERIES,
     62             MapFeature.GENERAL_PURPOSE,
     63             MapFeature.REJECTS_DUPLICATES_AT_CREATION)
     64         .createTestSuite());
     65     suite.addTestSuite(ConstrainedBiMapTest.class);
     66     return suite;
     67   }
     68 
     69   public void testPutWithForbiddenKeyForbiddenValue() {
     70     BiMap<String, String> map = MapConstraints.constrainedBiMap(
     71         HashBiMap.<String, String> create(),
     72         TEST_CONSTRAINT);
     73     try {
     74       map.put(TEST_KEY, TEST_VALUE);
     75       fail("Expected IllegalArgumentException");
     76     } catch (IllegalArgumentException expected) {
     77       // success
     78     }
     79   }
     80 
     81   public void testPutWithForbiddenKeyAllowedValue() {
     82     BiMap<String, String> map = MapConstraints.constrainedBiMap(
     83         HashBiMap.<String, String> create(),
     84         TEST_CONSTRAINT);
     85     try {
     86       map.put(TEST_KEY, "allowed");
     87       fail("Expected IllegalArgumentException");
     88     } catch (IllegalArgumentException expected) {
     89       // success
     90     }
     91   }
     92 
     93   public void testPutWithAllowedKeyForbiddenValue() {
     94     BiMap<String, String> map = MapConstraints.constrainedBiMap(
     95         HashBiMap.<String, String> create(),
     96         TEST_CONSTRAINT);
     97     try {
     98       map.put("allowed", TEST_VALUE);
     99       fail("Expected IllegalArgumentException");
    100     } catch (IllegalArgumentException expected) {
    101       // success
    102     }
    103   }
    104 
    105   public static final class ConstrainedBiMapGenerator extends TestStringBiMapGenerator {
    106     @Override
    107     protected BiMap<String, String> create(Entry<String, String>[] entries) {
    108       BiMap<String, String> bimap = MapConstraints.constrainedBiMap(
    109           HashBiMap.<String, String> create(),
    110           TEST_CONSTRAINT);
    111       for (Entry<String, String> entry : entries) {
    112         checkArgument(!bimap.containsKey(entry.getKey()));
    113         bimap.put(entry.getKey(), entry.getValue());
    114       }
    115       return bimap;
    116     }
    117   }
    118 
    119   private static final class TestConstraint implements MapConstraint<String, String> {
    120     @Override
    121     public void checkKeyValue(String key, String value) {
    122       if (TEST_KEY.equals(key)) {
    123         throw new TestKeyException();
    124       }
    125       if (TEST_VALUE.equals(value)) {
    126         throw new TestValueException();
    127       }
    128     }
    129 
    130     private static final long serialVersionUID = 0;
    131   }
    132 }
    133