// 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 com.google.protobuf.Descriptors.Descriptor; import com.google.protobuf.Descriptors.EnumDescriptor; import com.google.protobuf.Descriptors.EnumValueDescriptor; import com.google.protobuf.Descriptors.FieldDescriptor; import map_test.MapTestProto.BizarroTestMap; import map_test.MapTestProto.ReservedAsMapField; import map_test.MapTestProto.ReservedAsMapFieldWithEnumValue; import map_test.MapTestProto.TestMap; import map_test.MapTestProto.TestMap.MessageValue; import map_test.MapTestProto.TestMapOrBuilder; import map_test.MapTestProto.TestOnChangeEventPropagation; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.util.ArrayList; import java.util.Arrays; import java.util.HashMap; import java.util.List; import java.util.Map; import junit.framework.TestCase; /** * Unit tests for map fields. */ public class MapTest extends TestCase { private void setMapValuesUsingMutableMap(TestMap.Builder builder) { builder.getMutableInt32ToInt32Field().put(1, 11); builder.getMutableInt32ToInt32Field().put(2, 22); builder.getMutableInt32ToInt32Field().put(3, 33); // builder.getMutableInt32ToStringField().put(1, "11"); builder.getMutableInt32ToStringField().put(2, "22"); builder.getMutableInt32ToStringField().put(3, "33"); // builder.getMutableInt32ToBytesField().put(1, TestUtil.toBytes("11")); builder.getMutableInt32ToBytesField().put(2, TestUtil.toBytes("22")); builder.getMutableInt32ToBytesField().put(3, TestUtil.toBytes("33")); // builder.getMutableInt32ToEnumField().put(1, TestMap.EnumValue.FOO); builder.getMutableInt32ToEnumField().put(2, TestMap.EnumValue.BAR); builder.getMutableInt32ToEnumField().put(3, TestMap.EnumValue.BAZ); // builder.getMutableInt32ToMessageField().put( 1, MessageValue.newBuilder().setValue(11).build()); builder.getMutableInt32ToMessageField().put( 2, MessageValue.newBuilder().setValue(22).build()); builder.getMutableInt32ToMessageField().put( 3, MessageValue.newBuilder().setValue(33).build()); // builder.getMutableStringToInt32Field().put("1", 11); builder.getMutableStringToInt32Field().put("2", 22); builder.getMutableStringToInt32Field().put("3", 33); } private void setMapValuesUsingAccessors(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 usingMutableMapBuilder = TestMap.newBuilder(); setMapValuesUsingMutableMap(usingMutableMapBuilder); TestMap usingMutableMap = usingMutableMapBuilder.build(); assertMapValuesSet(usingMutableMap); TestMap.Builder usingAccessorsBuilder = TestMap.newBuilder(); setMapValuesUsingAccessors(usingAccessorsBuilder); TestMap usingAccessors = usingAccessorsBuilder.build(); assertMapValuesSet(usingAccessors); assertEquals(usingAccessors, usingMutableMap); } 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 updateMapValuesUsingMutableMap(TestMap.Builder builder) { builder.getMutableInt32ToInt32Field().put(1, 111); builder.getMutableInt32ToInt32Field().remove(2); builder.getMutableInt32ToInt32Field().put(4, 44); // builder.getMutableInt32ToStringField().put(1, "111"); builder.getMutableInt32ToStringField().remove(2); builder.getMutableInt32ToStringField().put(4, "44"); // builder.getMutableInt32ToBytesField().put(1, TestUtil.toBytes("111")); builder.getMutableInt32ToBytesField().remove(2); builder.getMutableInt32ToBytesField().put(4, TestUtil.toBytes("44")); // builder.getMutableInt32ToEnumField().put(1, TestMap.EnumValue.BAR); builder.getMutableInt32ToEnumField().remove(2); builder.getMutableInt32ToEnumField().put(4, TestMap.EnumValue.QUX); // builder.getMutableInt32ToMessageField().put( 1, MessageValue.newBuilder().setValue(111).build()); builder.getMutableInt32ToMessageField().remove(2); builder.getMutableInt32ToMessageField().put( 4, MessageValue.newBuilder().setValue(44).build()); // builder.getMutableStringToInt32Field().put("1", 111); builder.getMutableStringToInt32Field().remove("2"); builder.getMutableStringToInt32Field().put("4", 44); } private void updateMapValuesUsingAccessors(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 usingMutableMapBuilder = TestMap.newBuilder(); setMapValuesUsingMutableMap(usingMutableMapBuilder); TestMap usingMutableMap = usingMutableMapBuilder.build(); assertMapValuesSet(usingMutableMap); TestMap.Builder usingAccessorsBuilder = TestMap.newBuilder(); setMapValuesUsingAccessors(usingAccessorsBuilder); TestMap usingAccessors = usingAccessorsBuilder.build(); assertMapValuesSet(usingAccessors); assertEquals(usingAccessors, usingMutableMap); // usingMutableMapBuilder = usingMutableMap.toBuilder(); updateMapValuesUsingMutableMap(usingMutableMapBuilder); usingMutableMap = usingMutableMapBuilder.build(); assertMapValuesUpdated(usingMutableMap); usingAccessorsBuilder = usingAccessors.toBuilder(); updateMapValuesUsingAccessors(usingAccessorsBuilder); usingAccessors = usingAccessorsBuilder.build(); assertMapValuesUpdated(usingAccessors); assertEquals(usingAccessors, usingMutableMap); } 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 testGetMapIsImmutable() { TestMap.Builder builder = TestMap.newBuilder(); assertMapsAreImmutable(builder); assertMapsAreImmutable(builder.build()); setMapValuesUsingAccessors(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 void assertImmutable(Map map, K key, V value) { try { map.put(key, value); fail(); } catch (UnsupportedOperationException e) { // expected } } public void testMutableMapLifecycle() { TestMap.Builder builder = TestMap.newBuilder(); Map intMap = builder.getMutableInt32ToInt32Field(); intMap.put(1, 2); assertEquals(newMap(1, 2), builder.build().getInt32ToInt32Field()); try { intMap.put(2, 3); fail(); } catch (UnsupportedOperationException e) { // expected } assertEquals(newMap(1, 2), builder.getInt32ToInt32Field()); builder.getMutableInt32ToInt32Field().put(2, 3); assertEquals(newMap(1, 2, 2, 3), builder.getInt32ToInt32Field()); // Map enumMap = builder.getMutableInt32ToEnumField(); enumMap.put(1, TestMap.EnumValue.BAR); assertEquals(newMap(1, TestMap.EnumValue.BAR), builder.build().getInt32ToEnumField()); try { enumMap.put(2, TestMap.EnumValue.FOO); fail(); } catch (UnsupportedOperationException e) { // expected } assertEquals(newMap(1, TestMap.EnumValue.BAR), builder.getInt32ToEnumField()); builder.getMutableInt32ToEnumField().put(2, TestMap.EnumValue.FOO); assertEquals( newMap(1, TestMap.EnumValue.BAR, 2, TestMap.EnumValue.FOO), builder.getInt32ToEnumField()); // Map stringMap = builder.getMutableInt32ToStringField(); stringMap.put(1, "1"); assertEquals(newMap(1, "1"), builder.build().getInt32ToStringField()); try { stringMap.put(2, "2"); fail(); } catch (UnsupportedOperationException e) { // expected } assertEquals(newMap(1, "1"), builder.getInt32ToStringField()); builder.putInt32ToStringField(2, "2"); assertEquals( newMap(1, "1", 2, "2"), builder.getInt32ToStringField()); // Map messageMap = builder.getMutableInt32ToMessageField(); messageMap.put(1, TestMap.MessageValue.getDefaultInstance()); assertEquals(newMap(1, TestMap.MessageValue.getDefaultInstance()), builder.build().getInt32ToMessageField()); try { messageMap.put(2, TestMap.MessageValue.getDefaultInstance()); fail(); } catch (UnsupportedOperationException e) { // expected } 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 testMutableMapLifecycle_collections() { TestMap.Builder builder = TestMap.newBuilder(); Map intMap = builder.getMutableInt32ToInt32Field(); intMap.put(1, 2); assertEquals(newMap(1, 2), builder.build().getInt32ToInt32Field()); try { intMap.remove(2); fail(); } catch (UnsupportedOperationException e) { // expected } try { intMap.entrySet().remove(new Object()); fail(); } catch (UnsupportedOperationException e) { // expected } try { intMap.entrySet().iterator().remove(); fail(); } catch (UnsupportedOperationException e) { // expected } try { intMap.keySet().remove(new Object()); fail(); } catch (UnsupportedOperationException e) { // expected } try { intMap.values().remove(new Object()); fail(); } catch (UnsupportedOperationException e) { // expected } try { intMap.values().iterator().remove(); fail(); } catch (UnsupportedOperationException e) { // expected } assertEquals(newMap(1, 2), intMap); assertEquals(newMap(1, 2), builder.getInt32ToInt32Field()); assertEquals(newMap(1, 2), builder.build().getInt32ToInt32Field()); } public void testGettersAndSetters() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); TestMap message = builder.build(); assertMapValuesCleared(message); builder = message.toBuilder(); setMapValuesUsingAccessors(builder); message = builder.build(); assertMapValuesSet(message); builder = message.toBuilder(); updateMapValuesUsingAccessors(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(); setMapValuesUsingAccessors(sourceBuilder); TestMap source = sourceBuilder.build(); assertMapValuesSet(source); TestMap.Builder destination = TestMap.newBuilder(); copyMapValues(source, destination); assertMapValuesSet(destination.build()); } public void testPutAllForUnknownEnumValues() throws Exception { TestMap source = TestMap.newBuilder() .putAllInt32ToEnumFieldValue(newMap( 0, 0, 1, 1, 2, 1000)) // unknown value. .build(); TestMap destination = TestMap.newBuilder() .putAllInt32ToEnumFieldValue(source.getInt32ToEnumFieldValue()) .build(); assertEquals(0, destination.getInt32ToEnumFieldValue().get(0).intValue()); assertEquals(1, destination.getInt32ToEnumFieldValue().get(1).intValue()); assertEquals(1000, destination.getInt32ToEnumFieldValue().get(2).intValue()); assertEquals(3, destination.getInt32ToEnumFieldCount()); } public void testPutForUnknownEnumValues() throws Exception { TestMap.Builder builder = TestMap.newBuilder() .putInt32ToEnumFieldValue(0, 0) .putInt32ToEnumFieldValue(1, 1); try { builder.putInt32ToEnumFieldValue(2, 1000); // unknown value. fail(); } catch (IllegalArgumentException e) { // expected } TestMap message = builder.build(); assertEquals(0, message.getInt32ToEnumFieldValueOrThrow(0)); assertEquals(1, message.getInt32ToEnumFieldValueOrThrow(1)); assertEquals(2, message.getInt32ToEnumFieldCount()); } 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(); setMapValuesUsingAccessors(builder); TestMap message = builder.build(); assertEquals(message.getSerializedSize(), message.toByteString().size()); message = TestMap.parser().parseFrom(message.toByteString()); assertMapValuesSet(message); builder = message.toBuilder(); updateMapValuesUsingAccessors(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(map.getInt32ToInt32FieldOrDefault(5, -1), 0); 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(map.getInt32ToEnumFieldOrDefault(0, null), TestMap.EnumValue.FOO); 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(map.getStringToInt32FieldOrDefault(stringKey, -1), 0); } public void testMergeFrom() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(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. // Regression test for b/18549190: if a map is a subset of the other map, // equals() should return false. b2.removeInt32ToInt32Field(1); m2 = b2.build(); assertFalse(m1.equals(m2)); assertFalse(m2.equals(m1)); } public void testNestedBuilderOnChangeEventPropagation() { TestOnChangeEventPropagation.Builder parent = TestOnChangeEventPropagation.newBuilder(); parent.getOptionalMessageBuilder().putInt32ToInt32Field(1, 2); TestOnChangeEventPropagation message = parent.build(); assertEquals(2, message.getOptionalMessage().getInt32ToInt32Field().get(1).intValue()); // Make a change using nested builder. parent.getOptionalMessageBuilder().putInt32ToInt32Field(1, 3); // Should be able to observe the change. message = parent.build(); assertEquals(3, message.getOptionalMessage().getInt32ToInt32Field().get(1).intValue()); // Make another change using mergeFrom() TestMap.Builder other = TestMap.newBuilder().putInt32ToInt32Field(1, 4); parent.getOptionalMessageBuilder().mergeFrom(other.build()); // Should be able to observe the change. message = parent.build(); assertEquals(4, message.getOptionalMessage().getInt32ToInt32Field().get(1).intValue()); // Make yet another change by clearing the nested builder. parent.getOptionalMessageBuilder().clear(); // Should be able to observe the change. message = parent.build(); assertEquals(0, message.getOptionalMessage().getInt32ToInt32Field().size()); } public void testNestedBuilderOnChangeEventPropagationReflection() { FieldDescriptor intMapField = f("int32_to_int32_field"); // Create an outer message builder with nested builder. TestOnChangeEventPropagation.Builder parentBuilder = TestOnChangeEventPropagation.newBuilder(); TestMap.Builder testMapBuilder = parentBuilder.getOptionalMessageBuilder(); // Create a map entry message. TestMap.Builder entryBuilder = TestMap.newBuilder().putInt32ToInt32Field(1, 1); // Put the entry into the nested builder. testMapBuilder.addRepeatedField( intMapField, entryBuilder.getRepeatedField(intMapField, 0)); // Should be able to observe the change. TestOnChangeEventPropagation message = parentBuilder.build(); assertEquals(1, message.getOptionalMessage().getInt32ToInt32Field().size()); // Change the entry value. entryBuilder.putInt32ToInt32Field(1, 4); testMapBuilder = parentBuilder.getOptionalMessageBuilder(); testMapBuilder.setRepeatedField( intMapField, 0, entryBuilder.getRepeatedField(intMapField, 0)); // Should be able to observe the change. message = parentBuilder.build(); assertEquals(4, message.getOptionalMessage().getInt32ToInt32Field().get(1).intValue()); // Clear the nested builder. testMapBuilder = parentBuilder.getOptionalMessageBuilder(); testMapBuilder.clearField(intMapField); // Should be able to observe the change. message = parentBuilder.build(); assertEquals(0, message.getOptionalMessage().getInt32ToInt32Field().size()); } // The following methods are used to test reflection API. private static FieldDescriptor f(String name) { return TestMap.getDescriptor().findFieldByName(name); } private static Object getFieldValue(Message mapEntry, String name) { FieldDescriptor field = mapEntry.getDescriptorForType().findFieldByName(name); return mapEntry.getField(field); } private static Message.Builder setFieldValue( Message.Builder mapEntry, String name, Object value) { FieldDescriptor field = mapEntry.getDescriptorForType().findFieldByName(name); mapEntry.setField(field, value); return mapEntry; } private static void assertHasMapValues(Message message, String name, Map values) { FieldDescriptor field = f(name); for (Object entry : (List) message.getField(field)) { Message mapEntry = (Message) entry; Object key = getFieldValue(mapEntry, "key"); Object value = getFieldValue(mapEntry, "value"); assertTrue(values.containsKey(key)); assertEquals(value, values.get(key)); } assertEquals(values.size(), message.getRepeatedFieldCount(field)); for (int i = 0; i < message.getRepeatedFieldCount(field); i++) { Message mapEntry = (Message) message.getRepeatedField(field, i); Object key = getFieldValue(mapEntry, "key"); Object value = getFieldValue(mapEntry, "value"); assertTrue(values.containsKey(key)); assertEquals(value, values.get(key)); } } private static Message newMapEntry(Message.Builder builder, String name, KeyType key, ValueType value) { FieldDescriptor field = builder.getDescriptorForType().findFieldByName(name); Message.Builder entryBuilder = builder.newBuilderForField(field); FieldDescriptor keyField = entryBuilder.getDescriptorForType().findFieldByName("key"); FieldDescriptor valueField = entryBuilder.getDescriptorForType().findFieldByName("value"); entryBuilder.setField(keyField, key); entryBuilder.setField(valueField, value); return entryBuilder.build(); } private static void setMapValues(Message.Builder builder, String name, Map values) { List entryList = new ArrayList(); for (Map.Entry entry : values.entrySet()) { entryList.add(newMapEntry(builder, name, entry.getKey(), entry.getValue())); } FieldDescriptor field = builder.getDescriptorForType().findFieldByName(name); builder.setField(field, entryList); } private static Map mapForValues( KeyType key1, ValueType value1, KeyType key2, ValueType value2) { Map map = new HashMap(); map.put(key1, value1); map.put(key2, value2); return map; } public void testReflectionApi() throws Exception { // In reflection API, map fields are just repeated message fields. TestMap.Builder builder = TestMap.newBuilder() .putInt32ToInt32Field(1, 2) .putInt32ToInt32Field(3, 4) .putInt32ToMessageField(11, MessageValue.newBuilder().setValue(22).build()) .putInt32ToMessageField(33, MessageValue.newBuilder().setValue(44).build()); TestMap message = builder.build(); // Test getField(), getRepeatedFieldCount(), getRepeatedField(). assertHasMapValues(message, "int32_to_int32_field", mapForValues(1, 2, 3, 4)); assertHasMapValues(message, "int32_to_message_field", mapForValues( 11, MessageValue.newBuilder().setValue(22).build(), 33, MessageValue.newBuilder().setValue(44).build())); // Test clearField() builder.clearField(f("int32_to_int32_field")); builder.clearField(f("int32_to_message_field")); message = builder.build(); assertEquals(0, message.getInt32ToInt32Field().size()); assertEquals(0, message.getInt32ToMessageField().size()); // Test setField() setMapValues(builder, "int32_to_int32_field", mapForValues(11, 22, 33, 44)); setMapValues(builder, "int32_to_message_field", mapForValues( 111, MessageValue.newBuilder().setValue(222).build(), 333, MessageValue.newBuilder().setValue(444).build())); message = builder.build(); assertEquals(22, message.getInt32ToInt32Field().get(11).intValue()); assertEquals(44, message.getInt32ToInt32Field().get(33).intValue()); assertEquals(222, message.getInt32ToMessageField().get(111).getValue()); assertEquals(444, message.getInt32ToMessageField().get(333).getValue()); // Test addRepeatedField builder.addRepeatedField(f("int32_to_int32_field"), newMapEntry(builder, "int32_to_int32_field", 55, 66)); builder.addRepeatedField(f("int32_to_message_field"), newMapEntry(builder, "int32_to_message_field", 555, MessageValue.newBuilder().setValue(666).build())); message = builder.build(); assertEquals(66, message.getInt32ToInt32Field().get(55).intValue()); assertEquals(666, message.getInt32ToMessageField().get(555).getValue()); // Test addRepeatedField (overriding existing values) builder.addRepeatedField(f("int32_to_int32_field"), newMapEntry(builder, "int32_to_int32_field", 55, 55)); builder.addRepeatedField(f("int32_to_message_field"), newMapEntry(builder, "int32_to_message_field", 555, MessageValue.newBuilder().setValue(555).build())); message = builder.build(); assertEquals(55, message.getInt32ToInt32Field().get(55).intValue()); assertEquals(555, message.getInt32ToMessageField().get(555).getValue()); // Test setRepeatedField for (int i = 0; i < builder.getRepeatedFieldCount(f("int32_to_int32_field")); i++) { Message mapEntry = (Message) builder.getRepeatedField(f("int32_to_int32_field"), i); int oldKey = ((Integer) getFieldValue(mapEntry, "key")).intValue(); int oldValue = ((Integer) getFieldValue(mapEntry, "value")).intValue(); // Swap key with value for each entry. Message.Builder mapEntryBuilder = mapEntry.toBuilder(); setFieldValue(mapEntryBuilder, "key", oldValue); setFieldValue(mapEntryBuilder, "value", oldKey); builder.setRepeatedField(f("int32_to_int32_field"), i, mapEntryBuilder.build()); } message = builder.build(); assertEquals(11, message.getInt32ToInt32Field().get(22).intValue()); assertEquals(33, message.getInt32ToInt32Field().get(44).intValue()); assertEquals(55, message.getInt32ToInt32Field().get(55).intValue()); } public void testTextFormat() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(builder); TestMap message = builder.build(); String textData = TextFormat.printToString(message); builder = TestMap.newBuilder(); TextFormat.merge(textData, builder); message = builder.build(); assertMapValuesSet(message); } public void testDynamicMessage() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(builder); TestMap message = builder.build(); Message dynamicDefaultInstance = DynamicMessage.getDefaultInstance(TestMap.getDescriptor()); Message dynamicMessage = dynamicDefaultInstance .newBuilderForType().mergeFrom(message.toByteString()).build(); assertEquals(message, dynamicMessage); assertEquals(message.hashCode(), dynamicMessage.hashCode()); } public void testReflectionEqualsAndHashCode() throws Exception { // Test that generated equals() and hashCode() will disregard the order // of map entries when comparing/hashing map fields. // We use DynamicMessage to test reflection based equals()/hashCode(). Message dynamicDefaultInstance = DynamicMessage.getDefaultInstance(TestMap.getDescriptor()); FieldDescriptor field = f("int32_to_int32_field"); Message.Builder b1 = dynamicDefaultInstance.newBuilderForType(); b1.addRepeatedField(field, newMapEntry(b1, "int32_to_int32_field", 1, 2)); b1.addRepeatedField(field, newMapEntry(b1, "int32_to_int32_field", 3, 4)); b1.addRepeatedField(field, newMapEntry(b1, "int32_to_int32_field", 5, 6)); Message m1 = b1.build(); Message.Builder b2 = dynamicDefaultInstance.newBuilderForType(); b2.addRepeatedField(field, newMapEntry(b2, "int32_to_int32_field", 5, 6)); b2.addRepeatedField(field, newMapEntry(b2, "int32_to_int32_field", 1, 2)); b2.addRepeatedField(field, newMapEntry(b2, "int32_to_int32_field", 3, 4)); Message m2 = b2.build(); assertEquals(m1, m2); assertEquals(m1.hashCode(), m2.hashCode()); // Make sure we did compare map fields. b2.setRepeatedField(field, 0, newMapEntry(b1, "int32_to_int32_field", 0, 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 { TestMap.Builder builder = TestMap.newBuilder() .putAllInt32ToEnumFieldValue(newMap( 0, 0, 1, 1, 2, 1000)); // unknown value. TestMap message = builder.build(); assertEquals(TestMap.EnumValue.FOO, message.getInt32ToEnumField().get(0)); assertEquals(TestMap.EnumValue.BAR, message.getInt32ToEnumField().get(1)); assertEquals(TestMap.EnumValue.UNRECOGNIZED, message.getInt32ToEnumField().get(2)); assertEquals(1000, message.getInt32ToEnumFieldValue().get(2).intValue()); // Unknown enum values should be preserved after: // 1. Serialization and parsing. // 2. toBuild(). // 3. mergeFrom(). message = TestMap.parseFrom(message.toByteString()); assertEquals(1000, message.getInt32ToEnumFieldValue().get(2).intValue()); builder = message.toBuilder(); assertEquals(1000, builder.getInt32ToEnumFieldValue().get(2).intValue()); builder = TestMap.newBuilder().mergeFrom(message); assertEquals(1000, builder.getInt32ToEnumFieldValue().get(2).intValue()); // hashCode()/equals() should take unknown enum values into account. builder.putAllInt32ToEnumFieldValue(newMap(2, 1001)); TestMap message2 = builder.build(); assertFalse(message.hashCode() == message2.hashCode()); assertFalse(message.equals(message2)); // Unknown values will be converted to UNRECOGNIZED so the resulted enum map // should be the same. assertTrue(message.getInt32ToEnumField().equals(message2.getInt32ToEnumField())); } public void testUnknownEnumValuesInReflectionApi() throws Exception { Descriptor descriptor = TestMap.getDescriptor(); EnumDescriptor enumDescriptor = TestMap.EnumValue.getDescriptor(); FieldDescriptor field = descriptor.findFieldByName("int32_to_enum_field"); Map data = newMap( 0, 0, 1, 1, 2, 1000); // unknown value TestMap.Builder builder = TestMap.newBuilder() .putAllInt32ToEnumFieldValue(data); // Try to read unknown enum values using reflection API. for (int i = 0; i < builder.getRepeatedFieldCount(field); i++) { Message mapEntry = (Message) builder.getRepeatedField(field, i); int key = ((Integer) getFieldValue(mapEntry, "key")).intValue(); int value = ((EnumValueDescriptor) getFieldValue(mapEntry, "value")).getNumber(); assertEquals(data.get(key).intValue(), value); Message.Builder mapEntryBuilder = mapEntry.toBuilder(); // Increase the value by 1. setFieldValue(mapEntryBuilder, "value", enumDescriptor.findValueByNumberCreatingIfUnknown(value + 1)); builder.setRepeatedField(field, i, mapEntryBuilder.build()); } // Verify that enum values have been successfully updated. TestMap message = builder.build(); for (Map.Entry entry : message.getInt32ToEnumFieldValue().entrySet()) { assertEquals(data.get(entry.getKey()) + 1, entry.getValue().intValue()); } } public void testIterationOrder() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(builder); TestMap message = builder.build(); assertEquals(Arrays.asList("1", "2", "3"), new ArrayList(message.getStringToInt32Field().keySet())); } public void testGetMap() { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(builder); TestMap message = builder.build(); assertEquals( message.getStringToInt32Field(), message.getStringToInt32FieldMap()); assertEquals( message.getInt32ToBytesField(), message.getInt32ToBytesFieldMap()); assertEquals( message.getInt32ToEnumField(), message.getInt32ToEnumFieldMap()); assertEquals( message.getInt32ToEnumFieldValue(), message.getInt32ToEnumFieldValueMap()); assertEquals( message.getInt32ToMessageField(), message.getInt32ToMessageFieldMap()); } public void testContains() { TestMap.Builder builder = TestMap.newBuilder(); setMapValuesUsingAccessors(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); setMapValuesUsingAccessors(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); setMapValuesUsingAccessors(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( TestMap.EnumValue.BAR.getNumber(), (int) testMapOrBuilder.getInt32ToEnumFieldValueOrDefault(2, -1)); assertEquals(-1, testMapOrBuilder.getInt32ToEnumFieldValueOrDefault(-1000, -1)); 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); setMapValuesUsingAccessors(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( TestMap.EnumValue.BAR.getNumber(), testMapOrBuilder.getInt32ToEnumFieldValueOrThrow(2)); try { testMapOrBuilder.getInt32ToEnumFieldValueOrThrow(-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.putInt32ToEnumFieldValue(1, TestMap.EnumValue.BAR.getNumber()); assertEquals( TestMap.EnumValue.BAR.getNumber(), builder.getInt32ToEnumFieldValueOrThrow(1)); try { builder.putInt32ToEnumFieldValue(1, -1); fail(); } catch (IllegalArgumentException 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(); setMapValuesUsingAccessors(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 } } public void testReservedWordsFieldNames() { ReservedAsMapField.newBuilder().build(); ReservedAsMapFieldWithEnumValue.newBuilder().build(); } public void testDeterministicSerialziation() throws Exception { TestMap.Builder builder = TestMap.newBuilder(); // int32->int32 builder.putInt32ToInt32Field(5, 1); builder.putInt32ToInt32Field(1, 1); builder.putInt32ToInt32Field(4, 1); builder.putInt32ToInt32Field(-2, 1); builder.putInt32ToInt32Field(0, 1); // uint32->int32 builder.putUint32ToInt32Field(5, 1); builder.putUint32ToInt32Field(1, 1); builder.putUint32ToInt32Field(4, 1); builder.putUint32ToInt32Field(-2, 1); builder.putUint32ToInt32Field(0, 1); // int64->int32 builder.putInt64ToInt32Field(5L, 1); builder.putInt64ToInt32Field(1L, 1); builder.putInt64ToInt32Field(4L, 1); builder.putInt64ToInt32Field(-2L, 1); builder.putInt64ToInt32Field(0L, 1); // string->int32 builder.putStringToInt32Field("baz", 1); builder.putStringToInt32Field("foo", 1); builder.putStringToInt32Field("bar", 1); builder.putStringToInt32Field("", 1); builder.putStringToInt32Field("hello", 1); builder.putStringToInt32Field("world", 1); TestMap message = builder.build(); byte[] serialized = new byte[message.getSerializedSize()]; CodedOutputStream output = CodedOutputStream.newInstance(serialized); output.useDeterministicSerialization(); message.writeTo(output); output.flush(); CodedInputStream input = CodedInputStream.newInstance(serialized); List int32Keys = new ArrayList(); List uint32Keys = new ArrayList(); List int64Keys = new ArrayList(); List stringKeys = new ArrayList(); int tag; while (true) { tag = input.readTag(); if (tag == 0) { break; } int length = input.readRawVarint32(); int oldLimit = input.pushLimit(length); switch (WireFormat.getTagFieldNumber(tag)) { case TestMap.STRING_TO_INT32_FIELD_FIELD_NUMBER: stringKeys.add(readMapStringKey(input)); break; case TestMap.INT32_TO_INT32_FIELD_FIELD_NUMBER: int32Keys.add(readMapIntegerKey(input)); break; case TestMap.UINT32_TO_INT32_FIELD_FIELD_NUMBER: uint32Keys.add(readMapIntegerKey(input)); break; case TestMap.INT64_TO_INT32_FIELD_FIELD_NUMBER: int64Keys.add(readMapLongKey(input)); break; default: fail("Unexpected fields."); } input.popLimit(oldLimit); } assertEquals( Arrays.asList(-2, 0, 1, 4, 5), int32Keys); assertEquals( Arrays.asList(-2, 0, 1, 4, 5), uint32Keys); assertEquals( Arrays.asList(-2L, 0L, 1L, 4L, 5L), int64Keys); assertEquals( Arrays.asList("", "bar", "baz", "foo", "hello", "world"), stringKeys); } public void testInitFromPartialDynamicMessage() { FieldDescriptor fieldDescriptor = TestMap.getDescriptor().findFieldByNumber(TestMap.INT32_TO_MESSAGE_FIELD_FIELD_NUMBER); Descriptor mapEntryType = fieldDescriptor.getMessageType(); FieldDescriptor keyField = mapEntryType.findFieldByNumber(1); FieldDescriptor valueField = mapEntryType.findFieldByNumber(2); DynamicMessage dynamicMessage = DynamicMessage.newBuilder(TestMap.getDescriptor()) .addRepeatedField( fieldDescriptor, DynamicMessage.newBuilder(mapEntryType) .setField(keyField, 10) .setField(valueField, TestMap.MessageValue.newBuilder().setValue(10).build()) .build()) .build(); TestMap message = TestMap.newBuilder().mergeFrom(dynamicMessage).build(); assertEquals( TestMap.MessageValue.newBuilder().setValue(10).build(), message.getInt32ToMessageFieldMap().get(10)); } public void testInitFromFullyDynamicMessage() { FieldDescriptor fieldDescriptor = TestMap.getDescriptor().findFieldByNumber(TestMap.INT32_TO_MESSAGE_FIELD_FIELD_NUMBER); Descriptor mapEntryType = fieldDescriptor.getMessageType(); FieldDescriptor keyField = mapEntryType.findFieldByNumber(1); FieldDescriptor valueField = mapEntryType.findFieldByNumber(2); DynamicMessage dynamicMessage = DynamicMessage.newBuilder(TestMap.getDescriptor()) .addRepeatedField( fieldDescriptor, DynamicMessage.newBuilder(mapEntryType) .setField(keyField, 10) .setField( valueField, DynamicMessage.newBuilder(TestMap.MessageValue.getDescriptor()) .setField( TestMap.MessageValue.getDescriptor().findFieldByName("value"), 10) .build()) .build()) .build(); TestMap message = TestMap.newBuilder().mergeFrom(dynamicMessage).build(); assertEquals( TestMap.MessageValue.newBuilder().setValue(10).build(), message.getInt32ToMessageFieldMap().get(10)); } private int readMapIntegerKey(CodedInputStream input) throws IOException { int tag = input.readTag(); assertEquals(WireFormat.makeTag(1, WireFormat.WIRETYPE_VARINT), tag); int ret = input.readInt32(); // skip the value field. input.skipField(input.readTag()); assertTrue(input.isAtEnd()); return ret; } private long readMapLongKey(CodedInputStream input) throws IOException { int tag = input.readTag(); assertEquals(WireFormat.makeTag(1, WireFormat.WIRETYPE_VARINT), tag); long ret = input.readInt64(); // skip the value field. input.skipField(input.readTag()); assertTrue(input.isAtEnd()); return ret; } private String readMapStringKey(CodedInputStream input) throws IOException { int tag = input.readTag(); assertEquals(WireFormat.makeTag(1, WireFormat.WIRETYPE_LENGTH_DELIMITED), tag); String ret = input.readString(); // skip the value field. input.skipField(input.readTag()); assertTrue(input.isAtEnd()); return ret; } private static Map newMap(K key1, V value1) { Map map = new HashMap(); map.put(key1, value1); return map; } private static Map newMap(K key1, V value1, K key2, V value2) { Map map = new HashMap(); map.put(key1, value1); map.put(key2, value2); return map; } private static Map newMap(K key1, V value1, K key2, V value2, K key3, V value3) { Map map = new HashMap(); map.put(key1, value1); map.put(key2, value2); map.put(key3, value3); return map; } }