| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.commons.collections4.map; |
| |
| import java.util.Comparator; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.Map; |
| import java.util.SortedMap; |
| import java.util.TreeMap; |
| import java.util.concurrent.ConcurrentSkipListMap; |
| |
| import org.apache.commons.collections4.Predicate; |
| import org.apache.commons.collections4.functors.TruePredicate; |
| |
| /** |
| * Extension of {@link PredicatedMapTest} for exercising the |
| * {@link PredicatedSortedMap} implementation. |
| * |
| * @since 3.0 |
| */ |
| public class PredicatedSortedMapTest<K, V> extends AbstractSortedMapTest<K, V> { |
| |
| private class ReverseStringComparator implements Comparator<K> { |
| @Override |
| public int compare(K arg0, K arg1) { |
| return ((String) arg1).compareTo((String)arg0); |
| } |
| } |
| |
| protected static final Predicate<Object> truePredicate = TruePredicate.truePredicate(); |
| |
| protected static final Predicate<Object> testPredicate = o -> o instanceof String; |
| |
| protected final Comparator<K> reverseStringComparator = new ReverseStringComparator(); |
| |
| public PredicatedSortedMapTest(final String testName) { |
| super(testName); |
| } |
| |
| //----------------------------------------------------------------------- |
| protected SortedMap<K, V> decorateMap(final SortedMap<K, V> map, final Predicate<? super K> keyPredicate, |
| final Predicate<? super V> valuePredicate) { |
| return PredicatedSortedMap.predicatedSortedMap(map, keyPredicate, valuePredicate); |
| } |
| |
| @Override |
| public SortedMap<K, V> makeObject() { |
| return decorateMap(new TreeMap<K, V>(), truePredicate, truePredicate); |
| } |
| |
| public SortedMap<K, V> makeTestMap() { |
| return decorateMap(new TreeMap<K, V>(), testPredicate, testPredicate); |
| } |
| |
| public SortedMap<K, V> makeTestMapWithComparator() { |
| return decorateMap(new ConcurrentSkipListMap<K, V>(reverseStringComparator), testPredicate, testPredicate); |
| } |
| |
| @Override |
| public boolean isSubMapViewsSerializable() { |
| // TreeMap sub map views have a bug in deserialization. |
| return false; |
| } |
| |
| @Override |
| public boolean isAllowNullKey() { |
| return false; |
| } |
| |
| // from TestPredicatedMap |
| //----------------------------------------------------------------------- |
| @SuppressWarnings("unchecked") |
| public void testEntrySet() { |
| SortedMap<K, V> map = makeTestMap(); |
| assertTrue("returned entryset should not be null", |
| map.entrySet() != null); |
| map = decorateMap(new TreeMap<K, V>(), null, null); |
| map.put((K) "oneKey", (V) "oneValue"); |
| assertTrue("returned entryset should contain one entry", |
| map.entrySet().size() == 1); |
| map = decorateMap(map, null, null); |
| } |
| |
| @SuppressWarnings("unchecked") |
| public void testPut() { |
| final Map<K, V> map = makeTestMap(); |
| try { |
| map.put((K) "Hi", (V) Integer.valueOf(3)); |
| fail("Illegal value should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| |
| try { |
| map.put((K) Integer.valueOf(3), (V) "Hi"); |
| fail("Illegal key should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertTrue(!map.containsKey(Integer.valueOf(3))); |
| assertTrue(!map.containsValue(Integer.valueOf(3))); |
| |
| final Map<K, V> map2 = new HashMap<>(); |
| map2.put((K) "A", (V) "a"); |
| map2.put((K) "B", (V) "b"); |
| map2.put((K) "C", (V) "c"); |
| map2.put((K) "c", (V) Integer.valueOf(3)); |
| |
| try { |
| map.putAll(map2); |
| fail("Illegal value should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| |
| map.put((K) "E", (V) "e"); |
| Iterator<Map.Entry<K, V>> iterator = map.entrySet().iterator(); |
| try { |
| final Map.Entry<K, V> entry = iterator.next(); |
| entry.setValue((V) Integer.valueOf(3)); |
| fail("Illegal value should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| |
| map.put((K) "F", (V) "f"); |
| iterator = map.entrySet().iterator(); |
| final Map.Entry<K, V> entry = iterator.next(); |
| entry.setValue((V) "x"); |
| |
| } |
| |
| //----------------------------------------------------------------------- |
| @SuppressWarnings("unchecked") |
| public void testSortOrder() { |
| final SortedMap<K, V> map = makeTestMap(); |
| map.put((K) "A", (V) "a"); |
| map.put((K) "B", (V) "b"); |
| try { |
| map.put(null, (V) "c"); |
| fail("Null key should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| map.put((K) "C", (V) "c"); |
| try { |
| map.put((K) "D", null); |
| fail("Null value should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| assertEquals("First key should be A", "A", map.firstKey()); |
| assertEquals("Last key should be C", "C", map.lastKey()); |
| assertEquals("First key in tail map should be B", |
| "B", map.tailMap((K) "B").firstKey()); |
| assertEquals("Last key in head map should be B", |
| "B", map.headMap((K) "C").lastKey()); |
| assertEquals("Last key in submap should be B", |
| "B", map.subMap((K) "A",(K) "C").lastKey()); |
| |
| final Comparator<? super K> c = map.comparator(); |
| assertTrue("natural order, so comparator should be null", |
| c == null); |
| } |
| |
| @SuppressWarnings("unchecked") |
| public void testReverseSortOrder() { |
| final SortedMap<K, V> map = makeTestMapWithComparator(); |
| map.put((K) "A", (V) "a"); |
| map.put((K) "B", (V) "b"); |
| try { |
| map.put(null, (V) "c"); |
| fail("Null key should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| map.put((K) "C", (V) "c"); |
| try { |
| map.put((K) "D", null); |
| fail("Null value should raise IllegalArgument"); |
| } catch (final IllegalArgumentException e) { |
| // expected |
| } |
| assertEquals("Last key should be A", "A", map.lastKey()); |
| assertEquals("First key should be C", "C", map.firstKey()); |
| assertEquals("First key in tail map should be B", |
| "B", map.tailMap((K) "B").firstKey()); |
| assertEquals("Last key in head map should be B", |
| "B", map.headMap((K) "A").lastKey()); |
| assertEquals("Last key in submap should be B", |
| "B", map.subMap((K) "C",(K) "A").lastKey()); |
| |
| final Comparator<? super K> c = map.comparator(); |
| assertTrue("reverse order, so comparator should be reverseStringComparator", |
| c == reverseStringComparator); |
| } |
| |
| @Override |
| public String getCompatibilityVersion() { |
| return "4"; |
| } |
| |
| // public void testCreate() throws Exception { |
| // resetEmpty(); |
| // writeExternalFormToDisk( |
| // (java.io.Serializable) map, |
| // "src/test/resources/data/test/PredicatedSortedMap.emptyCollection.version4.obj"); |
| // resetFull(); |
| // writeExternalFormToDisk( |
| // (java.io.Serializable) map, |
| // "src/test/resources/data/test/PredicatedSortedMap.fullCollection.version4.obj"); |
| // } |
| |
| } |