| // 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. | 
 |  | 
 | #import <XCTest/XCTest.h> | 
 |  | 
 | #import "GPBUtilities_PackagePrivate.h" | 
 |  | 
 | #import <objc/runtime.h> | 
 |  | 
 | #import "GPBTestUtilities.h" | 
 |  | 
 | #import "GPBDescriptor.h" | 
 | #import "GPBDescriptor_PackagePrivate.h" | 
 | #import "GPBMessage.h" | 
 | #import "GPBUnknownField_PackagePrivate.h" | 
 |  | 
 | #import "google/protobuf/MapUnittest.pbobjc.h" | 
 | #import "google/protobuf/Unittest.pbobjc.h" | 
 | #import "google/protobuf/UnittestObjc.pbobjc.h" | 
 |  | 
 | @interface UtilitiesTests : GPBTestCase | 
 | @end | 
 |  | 
 | @implementation UtilitiesTests | 
 |  | 
 | - (void)testRightShiftFunctions { | 
 |   XCTAssertEqual((1UL << 31) >> 31, 1UL); | 
 |   XCTAssertEqual((int32_t)(1U << 31) >> 31, -1); | 
 |   XCTAssertEqual((1ULL << 63) >> 63, 1ULL); | 
 |   XCTAssertEqual((int64_t)(1ULL << 63) >> 63, -1LL); | 
 |  | 
 |   XCTAssertEqual(GPBLogicalRightShift32((1U << 31), 31), 1); | 
 |   XCTAssertEqual(GPBLogicalRightShift64((1ULL << 63), 63), 1LL); | 
 | } | 
 |  | 
 | - (void)testGPBDecodeTextFormatName { | 
 |   uint8_t decodeData[] = { | 
 |     0x6, | 
 |     // An inlined string (first to make sure the leading null is handled | 
 |     // correctly, and with a key of zero to check that). | 
 |     0x0, 0x0, 'z', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'I', 'J', 0x0, | 
 |     // All as is (00 op) | 
 |     0x1, 0x0A, 0x0, | 
 |     // Underscore, upper + 9 (10 op) | 
 |     0x3, 0xCA, 0x0, | 
 |     //  Upper + 3 (10 op), underscore, upper + 5 (10 op) | 
 |     0x2, 0x44, 0xC6, 0x0, | 
 |     // All Upper for 4 (11 op), underscore, underscore, upper + 5 (10 op), | 
 |     // underscore, lower + 0 (01 op) | 
 |     0x4, 0x64, 0x80, 0xC5, 0xA1, 0x0, | 
 |     // 2 byte key: as is + 3 (00 op), underscore, lower + 4 (01 op), | 
 |     //   underscore, lower + 3 (01 op), underscore, lower + 1 (01 op), | 
 |     //   underscore, lower + 30 (01 op), as is + 30 (00 op), as is + 13 (00 op), | 
 |     //   underscore, as is + 3 (00 op) | 
 |     0xE8, 0x07, 0x04, 0xA5, 0xA4, 0xA2, 0xBF, 0x1F, 0x0E, 0x84, 0x0, | 
 |   }; | 
 |   NSString *inputStr = @"abcdefghIJ"; | 
 |  | 
 |   // Empty inputs | 
 |  | 
 |   XCTAssertNil(GPBDecodeTextFormatName(nil, 1, NULL)); | 
 |   XCTAssertNil(GPBDecodeTextFormatName(decodeData, 1, NULL)); | 
 |   XCTAssertNil(GPBDecodeTextFormatName(nil, 1, inputStr)); | 
 |  | 
 |   // Keys not found. | 
 |  | 
 |   XCTAssertNil(GPBDecodeTextFormatName(decodeData, 5, inputStr)); | 
 |   XCTAssertNil(GPBDecodeTextFormatName(decodeData, -1, inputStr)); | 
 |  | 
 |   // Some name decodes. | 
 |  | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 1, inputStr), @"abcdefghIJ"); | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 2, inputStr), @"Abcd_EfghIJ"); | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 3, inputStr), @"_AbcdefghIJ"); | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 4, inputStr), @"ABCD__EfghI_j"); | 
 |  | 
 |   // An inlined string (and key of zero). | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 0, inputStr), @"zbcdefghIJ"); | 
 |  | 
 |   // Long name so multiple decode ops are needed. | 
 |   inputStr = @"longFieldNameIsLooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong1000"; | 
 |   XCTAssertEqualObjects(GPBDecodeTextFormatName(decodeData, 1000, inputStr), | 
 |                         @"long_field_name_is_looooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooong_1000"); | 
 | } | 
 |  | 
 | - (void)testTextFormat { | 
 |   TestAllTypes *message = [TestAllTypes message]; | 
 |  | 
 |   // Not kGPBDefaultRepeatCount because we are comparing to golden master file | 
 |   // which was generated with 2. | 
 |   [self setAllFields:message repeatedCount:2]; | 
 |  | 
 |   NSString *result = GPBTextFormatForMessage(message, nil); | 
 |  | 
 |   NSString *fileName = @"text_format_unittest_data.txt"; | 
 |   NSData *resultData = [result dataUsingEncoding:NSUTF8StringEncoding]; | 
 |   NSData *expectedData = | 
 |       [self getDataFileNamed:fileName dataToWrite:resultData]; | 
 |   NSString *expected = [[NSString alloc] initWithData:expectedData | 
 |                                              encoding:NSUTF8StringEncoding]; | 
 |   XCTAssertEqualObjects(expected, result); | 
 |   [expected release]; | 
 | } | 
 |  | 
 | - (void)testTextFormatExtra { | 
 |   // -testTextFormat uses all protos with fields that don't require special | 
 |   // handing for figuring out the names.  The ObjC proto has a bunch of oddball | 
 |   // field and enum names that require the decode info to get right, so this | 
 |   // confirms they generated and decoded correctly. | 
 |  | 
 |   self_Class *message = [self_Class message]; | 
 |   message.cmd = YES; | 
 |   message.isProxy_p = YES; | 
 |   message.subEnum = self_autorelease_RetainCount; | 
 |   message.new_p.copy_p = @"foo"; | 
 |  | 
 |   NSString *expected = @"_cmd: true\n" | 
 |                        @"isProxy: true\n" | 
 |                        @"SubEnum: retainCount\n" | 
 |                        @"New {\n" | 
 |                        @"  copy: \"foo\"\n" | 
 |                        @"}\n"; | 
 |   NSString *result = GPBTextFormatForMessage(message, nil); | 
 |   XCTAssertEqualObjects(expected, result); | 
 | } | 
 |  | 
 | - (void)testTextFormatMaps { | 
 |   TestMap *message = [TestMap message]; | 
 |  | 
 |   // Map iteration order doesn't have to be stable, so use only one entry. | 
 |   [self setAllMapFields:message numEntries:1]; | 
 |  | 
 |   NSString *result = GPBTextFormatForMessage(message, nil); | 
 |  | 
 |   NSString *fileName = @"text_format_map_unittest_data.txt"; | 
 |   NSData *resultData = [result dataUsingEncoding:NSUTF8StringEncoding]; | 
 |   NSData *expectedData = | 
 |       [self getDataFileNamed:fileName dataToWrite:resultData]; | 
 |   NSString *expected = [[NSString alloc] initWithData:expectedData | 
 |                                              encoding:NSUTF8StringEncoding]; | 
 |   XCTAssertEqualObjects(expected, result); | 
 |   [expected release]; | 
 | } | 
 |  | 
 | - (void)testTextFormatExtensions { | 
 |   TestAllExtensions *message = [TestAllExtensions message]; | 
 |  | 
 |   // Not kGPBDefaultRepeatCount because we are comparing to golden master file | 
 |   // which was generated with 2. | 
 |   [self setAllExtensions:message repeatedCount:2]; | 
 |  | 
 |   NSString *result = GPBTextFormatForMessage(message, nil); | 
 |  | 
 |   // NOTE: ObjC TextFormat doesn't have the proper extension names so it | 
 |   // uses comments for the ObjC name and raw numbers for the fields instead | 
 |   // of the bracketed extension name. | 
 |   NSString *fileName = @"text_format_extensions_unittest_data.txt"; | 
 |   NSData *resultData = [result dataUsingEncoding:NSUTF8StringEncoding]; | 
 |   NSData *expectedData = | 
 |       [self getDataFileNamed:fileName dataToWrite:resultData]; | 
 |   NSString *expected = [[NSString alloc] initWithData:expectedData | 
 |                                              encoding:NSUTF8StringEncoding]; | 
 |   XCTAssertEqualObjects(expected, result); | 
 |   [expected release]; | 
 | } | 
 |  | 
 | - (void)testSetRepeatedFields { | 
 |   TestAllTypes *message = [TestAllTypes message]; | 
 |  | 
 |   NSDictionary *repeatedFieldValues = @{ | 
 |     @"repeatedStringArray" : [@[@"foo", @"bar"] mutableCopy], | 
 |     @"repeatedBoolArray" : [GPBBoolArray arrayWithValue:YES], | 
 |     @"repeatedInt32Array" : [GPBInt32Array arrayWithValue:14], | 
 |     @"repeatedInt64Array" : [GPBInt64Array arrayWithValue:15], | 
 |     @"repeatedUint32Array" : [GPBUInt32Array arrayWithValue:16], | 
 |     @"repeatedUint64Array" : [GPBUInt64Array arrayWithValue:16], | 
 |     @"repeatedFloatArray" : [GPBFloatArray arrayWithValue:16], | 
 |     @"repeatedDoubleArray" : [GPBDoubleArray arrayWithValue:16], | 
 |     @"repeatedNestedEnumArray" : | 
 |         [GPBEnumArray arrayWithValidationFunction:TestAllTypes_NestedEnum_IsValidValue | 
 |                                          rawValue:TestAllTypes_NestedEnum_Foo], | 
 |   }; | 
 |   for (NSString *fieldName in repeatedFieldValues) { | 
 |     GPBFieldDescriptor *field = | 
 |         [message.descriptor fieldWithName:fieldName]; | 
 |     XCTAssertNotNil(field, @"No field with name: %@", fieldName); | 
 |     id expectedValues = repeatedFieldValues[fieldName]; | 
 |     GPBSetMessageRepeatedField(message, field, expectedValues); | 
 |     XCTAssertEqualObjects(expectedValues, | 
 |                           [message valueForKeyPath:fieldName]); | 
 |   } | 
 | } | 
 |  | 
 | // Helper to make an unknown field set with something in it. | 
 | static GPBUnknownFieldSet *UnknownFieldsSetHelper(int num) { | 
 |   GPBUnknownFieldSet *result = | 
 |       [[[GPBUnknownFieldSet alloc] init] autorelease]; | 
 |  | 
 |   GPBUnknownField *field = | 
 |       [[[GPBUnknownField alloc] initWithNumber:num] autorelease]; | 
 |   [field addVarint:num]; | 
 |   [result addField:field]; | 
 |  | 
 |   return result; | 
 | } | 
 |  | 
 | - (void)testDropMessageUnknownFieldsRecursively { | 
 |   TestAllExtensions *message = [TestAllExtensions message]; | 
 |  | 
 |   // Give it unknownFields. | 
 |   message.unknownFields = UnknownFieldsSetHelper(777); | 
 |  | 
 |   // Given it extensions that include a message with unknown fields of its own. | 
 |   { | 
 |     // Int | 
 |     [message setExtension:[UnittestRoot optionalInt32Extension] value:@5]; | 
 |  | 
 |     // Group | 
 |     OptionalGroup_extension *optionalGroup = [OptionalGroup_extension message]; | 
 |     optionalGroup.a = 123; | 
 |     optionalGroup.unknownFields = UnknownFieldsSetHelper(779); | 
 |     [message setExtension:[UnittestRoot optionalGroupExtension] | 
 |                     value:optionalGroup]; | 
 |  | 
 |     // Message | 
 |     TestAllTypes_NestedMessage *nestedMessage = | 
 |         [TestAllTypes_NestedMessage message]; | 
 |     nestedMessage.bb = 456; | 
 |     nestedMessage.unknownFields = UnknownFieldsSetHelper(778); | 
 |     [message setExtension:[UnittestRoot optionalNestedMessageExtension] | 
 |                     value:nestedMessage]; | 
 |  | 
 |     // Repeated Group | 
 |     RepeatedGroup_extension *repeatedGroup = | 
 |       [[RepeatedGroup_extension alloc] init]; | 
 |     repeatedGroup.a = 567; | 
 |     repeatedGroup.unknownFields = UnknownFieldsSetHelper(780); | 
 |     [message addExtension:[UnittestRoot repeatedGroupExtension] | 
 |                     value:repeatedGroup]; | 
 |     [repeatedGroup release]; | 
 |  | 
 |     // Repeated Message | 
 |     nestedMessage = [[TestAllTypes_NestedMessage alloc] init]; | 
 |     nestedMessage.bb = 678; | 
 |     nestedMessage.unknownFields = UnknownFieldsSetHelper(781); | 
 |     [message addExtension:[UnittestRoot repeatedNestedMessageExtension] | 
 |                     value:nestedMessage]; | 
 |     [nestedMessage release]; | 
 |   } | 
 |  | 
 |   // Confirm everything is there. | 
 |  | 
 |   XCTAssertNotNil(message); | 
 |   XCTAssertNotNil(message.unknownFields); | 
 |   XCTAssertTrue([message hasExtension:[UnittestRoot optionalInt32Extension]]); | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot optionalGroupExtension]]); | 
 |     OptionalGroup_extension *optionalGroup = | 
 |         [message getExtension:[UnittestRoot optionalGroupExtension]]; | 
 |     XCTAssertNotNil(optionalGroup); | 
 |     XCTAssertEqual(optionalGroup.a, 123); | 
 |     XCTAssertNotNil(optionalGroup.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot optionalNestedMessageExtension]]); | 
 |     TestAllTypes_NestedMessage *nestedMessage = | 
 |         [message getExtension:[UnittestRoot optionalNestedMessageExtension]]; | 
 |     XCTAssertNotNil(nestedMessage); | 
 |     XCTAssertEqual(nestedMessage.bb, 456); | 
 |     XCTAssertNotNil(nestedMessage.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot repeatedGroupExtension]]); | 
 |     NSArray *repeatedGroups = [message getExtension:[UnittestRoot repeatedGroupExtension]]; | 
 |     XCTAssertEqual(repeatedGroups.count, (NSUInteger)1); | 
 |     RepeatedGroup_extension *repeatedGroup = repeatedGroups.firstObject; | 
 |     XCTAssertNotNil(repeatedGroup); | 
 |     XCTAssertEqual(repeatedGroup.a, 567); | 
 |     XCTAssertNotNil(repeatedGroup.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot repeatedNestedMessageExtension]]); | 
 |     NSArray *repeatedNestedMessages = [message getExtension:[UnittestRoot repeatedNestedMessageExtension]]; | 
 |     XCTAssertEqual(repeatedNestedMessages.count, (NSUInteger)1); | 
 |     TestAllTypes_NestedMessage *repeatedNestedMessage = repeatedNestedMessages.firstObject; | 
 |     XCTAssertNotNil(repeatedNestedMessage); | 
 |     XCTAssertEqual(repeatedNestedMessage.bb, 678); | 
 |     XCTAssertNotNil(repeatedNestedMessage.unknownFields); | 
 |   } | 
 |  | 
 |   // Drop them. | 
 |   GPBMessageDropUnknownFieldsRecursively(message); | 
 |  | 
 |   // Confirm unknowns are gone from within the messages. | 
 |  | 
 |   XCTAssertNotNil(message); | 
 |   XCTAssertNil(message.unknownFields); | 
 |   XCTAssertTrue([message hasExtension:[UnittestRoot optionalInt32Extension]]); | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot optionalGroupExtension]]); | 
 |     OptionalGroup_extension *optionalGroup = | 
 |         [message getExtension:[UnittestRoot optionalGroupExtension]]; | 
 |     XCTAssertNotNil(optionalGroup); | 
 |     XCTAssertEqual(optionalGroup.a, 123); | 
 |     XCTAssertNil(optionalGroup.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot optionalNestedMessageExtension]]); | 
 |     TestAllTypes_NestedMessage *nestedMessage = | 
 |         [message getExtension:[UnittestRoot optionalNestedMessageExtension]]; | 
 |     XCTAssertNotNil(nestedMessage); | 
 |     XCTAssertEqual(nestedMessage.bb, 456); | 
 |     XCTAssertNil(nestedMessage.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot repeatedGroupExtension]]); | 
 |     NSArray *repeatedGroups = [message getExtension:[UnittestRoot repeatedGroupExtension]]; | 
 |     XCTAssertEqual(repeatedGroups.count, (NSUInteger)1); | 
 |     RepeatedGroup_extension *repeatedGroup = repeatedGroups.firstObject; | 
 |     XCTAssertNotNil(repeatedGroup); | 
 |     XCTAssertEqual(repeatedGroup.a, 567); | 
 |     XCTAssertNil(repeatedGroup.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     XCTAssertTrue([message hasExtension:[UnittestRoot repeatedNestedMessageExtension]]); | 
 |     NSArray *repeatedNestedMessages = [message getExtension:[UnittestRoot repeatedNestedMessageExtension]]; | 
 |     XCTAssertEqual(repeatedNestedMessages.count, (NSUInteger)1); | 
 |     TestAllTypes_NestedMessage *repeatedNestedMessage = repeatedNestedMessages.firstObject; | 
 |     XCTAssertNotNil(repeatedNestedMessage); | 
 |     XCTAssertEqual(repeatedNestedMessage.bb, 678); | 
 |     XCTAssertNil(repeatedNestedMessage.unknownFields); | 
 |   } | 
 |  | 
 | } | 
 |  | 
 | - (void)testDropMessageUnknownFieldsRecursively_Maps { | 
 |   TestMap *message = [TestMap message]; | 
 |  | 
 |   { | 
 |     ForeignMessage *foreignMessage = [ForeignMessage message]; | 
 |     foreignMessage.unknownFields = UnknownFieldsSetHelper(100); | 
 |     [message.mapInt32ForeignMessage setObject:foreignMessage forKey:100]; | 
 |  | 
 |     foreignMessage = [ForeignMessage message]; | 
 |     foreignMessage.unknownFields = UnknownFieldsSetHelper(101); | 
 |     [message.mapStringForeignMessage setObject:foreignMessage forKey:@"101"]; | 
 |   } | 
 |  | 
 |   // Confirm everything is there. | 
 |  | 
 |   XCTAssertNotNil(message); | 
 |  | 
 |   { | 
 |     ForeignMessage *foreignMessage = [message.mapInt32ForeignMessage objectForKey:100]; | 
 |     XCTAssertNotNil(foreignMessage); | 
 |     XCTAssertNotNil(foreignMessage.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     ForeignMessage *foreignMessage = [message.mapStringForeignMessage objectForKey:@"101"]; | 
 |     XCTAssertNotNil(foreignMessage); | 
 |     XCTAssertNotNil(foreignMessage.unknownFields); | 
 |   } | 
 |  | 
 |   GPBMessageDropUnknownFieldsRecursively(message); | 
 |  | 
 |   // Confirm unknowns are gone from within the messages. | 
 |  | 
 |   XCTAssertNotNil(message); | 
 |  | 
 |   { | 
 |     ForeignMessage *foreignMessage = [message.mapInt32ForeignMessage objectForKey:100]; | 
 |     XCTAssertNotNil(foreignMessage); | 
 |     XCTAssertNil(foreignMessage.unknownFields); | 
 |   } | 
 |  | 
 |   { | 
 |     ForeignMessage *foreignMessage = [message.mapStringForeignMessage objectForKey:@"101"]; | 
 |     XCTAssertNotNil(foreignMessage); | 
 |     XCTAssertNil(foreignMessage.unknownFields); | 
 |   } | 
 |  | 
 | } | 
 |  | 
 | @end |