Home | History | Annotate | Download | only in Tests
      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] autorelease];
    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] autorelease];
    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] autorelease];
    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] autorelease];
    255   [field2 mergeFromField:field1];
    256   XCTAssertEqualObjects(field1, field2);
    257 }
    258 
    259 @end
    260