| // 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.TestMap; |
| import map_lite_test.MapForProto2TestProto.TestMap.MessageValue; |
| import map_lite_test.MapForProto2TestProto.TestUnknownEnumValue; |
| |
| import junit.framework.TestCase; |
| |
| /** |
| * Unit tests for map fields. |
| */ |
| public class MapForProto2LiteTest extends TestCase { |
| private void setMapValues(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 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.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 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(TestMap message) { |
| assertEquals(0, message.getInt32ToInt32Field().size()); |
| assertEquals(0, message.getInt32ToStringField().size()); |
| assertEquals(0, message.getInt32ToBytesField().size()); |
| assertEquals(0, message.getInt32ToEnumField().size()); |
| assertEquals(0, message.getInt32ToMessageField().size()); |
| assertEquals(0, message.getStringToInt32Field().size()); |
| } |
| |
| 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(); |
| message = builder.build(); |
| assertMapValuesCleared(message); |
| } |
| |
| 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); |
| } |
| |
| 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(); |
| b1.getMutableInt32ToInt32Field().put(1, 2); |
| b1.getMutableInt32ToInt32Field().put(3, 4); |
| b1.getMutableInt32ToInt32Field().put(5, 6); |
| TestMap m1 = b1.build(); |
| |
| TestMap.Builder b2 = TestMap.newBuilder(); |
| b2.getMutableInt32ToInt32Field().put(5, 6); |
| b2.getMutableInt32ToInt32Field().put(1, 2); |
| b2.getMutableInt32ToInt32Field().put(3, 4); |
| TestMap m2 = b2.build(); |
| |
| assertEquals(m1, m2); |
| assertEquals(m1.hashCode(), m2.hashCode()); |
| |
| // Make sure we did compare map fields. |
| b2.getMutableInt32ToInt32Field().put(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(); |
| builder.getMutableInt32ToInt32Field().put(1, 1); |
| builder.getMutableInt32ToInt32Field().put(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()); |
| } |
| |
| } |