b4b900158065561ebbee7968a77a1071294f0786
[yangtools.git] / third-party / triemap / src / test / java / org / opendaylight / yangtools / triemap / TestConcurrentMapRemove.java
1 /*
2  * (C) Copyright 2016 Pantheon Technologies, s.r.o. and others.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.opendaylight.yangtools.triemap;
17
18 import static org.junit.Assert.assertFalse;
19 import static org.junit.Assert.assertNull;
20 import static org.junit.Assert.assertTrue;
21
22 import java.util.Map;
23 import java.util.concurrent.ConcurrentMap;
24 import org.junit.Test;
25
26 public class TestConcurrentMapRemove {
27     private static final int COUNT = 50 * 1000;
28
29     @Test
30     public void testConcurrentMapRemove() {
31         final ConcurrentMap<Integer, Object> map = TrieMap.create();
32
33         for (int i = 128; i < COUNT; i++) {
34             assertFalse(map.remove(i, i));
35             assertNull(map.put(i, i));
36             assertFalse(map.remove(i, "lol"));
37             assertTrue(map.containsKey(i));
38             assertTrue(map.remove(i, i));
39             assertFalse(map.containsKey(i));
40             assertNull(map.put(i, i));
41         }
42     }
43
44     @Test
45     public void testConflictingHash() {
46         final ZeroHashInt k1 = new ZeroHashInt(1);
47         final ZeroHashInt k2 = new ZeroHashInt(2);
48         final ZeroHashInt k3 = new ZeroHashInt(3);
49         final ZeroHashInt k3dup = new ZeroHashInt(3);
50         final ZeroHashInt v1 = new ZeroHashInt(4);
51         final ZeroHashInt v2 = new ZeroHashInt(5);
52         final ZeroHashInt v3 = new ZeroHashInt(6);
53         final ZeroHashInt v3dup = new ZeroHashInt(6);
54
55         final Map<ZeroHashInt, ZeroHashInt> map = TrieMap.create();
56         // Pre-populate an LNode
57         assertNull(map.putIfAbsent(k1, v1));
58         assertNull(map.putIfAbsent(k2, v2));
59         assertNull(map.putIfAbsent(k3, v3));
60
61         assertFalse(map.remove(k3, v2));
62         assertTrue(map.remove(k3dup, v3dup));
63     }
64 }