OLD | NEW |
(Empty) | |
| 1 // Protocol Buffers - Google's data interchange format |
| 2 // Copyright 2008 Google Inc. All rights reserved. |
| 3 // http://code.google.com/p/protobuf/ |
| 4 // |
| 5 // Redistribution and use in source and binary forms, with or without |
| 6 // modification, are permitted provided that the following conditions are |
| 7 // met: |
| 8 // |
| 9 // * Redistributions of source code must retain the above copyright |
| 10 // notice, this list of conditions and the following disclaimer. |
| 11 // * Redistributions in binary form must reproduce the above |
| 12 // copyright notice, this list of conditions and the following disclaimer |
| 13 // in the documentation and/or other materials provided with the |
| 14 // distribution. |
| 15 // * Neither the name of Google Inc. nor the names of its |
| 16 // contributors may be used to endorse or promote products derived from |
| 17 // this software without specific prior written permission. |
| 18 // |
| 19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 30 |
| 31 package com.google.protobuf; |
| 32 |
| 33 import protobuf_unittest.UnittestProto.TestAllTypes; |
| 34 import protobuf_unittest.UnittestProto.TestAllTypesOrBuilder; |
| 35 |
| 36 import junit.framework.TestCase; |
| 37 |
| 38 import java.util.Collections; |
| 39 import java.util.List; |
| 40 |
| 41 /** |
| 42 * Tests for {@link RepeatedFieldBuilder}. This tests basic functionality. |
| 43 * More extensive testing is provided via other tests that exercise the |
| 44 * builder. |
| 45 * |
| 46 * @author jonp@google.com (Jon Perlow) |
| 47 */ |
| 48 public class RepeatedFieldBuilderTest extends TestCase { |
| 49 |
| 50 public void testBasicUse() { |
| 51 TestUtil.MockBuilderParent mockParent = new TestUtil.MockBuilderParent(); |
| 52 RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 53 TestAllTypesOrBuilder> builder = newRepeatedFieldBuilder(mockParent); |
| 54 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(0).build()); |
| 55 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(1).build()); |
| 56 assertEquals(0, builder.getMessage(0).getOptionalInt32()); |
| 57 assertEquals(1, builder.getMessage(1).getOptionalInt32()); |
| 58 |
| 59 List<TestAllTypes> list = builder.build(); |
| 60 assertEquals(2, list.size()); |
| 61 assertEquals(0, list.get(0).getOptionalInt32()); |
| 62 assertEquals(1, list.get(1).getOptionalInt32()); |
| 63 assertIsUnmodifiable(list); |
| 64 |
| 65 // Make sure it doesn't change. |
| 66 List<TestAllTypes> list2 = builder.build(); |
| 67 assertSame(list, list2); |
| 68 assertEquals(0, mockParent.getInvalidationCount()); |
| 69 } |
| 70 |
| 71 public void testGoingBackAndForth() { |
| 72 TestUtil.MockBuilderParent mockParent = new TestUtil.MockBuilderParent(); |
| 73 RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 74 TestAllTypesOrBuilder> builder = newRepeatedFieldBuilder(mockParent); |
| 75 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(0).build()); |
| 76 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(1).build()); |
| 77 assertEquals(0, builder.getMessage(0).getOptionalInt32()); |
| 78 assertEquals(1, builder.getMessage(1).getOptionalInt32()); |
| 79 |
| 80 // Convert to list |
| 81 List<TestAllTypes> list = builder.build(); |
| 82 assertEquals(2, list.size()); |
| 83 assertEquals(0, list.get(0).getOptionalInt32()); |
| 84 assertEquals(1, list.get(1).getOptionalInt32()); |
| 85 assertIsUnmodifiable(list); |
| 86 |
| 87 // Update 0th item |
| 88 assertEquals(0, mockParent.getInvalidationCount()); |
| 89 builder.getBuilder(0).setOptionalString("foo"); |
| 90 assertEquals(1, mockParent.getInvalidationCount()); |
| 91 list = builder.build(); |
| 92 assertEquals(2, list.size()); |
| 93 assertEquals(0, list.get(0).getOptionalInt32()); |
| 94 assertEquals("foo", list.get(0).getOptionalString()); |
| 95 assertEquals(1, list.get(1).getOptionalInt32()); |
| 96 assertIsUnmodifiable(list); |
| 97 assertEquals(1, mockParent.getInvalidationCount()); |
| 98 } |
| 99 |
| 100 public void testVariousMethods() { |
| 101 TestUtil.MockBuilderParent mockParent = new TestUtil.MockBuilderParent(); |
| 102 RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 103 TestAllTypesOrBuilder> builder = newRepeatedFieldBuilder(mockParent); |
| 104 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(1).build()); |
| 105 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(2).build()); |
| 106 builder.addBuilder(0, TestAllTypes.getDefaultInstance()) |
| 107 .setOptionalInt32(0); |
| 108 builder.addBuilder(TestAllTypes.getDefaultInstance()).setOptionalInt32(3); |
| 109 |
| 110 assertEquals(0, builder.getMessage(0).getOptionalInt32()); |
| 111 assertEquals(1, builder.getMessage(1).getOptionalInt32()); |
| 112 assertEquals(2, builder.getMessage(2).getOptionalInt32()); |
| 113 assertEquals(3, builder.getMessage(3).getOptionalInt32()); |
| 114 |
| 115 assertEquals(0, mockParent.getInvalidationCount()); |
| 116 List<TestAllTypes> messages = builder.build(); |
| 117 assertEquals(4, messages.size()); |
| 118 assertSame(messages, builder.build()); // expect same list |
| 119 |
| 120 // Remove a message. |
| 121 builder.remove(2); |
| 122 assertEquals(1, mockParent.getInvalidationCount()); |
| 123 assertEquals(3, builder.getCount()); |
| 124 assertEquals(0, builder.getMessage(0).getOptionalInt32()); |
| 125 assertEquals(1, builder.getMessage(1).getOptionalInt32()); |
| 126 assertEquals(3, builder.getMessage(2).getOptionalInt32()); |
| 127 |
| 128 // Remove a builder. |
| 129 builder.remove(0); |
| 130 assertEquals(1, mockParent.getInvalidationCount()); |
| 131 assertEquals(2, builder.getCount()); |
| 132 assertEquals(1, builder.getMessage(0).getOptionalInt32()); |
| 133 assertEquals(3, builder.getMessage(1).getOptionalInt32()); |
| 134 |
| 135 // Test clear. |
| 136 builder.clear(); |
| 137 assertEquals(1, mockParent.getInvalidationCount()); |
| 138 assertEquals(0, builder.getCount()); |
| 139 assertTrue(builder.isEmpty()); |
| 140 } |
| 141 |
| 142 public void testLists() { |
| 143 TestUtil.MockBuilderParent mockParent = new TestUtil.MockBuilderParent(); |
| 144 RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 145 TestAllTypesOrBuilder> builder = newRepeatedFieldBuilder(mockParent); |
| 146 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(1).build()); |
| 147 builder.addMessage(0, |
| 148 TestAllTypes.newBuilder().setOptionalInt32(0).build()); |
| 149 assertEquals(0, builder.getMessage(0).getOptionalInt32()); |
| 150 assertEquals(1, builder.getMessage(1).getOptionalInt32()); |
| 151 |
| 152 // Use list of builders. |
| 153 List<TestAllTypes.Builder> builders = builder.getBuilderList(); |
| 154 assertEquals(0, builders.get(0).getOptionalInt32()); |
| 155 assertEquals(1, builders.get(1).getOptionalInt32()); |
| 156 builders.get(0).setOptionalInt32(10); |
| 157 builders.get(1).setOptionalInt32(11); |
| 158 |
| 159 // Use list of protos |
| 160 List<TestAllTypes> protos = builder.getMessageList(); |
| 161 assertEquals(10, protos.get(0).getOptionalInt32()); |
| 162 assertEquals(11, protos.get(1).getOptionalInt32()); |
| 163 |
| 164 // Add an item to the builders and verify it's updated in both |
| 165 builder.addMessage(TestAllTypes.newBuilder().setOptionalInt32(12).build()); |
| 166 assertEquals(3, builders.size()); |
| 167 assertEquals(3, protos.size()); |
| 168 } |
| 169 |
| 170 private void assertIsUnmodifiable(List<?> list) { |
| 171 if (list == Collections.emptyList()) { |
| 172 // OKAY -- Need to check this b/c EmptyList allows you to call clear. |
| 173 } else { |
| 174 try { |
| 175 list.clear(); |
| 176 fail("List wasn't immutable"); |
| 177 } catch (UnsupportedOperationException e) { |
| 178 // good |
| 179 } |
| 180 } |
| 181 } |
| 182 |
| 183 private RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 184 TestAllTypesOrBuilder> |
| 185 newRepeatedFieldBuilder(GeneratedMessage.BuilderParent parent) { |
| 186 return new RepeatedFieldBuilder<TestAllTypes, TestAllTypes.Builder, |
| 187 TestAllTypesOrBuilder>(Collections.<TestAllTypes>emptyList(), false, |
| 188 parent, false); |
| 189 } |
| 190 } |
OLD | NEW |