TroveTest.java

/*
 * Copyright 2000-2015 JetBrains s.r.o.
 *
 * Licensed 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 com.intellij.util.containers;


import gnu.trove.THashMap;
import gnu.trove.TIntObjectHashMap;
import gnu.trove.TObjectIntHashMap;
import junit.framework.TestCase;

import java.util.Map;
import java.util.Set;

public class TroveTest extends TestCase {
  public void testObjectInt() {
    TObjectIntHashMap<String> map = new TObjectIntHashMap<>();
    map.trimToSize();
    for (int i = 0; i < 100; i++) {
      String key = String.valueOf(i);
      map.put(key, i);
      map.put(key+"a", i);
    }
    for (int i = 0; i < 100; i++) {
      String key = String.valueOf(i);
      assertEquals(i, map.get(key));
      assertEquals(i, map.get(key + "a"));
    }
    assertEquals(200, map.size());
  }

  public void testTHashMap_Entry() {
    Map<Object, Object> map = new THashMap<>();
    map.put("1", "2");

    Map.Entry<Object, Object> entry = map.entrySet().iterator().next();

    assertEquals("1", entry.getKey());
    assertEquals("2", entry.getValue());
    Object old = entry.setValue("3");
    assertEquals("2", old);
    assertEquals("1", entry.getKey());
    assertEquals("3", entry.getValue());

    Map.Entry<Object, Object> refreshed = map.entrySet().iterator().next();
    assertEquals("1", refreshed.getKey());
    assertEquals("3", refreshed.getValue());
  }

  public void testKObjectMapCloneDoesNotDependOnTheSource() {
    TIntObjectHashMap<int[]> map = new TIntObjectHashMap<>();
    map.put(0, new int[2]);
    map.put(1, new int[2]);

    TIntObjectHashMap<int[]> clone = map.clone();
    assertEquals(clone.size(), 2);
    int[] keys = clone.keys();
    assertEquals(keys.length, 2);
    assertEquals(ContainerUtil.newHashSet(0,1), ContainerUtil.newHashSet(keys[0],keys[1]));

    map.clear();

    assertEquals(clone.size(), 2);
    keys = clone.keys();
    assertEquals(keys.length, 2);
    assertEquals(ContainerUtil.newHashSet(0,1), ContainerUtil.newHashSet(keys[0],keys[1]));
  }

  public void testHashMapCloneDoesNotDependOnTheSource() {
    THashMap<Integer, int[]> map = new THashMap<>();
    map.put(0, new int[2]);
    map.put(1, new int[2]);

    THashMap<Integer, int[]> clone = map.clone();
    assertEquals(clone.size(), 2);
    Set<Integer> keys = clone.keySet();
    assertEquals(keys.size(), 2);
    assertEquals(ContainerUtil.newHashSet(0,1), keys);

    map.clear();

    assertEquals(clone.size(), 2);
    keys = clone.keySet();
    assertEquals(keys.size(), 2);
    assertEquals(ContainerUtil.newHashSet(0,1), keys);
  }
}