1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright 2017 The Bazel Authors. All rights reserved.
//
// 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.google.devtools.build.lib.collect;
import static com.google.common.truth.Truth.assertThat;
import static junit.framework.TestCase.fail;
import com.google.common.collect.ImmutableMap;
import com.google.common.testing.EqualsTester;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/** Tests for {@link ImmutableSharedKeyMap}. */
@RunWith(JUnit4.class)
public final class ImmutableSharedKeyMapTest {
@Test
public void testBasicFunctionality() throws Exception {
Object valueA = new Object();
Object valueB = new Object();
ImmutableSharedKeyMap<String, Object> map =
ImmutableSharedKeyMap.<String, Object>builder().put("a", valueA).put("b", valueB).build();
assertThat(map.get("a")).isSameAs(valueA);
assertThat(map.get("b")).isSameAs(valueB);
assertThat(map.get("c")).isNull();
// Verify that we can find all items both by iteration and indexing
ImmutableMap.Builder<String, Object> iterationCopy = ImmutableMap.builder();
for (String key : map) {
iterationCopy.put(key, map.get(key));
}
assertThat(iterationCopy.build()).isEqualTo(ImmutableMap.of("a", valueA, "b", valueB));
ImmutableMap.Builder<String, Object> arrayIterationCopy = ImmutableMap.builder();
for (int i = 0; i < map.size(); ++i) {
arrayIterationCopy.put(map.keyAt(i), map.valueAt(i));
}
assertThat(arrayIterationCopy.build()).isEqualTo(ImmutableMap.of("a", valueA, "b", valueB));
}
@Test
public void testEquality() throws Exception {
ImmutableSharedKeyMap<String, Object> emptyMap =
ImmutableSharedKeyMap.<String, Object>builder().build();
Object valueA = new Object();
Object valueB = new Object();
ImmutableSharedKeyMap<String, Object> map =
ImmutableSharedKeyMap.<String, Object>builder().put("a", valueA).put("b", valueB).build();
// Two identically ordered maps are equal
ImmutableSharedKeyMap<String, Object> exactCopy =
ImmutableSharedKeyMap.<String, Object>builder().put("a", valueA).put("b", valueB).build();
// The map is order sensitive, so different insertion orders aren't equal
ImmutableSharedKeyMap<String, Object> oppositeOrderMap =
ImmutableSharedKeyMap.<String, Object>builder().put("b", valueB).put("a", valueA).build();
Object valueC = new Object();
ImmutableSharedKeyMap<String, Object> biggerMap =
ImmutableSharedKeyMap.<String, Object>builder()
.put("a", valueA)
.put("b", valueB)
.put("c", valueC)
.build();
new EqualsTester()
.addEqualityGroup(emptyMap)
.addEqualityGroup(map, exactCopy)
.addEqualityGroup(oppositeOrderMap)
.addEqualityGroup(biggerMap)
.testEquals();
}
@Test
public void testMultipleIdenticalKeysThrowsException() throws Exception {
Object valueA = new Object();
Object valueB = new Object();
Object valueC = new Object();
ImmutableSharedKeyMap.Builder<String, Object> map =
ImmutableSharedKeyMap.<String, Object>builder()
.put("key", valueA)
.put("key", valueB)
.put("key", valueC);
try {
map.build();
fail();
} catch (IllegalArgumentException e) {
// Expected
}
}
private static class SameHashCodeClass {
@Override
public int hashCode() {
return 0;
}
}
@Test
public void testTwoKeysWithTheSameHashCode() throws Exception {
SameHashCodeClass keyA = new SameHashCodeClass();
SameHashCodeClass keyB = new SameHashCodeClass();
Object valueA = new Object();
Object valueB = new Object();
ImmutableSharedKeyMap<SameHashCodeClass, Object> map =
ImmutableSharedKeyMap.<SameHashCodeClass, Object>builder()
.put(keyA, valueA)
.put(keyB, valueB)
.build();
assertThat(map.get(keyA)).isSameAs(valueA);
assertThat(map.get(keyB)).isSameAs(valueB);
}
}
|
