| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. All rights reserved. |
| // https://developers.google.com/protocol-buffers/ |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are |
| // met: |
| // |
| // * Redistributions of source code must retain the above copyright |
| // notice, this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above |
| // copyright notice, this list of conditions and the following disclaimer |
| // in the documentation and/or other materials provided with the |
| // distribution. |
| // * Neither the name of Google Inc. nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| package com.google.protobuf; |
| |
| import map_lite_test.MapForProto2TestProto.BizarroTestMap; |
| import map_lite_test.MapForProto2TestProto.TestMap; |
| import map_lite_test.MapForProto2TestProto.TestMap.MessageValue; |
| import map_lite_test.MapForProto2TestProto.TestMapOrBuilder; |
| import map_lite_test.MapForProto2TestProto.TestUnknownEnumValue; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.HashMap; |
| import java.util.Map; |
| import junit.framework.TestCase; |
| |
| /** |
| * Unit tests for map fields. |
| */ |
| public final class MapForProto2LiteTest extends TestCase { |
| |
| private void setMapValues(TestMap.Builder builder) { |
| builder |
| .putInt32ToInt32Field(1, 11) |
| .putInt32ToInt32Field(2, 22) |
| .putInt32ToInt32Field(3, 33) |
| |
| .putInt32ToStringField(1, "11") |
| .putInt32ToStringField(2, "22") |
| .putInt32ToStringField(3, "33") |
| |
| .putInt32ToBytesField(1, TestUtil.toBytes("11")) |
| .putInt32ToBytesField(2, TestUtil.toBytes("22")) |
| .putInt32ToBytesField(3, TestUtil.toBytes("33")) |
| |
| .putInt32ToEnumField(1, TestMap.EnumValue.FOO) |
| .putInt32ToEnumField(2, TestMap.EnumValue.BAR) |
| .putInt32ToEnumField(3, TestMap.EnumValue.BAZ) |
| |
| .putInt32ToMessageField(1, MessageValue.newBuilder().setValue(11).build()) |
| .putInt32ToMessageField(2, MessageValue.newBuilder().setValue(22).build()) |
| .putInt32ToMessageField(3, MessageValue.newBuilder().setValue(33).build()) |
| |
| .putStringToInt32Field("1", 11) |
| .putStringToInt32Field("2", 22) |
| .putStringToInt32Field("3", 33); |
| } |
| |
| public void testSetMapValues() { |
| TestMap.Builder mapBuilder = TestMap.newBuilder(); |
| setMapValues(mapBuilder); |
| TestMap map = mapBuilder.build(); |
| assertMapValuesSet(map); |
| } |
| |
| private void copyMapValues(TestMap source, TestMap.Builder destination) { |
| destination |
| .putAllInt32ToInt32Field(source.getInt32ToInt32Field()) |
| .putAllInt32ToStringField(source.getInt32ToStringField()) |
| .putAllInt32ToBytesField(source.getInt32ToBytesField()) |
| .putAllInt32ToEnumField(source.getInt32ToEnumField()) |
| .putAllInt32ToMessageField(source.getInt32ToMessageField()) |
| .putAllStringToInt32Field(source.getStringToInt32Field()); |
| } |
| |
| private void assertMapValuesSet(TestMap message) { |
| assertEquals(3, message.getInt32ToInt32Field().size()); |
| assertEquals(11, message.getInt32ToInt32Field().get(1).intValue()); |
| assertEquals(22, message.getInt32ToInt32Field().get(2).intValue()); |
| assertEquals(33, message.getInt32ToInt32Field().get(3).intValue()); |
| |
| assertEquals(3, message.getInt32ToStringField().size()); |
| assertEquals("11", message.getInt32ToStringField().get(1)); |
| assertEquals("22", message.getInt32ToStringField().get(2)); |
| assertEquals("33", message.getInt32ToStringField().get(3)); |
| |
| assertEquals(3, message.getInt32ToBytesField().size()); |
| assertEquals(TestUtil.toBytes("11"), message.getInt32ToBytesField().get(1)); |
| assertEquals(TestUtil.toBytes("22"), message.getInt32ToBytesField().get(2)); |
| assertEquals(TestUtil.toBytes("33"), message.getInt32ToBytesField().get(3)); |
| |
| assertEquals(3, message.getInt32ToEnumField().size()); |
| assertEquals(TestMap.EnumValue.FOO, message.getInt32ToEnumField().get(1)); |
| assertEquals(TestMap.EnumValue.BAR, message.getInt32ToEnumField().get(2)); |
| assertEquals(TestMap.EnumValue.BAZ, message.getInt32ToEnumField().get(3)); |
| |
| assertEquals(3, message.getInt32ToMessageField().size()); |
| assertEquals(11, message.getInt32ToMessageField().get(1).getValue()); |
| assertEquals(22, message.getInt32ToMessageField().get(2).getValue()); |
| assertEquals(33, message.getInt32ToMessageField().get(3).getValue()); |
| |
| assertEquals(3, message.getStringToInt32Field().size()); |
| assertEquals(11, message.getStringToInt32Field().get("1").intValue()); |
| assertEquals(22, message.getStringToInt32Field().get("2").intValue()); |
| assertEquals(33, message.getStringToInt32Field().get("3").intValue()); |
| } |
| |
| private void updateMapValues(TestMap.Builder builder) { |
| builder |
| .putInt32ToInt32Field(1, 111) |
| .removeInt32ToInt32Field(2) |
| .putInt32ToInt32Field(4, 44) |
| |
| .putInt32ToStringField(1, "111") |
| .removeInt32ToStringField(2) |
| .putInt32ToStringField(4, "44") |
| |
| .putInt32ToBytesField(1, TestUtil.toBytes("111")) |
| .removeInt32ToBytesField(2) |
| .putInt32ToBytesField(4, TestUtil.toBytes("44")) |
| |
| .putInt32ToEnumField(1, TestMap.EnumValue.BAR) |
| .removeInt32ToEnumField(2) |
| .putInt32ToEnumField(4, TestMap.EnumValue.QUX) |
| |
| .putInt32ToMessageField(1, MessageValue.newBuilder().setValue(111).build()) |
| .removeInt32ToMessageField(2) |
| .putInt32ToMessageField(4, MessageValue.newBuilder().setValue(44).build()) |
| |
| .putStringToInt32Field("1", 111) |
| .removeStringToInt32Field("2") |
| .putStringToInt32Field("4", 44); |
| } |
| |
| public void testUpdateMapValues() { |
| TestMap.Builder mapBuilder = TestMap.newBuilder(); |
| setMapValues(mapBuilder); |
| TestMap map = mapBuilder.build(); |
| assertMapValuesSet(map); |
| |
| mapBuilder = map.toBuilder(); |
| updateMapValues(mapBuilder); |
| map = mapBuilder.build(); |
| assertMapValuesUpdated(map); |
| } |
| |
| private void assertMapValuesUpdated(TestMap message) { |
| assertEquals(3, message.getInt32ToInt32Field().size()); |
| assertEquals(111, message.getInt32ToInt32Field().get(1).intValue()); |
| assertEquals(33, message.getInt32ToInt32Field().get(3).intValue()); |
| assertEquals(44, message.getInt32ToInt32Field().get(4).intValue()); |
| |
| assertEquals(3, message.getInt32ToStringField().size()); |
| assertEquals("111", message.getInt32ToStringField().get(1)); |
| assertEquals("33", message.getInt32ToStringField().get(3)); |
| assertEquals("44", message.getInt32ToStringField().get(4)); |
| |
| assertEquals(3, message.getInt32ToBytesField().size()); |
| assertEquals(TestUtil.toBytes("111"), message.getInt32ToBytesField().get(1)); |
| assertEquals(TestUtil.toBytes("33"), message.getInt32ToBytesField().get(3)); |
| assertEquals(TestUtil.toBytes("44"), message.getInt32ToBytesField().get(4)); |
| |
| assertEquals(3, message.getInt32ToEnumField().size()); |
| assertEquals(TestMap.EnumValue.BAR, message.getInt32ToEnumField().get(1)); |
| assertEquals(TestMap.EnumValue.BAZ, message.getInt32ToEnumField().get(3)); |
| assertEquals(TestMap.EnumValue.QUX, message.getInt32ToEnumField().get(4)); |
| |
| assertEquals(3, message.getInt32ToMessageField().size()); |
| assertEquals(111, message.getInt32ToMessageField().get(1).getValue()); |
| assertEquals(33, message.getInt32ToMessageField().get(3).getValue()); |
| assertEquals(44, message.getInt32ToMessageField().get(4).getValue()); |
| |
| assertEquals(3, message.getStringToInt32Field().size()); |
| assertEquals(111, message.getStringToInt32Field().get("1").intValue()); |
| assertEquals(33, message.getStringToInt32Field().get("3").intValue()); |
| assertEquals(44, message.getStringToInt32Field().get("4").intValue()); |
| } |
| |
| private void assertMapValuesCleared(TestMapOrBuilder testMapOrBuilder) { |
| assertEquals(0, testMapOrBuilder.getInt32ToInt32Field().size()); |
| assertEquals(0, testMapOrBuilder.getInt32ToInt32FieldCount()); |
| assertEquals(0, testMapOrBuilder.getInt32ToStringField().size()); |
| assertEquals(0, testMapOrBuilder.getInt32ToStringFieldCount()); |
| assertEquals(0, testMapOrBuilder.getInt32ToBytesField().size()); |
| assertEquals(0, testMapOrBuilder.getInt32ToBytesFieldCount()); |
| assertEquals(0, testMapOrBuilder.getInt32ToEnumField().size()); |
| assertEquals(0, testMapOrBuilder.getInt32ToEnumFieldCount()); |
| assertEquals(0, testMapOrBuilder.getInt32ToMessageField().size()); |
| assertEquals(0, testMapOrBuilder.getInt32ToMessageFieldCount()); |
| assertEquals(0, testMapOrBuilder.getStringToInt32Field().size()); |
| assertEquals(0, testMapOrBuilder.getStringToInt32FieldCount()); |
| } |
| |
| public void testSanityCopyOnWrite() throws InvalidProtocolBufferException { |
| // Since builders are implemented as a thin wrapper around a message |
| // instance, we attempt to verify that we can't cause the builder to modify |
| // a produced message. |
| |
| TestMap.Builder builder = TestMap.newBuilder(); |
| TestMap message = builder.build(); |
| builder.putInt32ToInt32Field(1, 2); |
| assertTrue(message.getInt32ToInt32Field().isEmpty()); |
| message = builder.build(); |
| assertEquals(newMap(1, 2), message.getInt32ToInt32Field()); |
| assertEquals(newMap(1, 2), builder.getInt32ToInt32Field()); |
| builder.putInt32ToInt32Field(2, 3); |
| assertEquals(newMap(1, 2), message.getInt32ToInt32Field()); |
| assertEquals(newMap(1, 2, 2, 3), builder.getInt32ToInt32Field()); |
| } |
| |
| public void testGetMapIsImmutable() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| assertMapsAreImmutable(builder); |
| assertMapsAreImmutable(builder.build()); |
| |
| setMapValues(builder); |
| assertMapsAreImmutable(builder); |
| assertMapsAreImmutable(builder.build()); |
| } |
| |
| private void assertMapsAreImmutable(TestMapOrBuilder testMapOrBuilder) { |
| assertImmutable(testMapOrBuilder.getInt32ToInt32Field(), 1, 2); |
| assertImmutable(testMapOrBuilder.getInt32ToStringField(), 1, "2"); |
| assertImmutable(testMapOrBuilder.getInt32ToBytesField(), 1, TestUtil.toBytes("2")); |
| assertImmutable(testMapOrBuilder.getInt32ToEnumField(), 1, TestMap.EnumValue.FOO); |
| assertImmutable( |
| testMapOrBuilder.getInt32ToMessageField(), 1, MessageValue.getDefaultInstance()); |
| assertImmutable(testMapOrBuilder.getStringToInt32Field(), "1", 2); |
| } |
| |
| private <K, V> void assertImmutable(Map<K, V> map, K key, V value) { |
| try { |
| map.put(key, value); |
| fail(); |
| } catch (UnsupportedOperationException e) { |
| // expected |
| } |
| if (!map.isEmpty()) { |
| try { |
| map.entrySet().remove(map.entrySet().iterator().next()); |
| fail(); |
| } catch (UnsupportedOperationException e) { |
| // expected |
| } |
| } |
| } |
| |
| public void testMutableMapLifecycle() { |
| TestMap.Builder builder = TestMap.newBuilder() |
| .putInt32ToInt32Field(1, 2); |
| assertEquals(newMap(1, 2), builder.build().getInt32ToInt32Field()); |
| assertEquals(newMap(1, 2), builder.getInt32ToInt32Field()); |
| builder.putInt32ToInt32Field(2, 3); |
| assertEquals(newMap(1, 2, 2, 3), builder.getInt32ToInt32Field()); |
| |
| builder.putInt32ToEnumField(1, TestMap.EnumValue.BAR); |
| assertEquals(newMap(1, TestMap.EnumValue.BAR), builder.build().getInt32ToEnumField()); |
| assertEquals(newMap(1, TestMap.EnumValue.BAR), builder.getInt32ToEnumField()); |
| builder.putInt32ToEnumField(2, TestMap.EnumValue.FOO); |
| assertEquals( |
| newMap(1, TestMap.EnumValue.BAR, 2, TestMap.EnumValue.FOO), |
| builder.getInt32ToEnumField()); |
| |
| builder.putInt32ToStringField(1, "1"); |
| assertEquals(newMap(1, "1"), builder.build().getInt32ToStringField()); |
| assertEquals(newMap(1, "1"), builder.getInt32ToStringField()); |
| builder.putInt32ToStringField(2, "2"); |
| assertEquals(newMap(1, "1", 2, "2"), builder.getInt32ToStringField()); |
| |
| builder.putInt32ToMessageField(1, TestMap.MessageValue.getDefaultInstance()); |
| assertEquals(newMap(1, TestMap.MessageValue.getDefaultInstance()), |
| builder.build().getInt32ToMessageField()); |
| assertEquals(newMap(1, TestMap.MessageValue.getDefaultInstance()), |
| builder.getInt32ToMessageField()); |
| builder.putInt32ToMessageField(2, TestMap.MessageValue.getDefaultInstance()); |
| assertEquals( |
| newMap(1, TestMap.MessageValue.getDefaultInstance(), |
| 2, TestMap.MessageValue.getDefaultInstance()), |
| builder.getInt32ToMessageField()); |
| } |
| |
| public void testGettersAndSetters() throws Exception { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| TestMap message = builder.build(); |
| assertMapValuesCleared(message); |
| |
| builder = message.toBuilder(); |
| setMapValues(builder); |
| message = builder.build(); |
| assertMapValuesSet(message); |
| |
| builder = message.toBuilder(); |
| updateMapValues(builder); |
| message = builder.build(); |
| assertMapValuesUpdated(message); |
| |
| builder = message.toBuilder(); |
| builder.clear(); |
| assertMapValuesCleared(builder); |
| message = builder.build(); |
| assertMapValuesCleared(message); |
| } |
| |
| public void testPutAll() throws Exception { |
| TestMap.Builder sourceBuilder = TestMap.newBuilder(); |
| setMapValues(sourceBuilder); |
| TestMap source = sourceBuilder.build(); |
| assertMapValuesSet(source); |
| |
| TestMap.Builder destination = TestMap.newBuilder(); |
| copyMapValues(source, destination); |
| assertMapValuesSet(destination.build()); |
| } |
| |
| public void testPutChecksNullKeysAndValues() throws Exception { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| |
| try { |
| builder.putInt32ToStringField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected. |
| } |
| |
| try { |
| builder.putInt32ToBytesField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected. |
| } |
| |
| try { |
| builder.putInt32ToEnumField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected. |
| } |
| |
| try { |
| builder.putInt32ToMessageField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected. |
| } |
| |
| try { |
| builder.putStringToInt32Field(null, 1); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected. |
| } |
| } |
| |
| public void testSerializeAndParse() throws Exception { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| TestMap message = builder.build(); |
| assertEquals(message.getSerializedSize(), message.toByteString().size()); |
| message = TestMap.parser().parseFrom(message.toByteString()); |
| assertMapValuesSet(message); |
| |
| builder = message.toBuilder(); |
| updateMapValues(builder); |
| message = builder.build(); |
| assertEquals(message.getSerializedSize(), message.toByteString().size()); |
| message = TestMap.parser().parseFrom(message.toByteString()); |
| assertMapValuesUpdated(message); |
| |
| builder = message.toBuilder(); |
| builder.clear(); |
| message = builder.build(); |
| assertEquals(message.getSerializedSize(), message.toByteString().size()); |
| message = TestMap.parser().parseFrom(message.toByteString()); |
| assertMapValuesCleared(message); |
| } |
| |
| private TestMap tryParseTestMap(BizarroTestMap bizarroMap) throws IOException { |
| ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); |
| CodedOutputStream output = CodedOutputStream.newInstance(byteArrayOutputStream); |
| bizarroMap.writeTo(output); |
| output.flush(); |
| return TestMap.parser().parseFrom(ByteString.copyFrom(byteArrayOutputStream.toByteArray())); |
| } |
| |
| public void testParseError() throws Exception { |
| ByteString bytes = TestUtil.toBytes("SOME BYTES"); |
| String stringKey = "a string key"; |
| |
| TestMap map = tryParseTestMap(BizarroTestMap.newBuilder() |
| .putInt32ToInt32Field(5, bytes) |
| .build()); |
| assertEquals(0, map.getInt32ToInt32FieldOrDefault(5, -1)); |
| |
| map = tryParseTestMap(BizarroTestMap.newBuilder() |
| .putInt32ToStringField(stringKey, 5) |
| .build()); |
| assertEquals("", map.getInt32ToStringFieldOrDefault(0, null)); |
| |
| map = tryParseTestMap(BizarroTestMap.newBuilder() |
| .putInt32ToBytesField(stringKey, 5) |
| .build()); |
| assertEquals(map.getInt32ToBytesFieldOrDefault(0, null), ByteString.EMPTY); |
| |
| map = tryParseTestMap(BizarroTestMap.newBuilder() |
| .putInt32ToEnumField(stringKey, bytes) |
| .build()); |
| assertEquals(TestMap.EnumValue.FOO, map.getInt32ToEnumFieldOrDefault(0, null)); |
| |
| try { |
| tryParseTestMap(BizarroTestMap.newBuilder() |
| .putInt32ToMessageField(stringKey, bytes) |
| .build()); |
| fail(); |
| } catch (InvalidProtocolBufferException expected) { |
| assertTrue(expected.getUnfinishedMessage() instanceof TestMap); |
| map = (TestMap) expected.getUnfinishedMessage(); |
| assertTrue(map.getInt32ToMessageField().isEmpty()); |
| } |
| |
| map = tryParseTestMap(BizarroTestMap.newBuilder() |
| .putStringToInt32Field(stringKey, bytes) |
| .build()); |
| assertEquals(0, map.getStringToInt32FieldOrDefault(stringKey, -1)); |
| } |
| |
| public void testMergeFrom() throws Exception { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| TestMap message = builder.build(); |
| |
| TestMap.Builder other = TestMap.newBuilder(); |
| other.mergeFrom(message); |
| assertMapValuesSet(other.build()); |
| } |
| |
| public void testEqualsAndHashCode() throws Exception { |
| // Test that generated equals() and hashCode() will disregard the order |
| // of map entries when comparing/hashing map fields. |
| |
| // We can't control the order of elements in a HashMap. The best we can do |
| // here is to add elements in different order. |
| TestMap.Builder b1 = TestMap.newBuilder() |
| .putInt32ToInt32Field(1, 2) |
| .putInt32ToInt32Field(3, 4) |
| .putInt32ToInt32Field(5, 6); |
| TestMap m1 = b1.build(); |
| |
| TestMap.Builder b2 = TestMap.newBuilder() |
| .putInt32ToInt32Field(5, 6) |
| .putInt32ToInt32Field(1, 2) |
| .putInt32ToInt32Field(3, 4); |
| TestMap m2 = b2.build(); |
| |
| assertEquals(m1, m2); |
| assertEquals(m1.hashCode(), m2.hashCode()); |
| |
| // Make sure we did compare map fields. |
| b2.putInt32ToInt32Field(1, 0); |
| m2 = b2.build(); |
| assertFalse(m1.equals(m2)); |
| // Don't check m1.hashCode() != m2.hashCode() because it's not guaranteed |
| // to be different. |
| } |
| |
| public void testUnknownEnumValues() throws Exception { |
| TestUnknownEnumValue.Builder builder = TestUnknownEnumValue.newBuilder() |
| .putInt32ToInt32Field(1, 1) |
| .putInt32ToInt32Field(2, 54321); |
| ByteString data = builder.build().toByteString(); |
| |
| TestMap message = TestMap.parseFrom(data); |
| // Entries with unknown enum values will be stored into UnknownFieldSet so |
| // there is only one entry in the map. |
| assertEquals(1, message.getInt32ToEnumField().size()); |
| assertEquals(TestMap.EnumValue.BAR, message.getInt32ToEnumField().get(1)); |
| // Serializing and parsing should preserve the unknown entry. |
| data = message.toByteString(); |
| TestUnknownEnumValue messageWithUnknownEnums = |
| TestUnknownEnumValue.parseFrom(data); |
| assertEquals(2, messageWithUnknownEnums.getInt32ToInt32Field().size()); |
| assertEquals(1, messageWithUnknownEnums.getInt32ToInt32Field().get(1).intValue()); |
| assertEquals(54321, messageWithUnknownEnums.getInt32ToInt32Field().get(2).intValue()); |
| } |
| |
| public void testIterationOrder() throws Exception { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| TestMap message = builder.build(); |
| |
| assertEquals(Arrays.asList("1", "2", "3"), |
| new ArrayList<String>(message.getStringToInt32Field().keySet())); |
| } |
| |
| private static <K, V> Map<K, V> newMap(K key1, V value1) { |
| Map<K, V> map = new HashMap<K, V>(); |
| map.put(key1, value1); |
| return map; |
| } |
| |
| private static <K, V> Map<K, V> newMap(K key1, V value1, K key2, V value2) { |
| Map<K, V> map = new HashMap<K, V>(); |
| map.put(key1, value1); |
| map.put(key2, value2); |
| return map; |
| } |
| |
| public void testGetMap() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| TestMap message = builder.build(); |
| assertEquals( |
| message.getStringToInt32Field(), |
| message.getStringToInt32FieldMap()); |
| assertEquals( |
| message.getInt32ToBytesField(), |
| message.getInt32ToBytesFieldMap()); |
| assertEquals( |
| message.getInt32ToEnumField(), |
| message.getInt32ToEnumFieldMap()); |
| assertEquals( |
| message.getInt32ToMessageField(), |
| message.getInt32ToMessageFieldMap()); |
| } |
| |
| public void testContains() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| assertMapContainsSetValues(builder); |
| assertMapContainsSetValues(builder.build()); |
| } |
| |
| private void assertMapContainsSetValues(TestMapOrBuilder testMapOrBuilder) { |
| assertTrue(testMapOrBuilder.containsInt32ToInt32Field(1)); |
| assertTrue(testMapOrBuilder.containsInt32ToInt32Field(2)); |
| assertTrue(testMapOrBuilder.containsInt32ToInt32Field(3)); |
| assertFalse(testMapOrBuilder.containsInt32ToInt32Field(-1)); |
| |
| assertTrue(testMapOrBuilder.containsInt32ToStringField(1)); |
| assertTrue(testMapOrBuilder.containsInt32ToStringField(2)); |
| assertTrue(testMapOrBuilder.containsInt32ToStringField(3)); |
| assertFalse(testMapOrBuilder.containsInt32ToStringField(-1)); |
| |
| assertTrue(testMapOrBuilder.containsInt32ToBytesField(1)); |
| assertTrue(testMapOrBuilder.containsInt32ToBytesField(2)); |
| assertTrue(testMapOrBuilder.containsInt32ToBytesField(3)); |
| assertFalse(testMapOrBuilder.containsInt32ToBytesField(-1)); |
| |
| assertTrue(testMapOrBuilder.containsInt32ToEnumField(1)); |
| assertTrue(testMapOrBuilder.containsInt32ToEnumField(2)); |
| assertTrue(testMapOrBuilder.containsInt32ToEnumField(3)); |
| assertFalse(testMapOrBuilder.containsInt32ToEnumField(-1)); |
| |
| assertTrue(testMapOrBuilder.containsInt32ToMessageField(1)); |
| assertTrue(testMapOrBuilder.containsInt32ToMessageField(2)); |
| assertTrue(testMapOrBuilder.containsInt32ToMessageField(3)); |
| assertFalse(testMapOrBuilder.containsInt32ToMessageField(-1)); |
| |
| assertTrue(testMapOrBuilder.containsStringToInt32Field("1")); |
| assertTrue(testMapOrBuilder.containsStringToInt32Field("2")); |
| assertTrue(testMapOrBuilder.containsStringToInt32Field("3")); |
| assertFalse(testMapOrBuilder.containsStringToInt32Field("-1")); |
| } |
| |
| public void testCount() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| assertMapCounts(0, builder); |
| |
| setMapValues(builder); |
| assertMapCounts(3, builder); |
| |
| TestMap message = builder.build(); |
| assertMapCounts(3, message); |
| |
| builder = message.toBuilder().putInt32ToInt32Field(4, 44); |
| assertEquals(4, builder.getInt32ToInt32FieldCount()); |
| assertEquals(4, builder.build().getInt32ToInt32FieldCount()); |
| |
| // already present - should be unchanged |
| builder.putInt32ToInt32Field(4, 44); |
| assertEquals(4, builder.getInt32ToInt32FieldCount()); |
| } |
| |
| private void assertMapCounts(int expectedCount, TestMapOrBuilder testMapOrBuilder) { |
| assertEquals(expectedCount, testMapOrBuilder.getInt32ToInt32FieldCount()); |
| assertEquals(expectedCount, testMapOrBuilder.getInt32ToStringFieldCount()); |
| assertEquals(expectedCount, testMapOrBuilder.getInt32ToBytesFieldCount()); |
| assertEquals(expectedCount, testMapOrBuilder.getInt32ToEnumFieldCount()); |
| assertEquals(expectedCount, testMapOrBuilder.getInt32ToMessageFieldCount()); |
| assertEquals(expectedCount, testMapOrBuilder.getStringToInt32FieldCount()); |
| } |
| |
| public void testGetOrDefault() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| assertMapCounts(0, builder); |
| setMapValues(builder); |
| doTestGetOrDefault(builder); |
| doTestGetOrDefault(builder.build()); |
| } |
| |
| public void doTestGetOrDefault(TestMapOrBuilder testMapOrBuilder) { |
| assertEquals(11, testMapOrBuilder.getInt32ToInt32FieldOrDefault(1, -11)); |
| assertEquals(-11, testMapOrBuilder.getInt32ToInt32FieldOrDefault(-1, -11)); |
| |
| assertEquals("11", testMapOrBuilder.getInt32ToStringFieldOrDefault(1, "-11")); |
| assertNull("-11", testMapOrBuilder.getInt32ToStringFieldOrDefault(-1, null)); |
| |
| assertEquals(TestUtil.toBytes("11"), testMapOrBuilder.getInt32ToBytesFieldOrDefault(1, null)); |
| assertNull(testMapOrBuilder.getInt32ToBytesFieldOrDefault(-1, null)); |
| |
| assertEquals(TestMap.EnumValue.FOO, testMapOrBuilder.getInt32ToEnumFieldOrDefault(1, null)); |
| assertNull(testMapOrBuilder.getInt32ToEnumFieldOrDefault(-1, null)); |
| |
| assertEquals(MessageValue.newBuilder().setValue(11).build(), |
| testMapOrBuilder.getInt32ToMessageFieldOrDefault(1, null)); |
| assertNull(testMapOrBuilder.getInt32ToMessageFieldOrDefault(-1, null)); |
| |
| assertEquals(11, testMapOrBuilder.getStringToInt32FieldOrDefault("1", -11)); |
| assertEquals(-11, testMapOrBuilder.getStringToInt32FieldOrDefault("-1", -11)); |
| |
| try { |
| testMapOrBuilder.getStringToInt32FieldOrDefault(null, -11); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| } |
| |
| public void testGetOrThrow() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| assertMapCounts(0, builder); |
| setMapValues(builder); |
| doTestGetOrDefault(builder); |
| doTestGetOrDefault(builder.build()); |
| } |
| |
| public void doTestGetOrThrow(TestMapOrBuilder testMapOrBuilder) { |
| assertEquals(11, testMapOrBuilder.getInt32ToInt32FieldOrThrow(1)); |
| try { |
| testMapOrBuilder.getInt32ToInt32FieldOrThrow(-1); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertEquals("11", testMapOrBuilder.getInt32ToStringFieldOrThrow(1)); |
| |
| try { |
| testMapOrBuilder.getInt32ToStringFieldOrThrow(-1); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertEquals(TestUtil.toBytes("11"), testMapOrBuilder.getInt32ToBytesFieldOrThrow(1)); |
| |
| try { |
| testMapOrBuilder.getInt32ToBytesFieldOrThrow(-1); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertEquals(TestMap.EnumValue.FOO, testMapOrBuilder.getInt32ToEnumFieldOrThrow(1)); |
| try { |
| testMapOrBuilder.getInt32ToEnumFieldOrThrow(-1); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertEquals(MessageValue.newBuilder().setValue(11).build(), |
| testMapOrBuilder.getInt32ToMessageFieldOrThrow(1)); |
| try { |
| testMapOrBuilder.getInt32ToMessageFieldOrThrow(-1); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| assertEquals(11, testMapOrBuilder.getStringToInt32FieldOrThrow("1")); |
| try { |
| testMapOrBuilder.getStringToInt32FieldOrThrow("-1"); |
| fail(); |
| } catch (IllegalArgumentException e) { |
| // expected |
| } |
| |
| try { |
| testMapOrBuilder.getStringToInt32FieldOrThrow(null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| } |
| |
| public void testPut() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| builder.putInt32ToInt32Field(1, 11); |
| assertEquals(11, builder.getInt32ToInt32FieldOrThrow(1)); |
| |
| builder.putInt32ToStringField(1, "a"); |
| assertEquals("a", builder.getInt32ToStringFieldOrThrow(1)); |
| try { |
| builder.putInt32ToStringField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| |
| builder.putInt32ToBytesField(1, TestUtil.toBytes("11")); |
| assertEquals(TestUtil.toBytes("11"), builder.getInt32ToBytesFieldOrThrow(1)); |
| try { |
| builder.putInt32ToBytesField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| |
| builder.putInt32ToEnumField(1, TestMap.EnumValue.FOO); |
| assertEquals(TestMap.EnumValue.FOO, builder.getInt32ToEnumFieldOrThrow(1)); |
| try { |
| builder.putInt32ToEnumField(1, null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| |
| builder.putStringToInt32Field("a", 1); |
| assertEquals(1, builder.getStringToInt32FieldOrThrow("a")); |
| try { |
| builder.putStringToInt32Field(null, -1); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| } |
| |
| public void testRemove() { |
| TestMap.Builder builder = TestMap.newBuilder(); |
| setMapValues(builder); |
| assertEquals(11, builder.getInt32ToInt32FieldOrThrow(1)); |
| for (int times = 0; times < 2; times++) { |
| builder.removeInt32ToInt32Field(1); |
| assertEquals(-1, builder.getInt32ToInt32FieldOrDefault(1, -1)); |
| } |
| |
| assertEquals("11", builder.getInt32ToStringFieldOrThrow(1)); |
| for (int times = 0; times < 2; times++) { |
| builder.removeInt32ToStringField(1); |
| assertNull(builder.getInt32ToStringFieldOrDefault(1, null)); |
| } |
| |
| assertEquals(TestUtil.toBytes("11"), builder.getInt32ToBytesFieldOrThrow(1)); |
| for (int times = 0; times < 2; times++) { |
| builder.removeInt32ToBytesField(1); |
| assertNull(builder.getInt32ToBytesFieldOrDefault(1, null)); |
| } |
| |
| assertEquals(TestMap.EnumValue.FOO, builder.getInt32ToEnumFieldOrThrow(1)); |
| for (int times = 0; times < 2; times++) { |
| builder.removeInt32ToEnumField(1); |
| assertNull(builder.getInt32ToEnumFieldOrDefault(1, null)); |
| } |
| |
| assertEquals(11, builder.getStringToInt32FieldOrThrow("1")); |
| for (int times = 0; times < 2; times++) { |
| builder.removeStringToInt32Field("1"); |
| assertEquals(-1, builder.getStringToInt32FieldOrDefault("1", -1)); |
| } |
| |
| try { |
| builder.removeStringToInt32Field(null); |
| fail(); |
| } catch (NullPointerException e) { |
| // expected |
| } |
| } |
| } |