1 /*

     2  * Copyright (c) 2017, 2017, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 package org.graalvm.collections.test;

    26 

    27 import java.util.LinkedHashMap;

    28 

    29 import org.graalvm.collections.EconomicMap;

    30 import org.graalvm.collections.UnmodifiableEconomicMap;

    31 import org.junit.Assert;

    32 import org.junit.Test;

    33 

    34 public class EconomicMapTest {

    35 

    36     @Test

    37     public void testMapGetDefault() {

    38         EconomicMap<Integer, Integer> map = EconomicMap.create();

    39         map.put(0, 1);

    40         Assert.assertEquals(map.get(0, 2), Integer.valueOf(1));

    41         Assert.assertEquals(map.get(1, 2), Integer.valueOf(2));

    42     }

    43 

    44     @Test

    45     public void testMapPutAll() {

    46         EconomicMap<Integer, Integer> map = EconomicMap.create();

    47         EconomicMap<Integer, Integer> newMap = EconomicMap.wrapMap(new LinkedHashMap<>());

    48         newMap.put(1, 1);

    49         newMap.put(2, 4);

    50         map.putAll(newMap);

    51         Assert.assertEquals(map.size(), 2);

    52 

    53         UnmodifiableEconomicMap<Integer, Integer> unmodifiableEconomicMap = EconomicMap.create(newMap);

    54 

    55         map.removeKey(1);

    56         map.put(2, 2);

    57         map.put(3, 9);

    58 

    59         map.putAll(unmodifiableEconomicMap);

    60         Assert.assertEquals(map.size(), 3);

    61         Assert.assertEquals(map.get(2), Integer.valueOf(4));

    62     }

    63 

    64     @Test

    65     public void testToString() {

    66         EconomicMap<Integer, Integer> map = EconomicMap.create();

    67         map.put(0, 0);

    68         map.put(1, 1);

    69         Assert.assertEquals(map.toString(), "map(size=2, {(0,0),(1,1)})");

    70     }

    71 

    72 }