/*
    * 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;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertFalse;
  import static org.junit.jupiter.api.Assertions.assertSame;
  import static org.junit.jupiter.api.Assertions.assertThrows;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  import java.util.HashMap;
  import java.util.Map;
  
  import org.apache.commons.collections4.functors.NOPTransformer;
  import org.apache.commons.collections4.map.HashedMap;
  import org.apache.commons.collections4.splitmap.TransformedSplitMap;
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.Test;
  
  /**
   * Tests for {@link TransformedSplitMap}
   */
  @SuppressWarnings("boxing")
  public class SplitMapUtilsTest {
  
      private Map<String, Integer> backingMap;
      private TransformedSplitMap<String, String, String, Integer> transformedMap;
  
      private final Transformer<String, Integer> stringToInt = Integer::valueOf;
  
      private void attemptGetOperation(final Runnable r) {
          assertThrows(UnsupportedOperationException.class, () -> r.run(),
                  "Put exposed as writable Map must not allow Get operations");
      }
  
      private void attemptPutOperation(final Runnable r) {
          assertThrows(UnsupportedOperationException.class, () -> r.run(),
                  "Get exposed as writable Map must not allow Put operations");
      }
  
      @BeforeEach
      public void setUp() throws Exception {
          backingMap = new HashMap<>();
          transformedMap = TransformedSplitMap.transformingMap(backingMap, NOPTransformer.<String>nopTransformer(),
                  stringToInt);
          for (int i = 0; i < 10; i++) {
              transformedMap.put(String.valueOf(i), String.valueOf(i));
          }
      }
  
      @Test
      public void testAlreadyReadableMap() {
          final HashedMap<String, Integer> hashedMap = new HashedMap<>();
          assertSame(hashedMap, SplitMapUtils.readableMap(hashedMap));
      }
  
      @Test
      public void testAlreadyWritableMap() {
          final HashedMap<String, String> hashedMap = new HashedMap<>();
          assertSame(hashedMap, SplitMapUtils.writableMap(hashedMap));
      }
  
      @Test
      public void testReadableMap() {
          final IterableMap<String, Integer> map = SplitMapUtils.readableMap(transformedMap);
  
          // basic
          for (int i = 0; i < 10; i++) {
              assertFalse(map.containsValue(String.valueOf(i)));
              assertEquals(i, map.get(String.valueOf(i)).intValue());
          }
  
          // mapIterator
          final MapIterator<String, Integer> it = map.mapIterator();
          while (it.hasNext()) {
              final String k = it.next();
              assertEquals(k, it.getKey());
              assertEquals(Integer.valueOf(k), it.getValue());
          }
  
          // unmodifiable
          assertTrue(map instanceof Unmodifiable);
  
          // check individual operations
          int sz = map.size();
 
         attemptPutOperation(map::clear);
 
         assertEquals(sz, map.size());
 
         attemptPutOperation(() -> map.put("foo", 100));
 
         final HashMap<String, Integer> m = new HashMap<>();
         m.put("foo", 100);
         m.put("bar", 200);
         m.put("baz", 300);
         attemptPutOperation(() -> map.putAll(m));
 
         // equals, hashCode
         final IterableMap<String, Integer> other = SplitMapUtils.readableMap(transformedMap);
         assertEquals(other, map);
         assertEquals(other.hashCode(), map.hashCode());
 
         // remove
         for (int i = 0; i < 10; i++) {
             assertEquals(i, map.remove(String.valueOf(i)).intValue());
             assertEquals(--sz, map.size());
         }
         assertTrue(map.isEmpty());
         assertSame(map, SplitMapUtils.readableMap(map));
     }
 
     @Test
     @SuppressWarnings("unchecked")
     public void testWritableMap() {
         final Map<String, String> map = SplitMapUtils.writableMap(transformedMap);
         attemptGetOperation(() -> map.get(null));
         attemptGetOperation(map::entrySet);
         attemptGetOperation(map::keySet);
         attemptGetOperation(map::values);
         attemptGetOperation(map::size);
         attemptGetOperation(map::isEmpty);
         attemptGetOperation(() -> map.containsKey(null));
         attemptGetOperation(() -> map.containsValue(null));
         attemptGetOperation(() -> map.remove(null));
 
         // equals, hashCode
         final Map<String, String> other = SplitMapUtils.writableMap(transformedMap);
         assertEquals(other, map);
         assertEquals(other.hashCode(), map.hashCode());
 
         // put
         int sz = backingMap.size();
         assertFalse(backingMap.containsKey("foo"));
         map.put("new", "66");
         assertEquals(++sz, backingMap.size());
 
         // putAll
         final Map<String, String> more = new HashMap<>();
         more.put("foo", "77");
         more.put("bar", "88");
         more.put("baz", "99");
         map.putAll(more);
         assertEquals(sz + more.size(), backingMap.size());
 
         // clear
         map.clear();
         assertTrue(backingMap.isEmpty());
         assertSame(map, SplitMapUtils.writableMap((Put<String, String>) map));
     }
 
 }