/*
    * 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 static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertFalse;
  import static org.junit.jupiter.api.Assertions.assertThrows;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  import java.util.Map;
  import java.util.Set;
  import java.util.SortedMap;
  import java.util.TreeMap;
  
  import org.apache.commons.collections4.Transformer;
  import org.apache.commons.collections4.TransformerUtils;
  import org.apache.commons.collections4.collection.TransformedCollectionTest;
  import org.junit.jupiter.api.Test;
  
  /**
   * Extension of {@link AbstractSortedMapTest} for exercising the {@link TransformedSortedMap}
   * implementation.
   *
   * @param <K> the key type.
   * @param <V> the value type.
   */
  public class TransformedSortedMapTest<K, V> extends AbstractSortedMapTest<K, V> {
  
      @Override
      public String getCompatibilityVersion() {
          return "4";
      }
  
      @Override
      public boolean isSubMapViewsSerializable() {
          // TreeMap sub map views have a bug in deserialization.
          return false;
      }
  
      @Override
      @SuppressWarnings("unchecked")
      public SortedMap<K, V> makeObject() {
          return TransformedSortedMap.transformingSortedMap(new TreeMap<>(),
                  (Transformer<? super K, ? extends K>) TransformerUtils.nopTransformer(),
                  (Transformer<? super V, ? extends V>) TransformerUtils.nopTransformer());
      }
  
      @Test
      @SuppressWarnings("unchecked")
      public void testFactory_Decorate() {
          final SortedMap<K, V> base = new TreeMap<>();
          base.put((K) "A", (V) "1");
          base.put((K) "B", (V) "2");
          base.put((K) "C", (V) "3");
  
          final SortedMap<K, V> trans = TransformedSortedMap
                  .transformingSortedMap(
                          base,
                          null,
                          (Transformer<? super V, ? extends V>) TransformedCollectionTest.STRING_TO_INTEGER_TRANSFORMER);
          assertEquals(3, trans.size());
          assertEquals("1", trans.get("A"));
          assertEquals("2", trans.get("B"));
          assertEquals("3", trans.get("C"));
          trans.put((K) "D", (V) "4");
          assertEquals(Integer.valueOf(4), trans.get("D"));
      }
  
      @Test
      @SuppressWarnings("unchecked")
      public void testFactory_decorateTransform() {
          final SortedMap<K, V> base = new TreeMap<>();
          base.put((K) "A", (V) "1");
          base.put((K) "B", (V) "2");
          base.put((K) "C", (V) "3");
  
          final SortedMap<K, V> trans = TransformedSortedMap
                  .transformedSortedMap(
                          base,
                          null,
                          (Transformer<? super V, ? extends V>) TransformedCollectionTest.STRING_TO_INTEGER_TRANSFORMER);
          assertEquals(3, trans.size());
          assertEquals(Integer.valueOf(1), trans.get("A"));
          assertEquals(Integer.valueOf(2), trans.get("B"));
          assertEquals(Integer.valueOf(3), trans.get("C"));
         trans.put((K) "D", (V) "4");
         assertEquals(Integer.valueOf(4), trans.get("D"));
     }
 
     @Test
     @SuppressWarnings("unchecked")
     public void testTransformedMap() {
         final Object[] els = { "1", "3", "5", "7", "2", "4", "6" };
 
         SortedMap<K, V> map = TransformedSortedMap
                 .transformingSortedMap(
                         new TreeMap<>(),
                         (Transformer<? super K, ? extends K>) TransformedCollectionTest.STRING_TO_INTEGER_TRANSFORMER,
                         null);
         assertEquals(0, map.size());
         for (int i = 0; i < els.length; i++) {
             map.put((K) els[i], (V) els[i]);
             assertEquals(i + 1, map.size());
             assertTrue(map.containsKey(Integer.valueOf((String) els[i])));
             final SortedMap<K, V> finalMap1 = map;
             final int finalI = i;
             assertThrows(ClassCastException.class, () -> finalMap1.containsKey(els[finalI]));
             assertTrue(map.containsValue(els[i]));
             assertEquals(els[i], map.get(Integer.valueOf((String) els[i])));
         }
 
         final SortedMap<K, V> finalMap = map;
         assertThrows(ClassCastException.class, () -> finalMap.remove(els[0]));
         assertEquals(els[0], map.remove(Integer.valueOf((String) els[0])));
 
         map = TransformedSortedMap
                 .transformingSortedMap(
                         new TreeMap<>(),
                         null,
                         (Transformer<? super V, ? extends V>) TransformedCollectionTest.STRING_TO_INTEGER_TRANSFORMER);
         assertEquals(0, map.size());
         for (int i = 0; i < els.length; i++) {
             map.put((K) els[i], (V) els[i]);
             assertEquals(i + 1, map.size());
             assertTrue(map.containsValue(Integer.valueOf((String) els[i])));
             assertFalse(map.containsValue(els[i]));
             assertTrue(map.containsKey(els[i]));
             assertEquals(Integer.valueOf((String) els[i]), map.get(els[i]));
         }
 
         assertEquals(Integer.valueOf((String) els[0]), map.remove(els[0]));
 
         final Set<Map.Entry<K, V>> entrySet = map.entrySet();
         final Map.Entry<K, V>[] array = entrySet.toArray(new Map.Entry[0]);
         array[0].setValue((V) "66");
         assertEquals(Integer.valueOf(66), array[0].getValue());
         assertEquals(Integer.valueOf(66), map.get(array[0].getKey()));
 
         final Map.Entry<K, V> entry = entrySet.iterator().next();
         entry.setValue((V) "88");
         assertEquals(Integer.valueOf(88), entry.getValue());
         assertEquals(Integer.valueOf(88), map.get(entry.getKey()));
     }
 
 //    public void testCreate() throws Exception {
 //        resetEmpty();
 //        writeExternalFormToDisk(
 //            (java.io.Serializable) map,
 //            "src/test/resources/data/test/TransformedSortedMap.emptyCollection.version4.obj");
 //        resetFull();
 //        writeExternalFormToDisk(
 //            (java.io.Serializable) map,
 //            "src/test/resources/data/test/TransformedSortedMap.fullCollection.version4.obj");
 //    }
 }