diff options
author | Googler <noreply@google.com> | 2017-06-19 19:46:47 -0400 |
---|---|---|
committer | Kristina Chodorow <kchodorow@google.com> | 2017-06-20 14:35:54 -0400 |
commit | 231e24e77f56a28d7032eee3e03bda78c8cc4589 (patch) | |
tree | f4d8c29cfe08ca544221187fdb1d81dd4263d5a9 /src/test/java/com/google/devtools/build/lib/collect | |
parent | 9c931b3dfe204e5c25d016876c6ccb3ea55e7998 (diff) |
Extract ImmutableSharedKeysMap class from TransitiveInfoProviderMap.
PiperOrigin-RevId: 159498323
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/collect')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/collect/ImmutableSharedKeyMapTest.java | 130 |
1 files changed, 130 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/collect/ImmutableSharedKeyMapTest.java b/src/test/java/com/google/devtools/build/lib/collect/ImmutableSharedKeyMapTest.java new file mode 100644 index 0000000000..8716c6bdcb --- /dev/null +++ b/src/test/java/com/google/devtools/build/lib/collect/ImmutableSharedKeyMapTest.java @@ -0,0 +1,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); + } +} |