| OLD | NEW |
| (Empty) |
| 1 // Protocol Buffers - Google's data interchange format | |
| 2 // Copyright 2008 Google Inc. All rights reserved. | |
| 3 // https://developers.google.com/protocol-buffers/ | |
| 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 #import "GPBTestUtilities.h" | |
| 32 | |
| 33 #import "GPBUnknownField_PackagePrivate.h" | |
| 34 #import "GPBUnknownFieldSet_PackagePrivate.h" | |
| 35 #import "google/protobuf/Unittest.pbobjc.h" | |
| 36 | |
| 37 @interface GPBUnknownFieldSet (GPBUnknownFieldSetTest) | |
| 38 - (void)getTags:(int32_t*)tags; | |
| 39 @end | |
| 40 | |
| 41 @interface UnknownFieldSetTest : GPBTestCase { | |
| 42 @private | |
| 43 TestAllTypes* allFields_; | |
| 44 NSData* allFieldsData_; | |
| 45 | |
| 46 // An empty message that has been parsed from allFieldsData. So, it has | |
| 47 // unknown fields of every type. | |
| 48 TestEmptyMessage* emptyMessage_; | |
| 49 GPBUnknownFieldSet* unknownFields_; | |
| 50 } | |
| 51 | |
| 52 @end | |
| 53 | |
| 54 @implementation UnknownFieldSetTest | |
| 55 | |
| 56 - (void)setUp { | |
| 57 allFields_ = [self allSetRepeatedCount:kGPBDefaultRepeatCount]; | |
| 58 allFieldsData_ = [allFields_ data]; | |
| 59 emptyMessage_ = [TestEmptyMessage parseFromData:allFieldsData_ error:NULL]; | |
| 60 unknownFields_ = emptyMessage_.unknownFields; | |
| 61 } | |
| 62 | |
| 63 - (GPBUnknownField *)getField:(int32_t)number { | |
| 64 return [unknownFields_ getField:number]; | |
| 65 } | |
| 66 | |
| 67 // Constructs a protocol buffer which contains fields with all the same | |
| 68 // numbers as allFieldsData except that each field is some other wire | |
| 69 // type. | |
| 70 - (NSData*)getBizarroData { | |
| 71 GPBUnknownFieldSet* bizarroFields = | |
| 72 [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 73 NSUInteger count = [unknownFields_ countOfFields]; | |
| 74 int32_t tags[count]; | |
| 75 [unknownFields_ getTags:tags]; | |
| 76 for (NSUInteger i = 0; i < count; ++i) { | |
| 77 int32_t tag = tags[i]; | |
| 78 GPBUnknownField* field = [unknownFields_ getField:tag]; | |
| 79 if (field.varintList.count == 0) { | |
| 80 // Original field is not a varint, so use a varint. | |
| 81 GPBUnknownField* varintField = | |
| 82 [[[GPBUnknownField alloc] initWithNumber:tag] autorelease]; | |
| 83 [varintField addVarint:1]; | |
| 84 [bizarroFields addField:varintField]; | |
| 85 } else { | |
| 86 // Original field *is* a varint, so use something else. | |
| 87 GPBUnknownField* fixed32Field = | |
| 88 [[[GPBUnknownField alloc] initWithNumber:tag] autorelease]; | |
| 89 [fixed32Field addFixed32:1]; | |
| 90 [bizarroFields addField:fixed32Field]; | |
| 91 } | |
| 92 } | |
| 93 | |
| 94 return [bizarroFields data]; | |
| 95 } | |
| 96 | |
| 97 - (void)testSerialize { | |
| 98 // Check that serializing the UnknownFieldSet produces the original data | |
| 99 // again. | |
| 100 NSData* data = [emptyMessage_ data]; | |
| 101 XCTAssertEqualObjects(allFieldsData_, data); | |
| 102 } | |
| 103 | |
| 104 - (void)testCopyFrom { | |
| 105 TestEmptyMessage* message = [TestEmptyMessage message]; | |
| 106 [message mergeFrom:emptyMessage_]; | |
| 107 | |
| 108 XCTAssertEqualObjects(emptyMessage_.data, message.data); | |
| 109 } | |
| 110 | |
| 111 - (void)testMergeFrom { | |
| 112 GPBUnknownFieldSet* set1 = [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 113 GPBUnknownField* field = [[[GPBUnknownField alloc] initWithNumber:2] autorelea
se]; | |
| 114 [field addVarint:2]; | |
| 115 [set1 addField:field]; | |
| 116 field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease]; | |
| 117 [field addVarint:4]; | |
| 118 [set1 addField:field]; | |
| 119 | |
| 120 GPBUnknownFieldSet* set2 = [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 121 field = [[[GPBUnknownField alloc] initWithNumber:1] autorelease]; | |
| 122 [field addVarint:1]; | |
| 123 [set2 addField:field]; | |
| 124 field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease]; | |
| 125 [field addVarint:3]; | |
| 126 [set2 addField:field]; | |
| 127 | |
| 128 GPBUnknownFieldSet* set3 = [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 129 field = [[[GPBUnknownField alloc] initWithNumber:1] autorelease]; | |
| 130 [field addVarint:1]; | |
| 131 [set3 addField:field]; | |
| 132 field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease]; | |
| 133 [field addVarint:4]; | |
| 134 [set3 addField:field]; | |
| 135 | |
| 136 GPBUnknownFieldSet* set4 = [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 137 field = [[[GPBUnknownField alloc] initWithNumber:2] autorelease]; | |
| 138 [field addVarint:2]; | |
| 139 [set4 addField:field]; | |
| 140 field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease]; | |
| 141 [field addVarint:3]; | |
| 142 [set4 addField:field]; | |
| 143 | |
| 144 TestEmptyMessage* source1 = [TestEmptyMessage message]; | |
| 145 [source1 setUnknownFields:set1]; | |
| 146 TestEmptyMessage* source2 = [TestEmptyMessage message]; | |
| 147 [source2 setUnknownFields:set2]; | |
| 148 TestEmptyMessage* source3 = [TestEmptyMessage message]; | |
| 149 [source3 setUnknownFields:set3]; | |
| 150 TestEmptyMessage* source4 = [TestEmptyMessage message]; | |
| 151 [source4 setUnknownFields:set4]; | |
| 152 | |
| 153 TestEmptyMessage* destination1 = [TestEmptyMessage message]; | |
| 154 [destination1 mergeFrom:source1]; | |
| 155 [destination1 mergeFrom:source2]; | |
| 156 | |
| 157 TestEmptyMessage* destination2 = [TestEmptyMessage message]; | |
| 158 [destination2 mergeFrom:source3]; | |
| 159 [destination2 mergeFrom:source4]; | |
| 160 | |
| 161 XCTAssertEqualObjects(destination1.data, destination2.data); | |
| 162 } | |
| 163 | |
| 164 - (void)testClearMessage { | |
| 165 TestEmptyMessage *message = [TestEmptyMessage message]; | |
| 166 [message mergeFrom:emptyMessage_]; | |
| 167 [message clear]; | |
| 168 XCTAssertEqual(message.serializedSize, (size_t)0); | |
| 169 } | |
| 170 | |
| 171 - (void)testParseKnownAndUnknown { | |
| 172 // Test mixing known and unknown fields when parsing. | |
| 173 GPBUnknownFieldSet *fields = [[unknownFields_ copy] autorelease]; | |
| 174 GPBUnknownField *field = | |
| 175 [[[GPBUnknownField alloc] initWithNumber:123456] autorelease]; | |
| 176 [field addVarint:654321]; | |
| 177 [fields addField:field]; | |
| 178 | |
| 179 NSData* data = fields.data; | |
| 180 TestAllTypes* destination = [TestAllTypes parseFromData:data error:NULL]; | |
| 181 | |
| 182 [self assertAllFieldsSet:destination repeatedCount:kGPBDefaultRepeatCount]; | |
| 183 XCTAssertEqual(destination.unknownFields.countOfFields, (NSUInteger)1); | |
| 184 | |
| 185 GPBUnknownField* field2 = [destination.unknownFields getField:123456]; | |
| 186 XCTAssertEqual(field2.varintList.count, (NSUInteger)1); | |
| 187 XCTAssertEqual(654321ULL, [field2.varintList valueAtIndex:0]); | |
| 188 } | |
| 189 | |
| 190 - (void)testWrongTypeTreatedAsUnknown { | |
| 191 // Test that fields of the wrong wire type are treated like unknown fields | |
| 192 // when parsing. | |
| 193 | |
| 194 NSData* bizarroData = [self getBizarroData]; | |
| 195 TestAllTypes* allTypesMessage = | |
| 196 [TestAllTypes parseFromData:bizarroData error:NULL]; | |
| 197 TestEmptyMessage* emptyMessage = | |
| 198 [TestEmptyMessage parseFromData:bizarroData error:NULL]; | |
| 199 | |
| 200 // All fields should have been interpreted as unknown, so the debug strings | |
| 201 // should be the same. | |
| 202 XCTAssertEqualObjects(emptyMessage.data, allTypesMessage.data); | |
| 203 } | |
| 204 | |
| 205 - (void)testUnknownExtensions { | |
| 206 // Make sure fields are properly parsed to the UnknownFieldSet even when | |
| 207 // they are declared as extension numbers. | |
| 208 | |
| 209 TestEmptyMessageWithExtensions* message = | |
| 210 [TestEmptyMessageWithExtensions parseFromData:allFieldsData_ error:NULL]; | |
| 211 | |
| 212 XCTAssertEqual(unknownFields_.countOfFields, | |
| 213 message.unknownFields.countOfFields); | |
| 214 XCTAssertEqualObjects(allFieldsData_, message.data); | |
| 215 } | |
| 216 | |
| 217 - (void)testWrongExtensionTypeTreatedAsUnknown { | |
| 218 // Test that fields of the wrong wire type are treated like unknown fields | |
| 219 // when parsing extensions. | |
| 220 | |
| 221 NSData* bizarroData = [self getBizarroData]; | |
| 222 TestAllExtensions* allExtensionsMessage = | |
| 223 [TestAllExtensions parseFromData:bizarroData error:NULL]; | |
| 224 TestEmptyMessage* emptyMessage = | |
| 225 [TestEmptyMessage parseFromData:bizarroData error:NULL]; | |
| 226 | |
| 227 // All fields should have been interpreted as unknown, so the debug strings | |
| 228 // should be the same. | |
| 229 XCTAssertEqualObjects(emptyMessage.data, allExtensionsMessage.data); | |
| 230 } | |
| 231 | |
| 232 - (void)testLargeVarint { | |
| 233 GPBUnknownFieldSet* fields = [[unknownFields_ copy] autorelease]; | |
| 234 GPBUnknownField* field = [[[GPBUnknownField alloc] initWithNumber:1] autorelea
se]; | |
| 235 [field addVarint:0x7FFFFFFFFFFFFFFFL]; | |
| 236 [fields addField:field]; | |
| 237 | |
| 238 NSData* data = [fields data]; | |
| 239 | |
| 240 GPBUnknownFieldSet* parsed = [[[GPBUnknownFieldSet alloc] init] autorelease]; | |
| 241 [parsed mergeFromData:data]; | |
| 242 GPBUnknownField* field2 = [parsed getField:1]; | |
| 243 XCTAssertEqual(field2.varintList.count, (NSUInteger)1); | |
| 244 XCTAssertEqual(0x7FFFFFFFFFFFFFFFULL, [field2.varintList valueAtIndex:0]); | |
| 245 } | |
| 246 | |
| 247 - (void)testMergingFields { | |
| 248 GPBUnknownField* field1 = [[[GPBUnknownField alloc] initWithNumber:1] autorele
ase]; | |
| 249 [field1 addVarint:1]; | |
| 250 [field1 addFixed32:2]; | |
| 251 [field1 addFixed64:3]; | |
| 252 [field1 addLengthDelimited:[NSData dataWithBytes:"hello" length:5]]; | |
| 253 [field1 addGroup:[[unknownFields_ copy] autorelease]]; | |
| 254 GPBUnknownField* field2 = [[[GPBUnknownField alloc] initWithNumber:2] autorele
ase]; | |
| 255 [field2 mergeFromField:field1]; | |
| 256 XCTAssertEqualObjects(field1, field2); | |
| 257 } | |
| 258 | |
| 259 @end | |
| OLD | NEW |