objectivec/Tests/GPBWireFormatTests.m - third_party/protobuf - Git at Google

 // 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 "GPBTestUtilities.h"

 #import "GPBCodedInputStream.h"
 #import "GPBMessage_PackagePrivate.h"
 #import "GPBUnknownField_PackagePrivate.h"
 #import "google/protobuf/Unittest.pbobjc.h"
 #import "google/protobuf/UnittestMset.pbobjc.h"
 #import "google/protobuf/UnittestMsetWireFormat.pbobjc.h"

 @interface WireFormatTests : GPBTestCase
 @end

 @implementation WireFormatTests

 - (void)testSerialization {
   TestAllTypes* message = [self allSetRepeatedCount:kGPBDefaultRepeatCount];

   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];
   XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

   TestAllTypes* message2 = [TestAllTypes parseFromData:rawBytes error:NULL];

   [self assertAllFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
 }

 - (void)testSerializationPacked {
   TestPackedTypes* message =
       [self packedSetRepeatedCount:kGPBDefaultRepeatCount];

   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];
   XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

   TestPackedTypes* message2 =
       [TestPackedTypes parseFromData:rawBytes error:NULL];

   [self assertPackedFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
 }

 - (void)testSerializeExtensions {
   // TestAllTypes and TestAllExtensions should have compatible wire formats,
   // so if we serealize a TestAllExtensions then parse it as TestAllTypes
   // it should work.

   TestAllExtensions* message =
       [self allExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];
   XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

   TestAllTypes* message2 = [TestAllTypes parseFromData:rawBytes error:NULL];

   [self assertAllFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
 }

 - (void)testSerializePackedExtensions {
   // TestPackedTypes and TestPackedExtensions should have compatible wire
   // formats; check that they serialize to the same string.
   TestPackedExtensions* message =
       [self packedExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];

   TestPackedTypes* message2 =
       [self packedSetRepeatedCount:kGPBDefaultRepeatCount];
   NSData* rawBytes2 = message2.data;

   XCTAssertEqualObjects(rawBytes, rawBytes2);
 }

 - (void)testParseExtensions {
   // TestAllTypes and TestAllExtensions should have compatible wire formats,
   // so if we serialize a TestAllTypes then parse it as TestAllExtensions
   // it should work.

   TestAllTypes* message = [self allSetRepeatedCount:kGPBDefaultRepeatCount];
   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];

   GPBExtensionRegistry* registry = [self extensionRegistry];

   TestAllExtensions* message2 = [TestAllExtensions parseFromData:rawBytes
                                                extensionRegistry:registry
                                                            error:NULL];

   [self assertAllExtensionsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
 }


 - (void)testExtensionsSerializedSize {
   size_t allSet = [self allSetRepeatedCount:kGPBDefaultRepeatCount].serializedSize;
   size_t extensionSet = [self allExtensionsSetRepeatedCount:kGPBDefaultRepeatCount].serializedSize;
   XCTAssertEqual(allSet, extensionSet);
 }

 - (void)testParsePackedExtensions {
   // Ensure that packed extensions can be properly parsed.
   TestPackedExtensions* message =
       [self packedExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
   NSData* rawBytes = message.data;
   [self assertFieldsInOrder:rawBytes];

   GPBExtensionRegistry* registry = [self extensionRegistry];

   TestPackedExtensions* message2 = [TestPackedExtensions parseFromData:rawBytes
                                                      extensionRegistry:registry
                                                                  error:NULL];

   [self assertPackedExtensionsSet:message2
                     repeatedCount:kGPBDefaultRepeatCount];
 }

 const int kUnknownTypeId = 1550055;

 - (void)testSerializeMessageSet {
   // Set up a TestMessageSet with two known messages and an unknown one.
   TestMessageSet* message_set = [TestMessageSet message];
   [[message_set getExtension:[TestMessageSetExtension1 messageSetExtension]]
       setI:123];
   [[message_set getExtension:[TestMessageSetExtension2 messageSetExtension]]
       setStr:@"foo"];
   GPBUnknownField* unknownField =
       [[[GPBUnknownField alloc] initWithNumber:kUnknownTypeId] autorelease];
   [unknownField addLengthDelimited:[NSData dataWithBytes:"bar" length:3]];
   GPBUnknownFieldSet* unknownFieldSet =
       [[[GPBUnknownFieldSet alloc] init] autorelease];
   [unknownFieldSet addField:unknownField];
   [message_set setUnknownFields:unknownFieldSet];

   NSData* data = [message_set data];

   // Parse back using RawMessageSet and check the contents.
   RawMessageSet* raw = [RawMessageSet parseFromData:data error:NULL];

   XCTAssertEqual([raw.unknownFields countOfFields], (NSUInteger)0);

   XCTAssertEqual(raw.itemArray.count, (NSUInteger)3);
   XCTAssertEqual((uint32_t)[raw.itemArray[0] typeId],
                  [TestMessageSetExtension1 messageSetExtension].fieldNumber);
   XCTAssertEqual((uint32_t)[raw.itemArray[1] typeId],
                  [TestMessageSetExtension2 messageSetExtension].fieldNumber);
   XCTAssertEqual([raw.itemArray[2] typeId], kUnknownTypeId);

   TestMessageSetExtension1* message1 =
       [TestMessageSetExtension1 parseFromData:[((RawMessageSet_Item*)raw.itemArray[0]) message]
                                         error:NULL];
   XCTAssertEqual(message1.i, 123);

   TestMessageSetExtension2* message2 =
       [TestMessageSetExtension2 parseFromData:[((RawMessageSet_Item*)raw.itemArray[1]) message]
                                         error:NULL];
   XCTAssertEqualObjects(message2.str, @"foo");

   XCTAssertEqualObjects([raw.itemArray[2] message],
                         [NSData dataWithBytes:"bar" length:3]);
 }

 - (void)testParseMessageSet {
   // Set up a RawMessageSet with two known messages and an unknown one.
   RawMessageSet* raw = [RawMessageSet message];

   {
     RawMessageSet_Item* item = [RawMessageSet_Item message];
     item.typeId = [TestMessageSetExtension1 messageSetExtension].fieldNumber;
     TestMessageSetExtension1* message = [TestMessageSetExtension1 message];
     message.i = 123;
     item.message = [message data];
     [raw.itemArray addObject:item];
   }

   {
     RawMessageSet_Item* item = [RawMessageSet_Item message];
     item.typeId = [TestMessageSetExtension2 messageSetExtension].fieldNumber;
     TestMessageSetExtension2* message = [TestMessageSetExtension2 message];
     message.str = @"foo";
     item.message = [message data];
     [raw.itemArray addObject:item];
   }

   {
     RawMessageSet_Item* item = [RawMessageSet_Item message];
     item.typeId = kUnknownTypeId;
     item.message = [NSData dataWithBytes:"bar" length:3];
     [raw.itemArray addObject:item];
   }

   NSData* data = [raw data];

   // Parse as a TestMessageSet and check the contents.
   TestMessageSet* messageSet =
       [TestMessageSet parseFromData:data
                   extensionRegistry:[UnittestMsetRoot extensionRegistry]
                               error:NULL];

   XCTAssertEqual(
       [[messageSet
           getExtension:[TestMessageSetExtension1 messageSetExtension]] i],
       123);
   XCTAssertEqualObjects(
       [[messageSet
           getExtension:[TestMessageSetExtension2 messageSetExtension]] str],
       @"foo");

   XCTAssertEqual([messageSet.unknownFields countOfFields], (NSUInteger)1);
   GPBUnknownField* unknownField = [messageSet.unknownFields getField:kUnknownTypeId];
   XCTAssertNotNil(unknownField);
   XCTAssertEqual(unknownField.lengthDelimitedList.count, (NSUInteger)1);
   XCTAssertEqualObjects(unknownField.lengthDelimitedList[0],
                         [NSData dataWithBytes:"bar" length:3]);
 }

 - (void)assertFieldsInOrder:(NSData*)data {
   GPBCodedInputStream* input = [GPBCodedInputStream streamWithData:data];
   int32_t previousTag = 0;

   while (YES) {
     int32_t tag = [input readTag];
     if (tag == 0) {
       break;
     }

     XCTAssertGreaterThan(tag, previousTag);
     [input skipField:tag];
   }
 }

 @end
	// 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 "GPBTestUtilities.h"

	#import "GPBCodedInputStream.h"
	#import "GPBMessage_PackagePrivate.h"
	#import "GPBUnknownField_PackagePrivate.h"
	#import "google/protobuf/Unittest.pbobjc.h"
	#import "google/protobuf/UnittestMset.pbobjc.h"
	#import "google/protobuf/UnittestMsetWireFormat.pbobjc.h"

	@interface WireFormatTests : GPBTestCase
	@end

	@implementation WireFormatTests

	- (void)testSerialization {
	TestAllTypes* message = [self allSetRepeatedCount:kGPBDefaultRepeatCount];

	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];
	XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

	TestAllTypes* message2 = [TestAllTypes parseFromData:rawBytes error:NULL];

	[self assertAllFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
	}

	- (void)testSerializationPacked {
	TestPackedTypes* message =
	[self packedSetRepeatedCount:kGPBDefaultRepeatCount];

	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];
	XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

	TestPackedTypes* message2 =
	[TestPackedTypes parseFromData:rawBytes error:NULL];

	[self assertPackedFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
	}

	- (void)testSerializeExtensions {
	// TestAllTypes and TestAllExtensions should have compatible wire formats,
	// so if we serealize a TestAllExtensions then parse it as TestAllTypes
	// it should work.

	TestAllExtensions* message =
	[self allExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];
	XCTAssertEqual(message.serializedSize, (size_t)rawBytes.length);

	TestAllTypes* message2 = [TestAllTypes parseFromData:rawBytes error:NULL];

	[self assertAllFieldsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
	}

	- (void)testSerializePackedExtensions {
	// TestPackedTypes and TestPackedExtensions should have compatible wire
	// formats; check that they serialize to the same string.
	TestPackedExtensions* message =
	[self packedExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];

	TestPackedTypes* message2 =
	[self packedSetRepeatedCount:kGPBDefaultRepeatCount];
	NSData* rawBytes2 = message2.data;

	XCTAssertEqualObjects(rawBytes, rawBytes2);
	}

	- (void)testParseExtensions {
	// TestAllTypes and TestAllExtensions should have compatible wire formats,
	// so if we serialize a TestAllTypes then parse it as TestAllExtensions
	// it should work.

	TestAllTypes* message = [self allSetRepeatedCount:kGPBDefaultRepeatCount];
	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];

	GPBExtensionRegistry* registry = [self extensionRegistry];

	TestAllExtensions* message2 = [TestAllExtensions parseFromData:rawBytes
	extensionRegistry:registry
	error:NULL];

	[self assertAllExtensionsSet:message2 repeatedCount:kGPBDefaultRepeatCount];
	}


	- (void)testExtensionsSerializedSize {
	size_t allSet = [self allSetRepeatedCount:kGPBDefaultRepeatCount].serializedSize;
	size_t extensionSet = [self allExtensionsSetRepeatedCount:kGPBDefaultRepeatCount].serializedSize;
	XCTAssertEqual(allSet, extensionSet);
	}

	- (void)testParsePackedExtensions {
	// Ensure that packed extensions can be properly parsed.
	TestPackedExtensions* message =
	[self packedExtensionsSetRepeatedCount:kGPBDefaultRepeatCount];
	NSData* rawBytes = message.data;
	[self assertFieldsInOrder:rawBytes];

	GPBExtensionRegistry* registry = [self extensionRegistry];

	TestPackedExtensions* message2 = [TestPackedExtensions parseFromData:rawBytes
	extensionRegistry:registry
	error:NULL];

	[self assertPackedExtensionsSet:message2
	repeatedCount:kGPBDefaultRepeatCount];
	}

	const int kUnknownTypeId = 1550055;

	- (void)testSerializeMessageSet {
	// Set up a TestMessageSet with two known messages and an unknown one.
	TestMessageSet* message_set = [TestMessageSet message];
	[[message_set getExtension:[TestMessageSetExtension1 messageSetExtension]]
	setI:123];
	[[message_set getExtension:[TestMessageSetExtension2 messageSetExtension]]
	setStr:@"foo"];
	GPBUnknownField* unknownField =
	[[[GPBUnknownField alloc] initWithNumber:kUnknownTypeId] autorelease];
	[unknownField addLengthDelimited:[NSData dataWithBytes:"bar" length:3]];
	GPBUnknownFieldSet* unknownFieldSet =
	[[[GPBUnknownFieldSet alloc] init] autorelease];
	[unknownFieldSet addField:unknownField];
	[message_set setUnknownFields:unknownFieldSet];

	NSData* data = [message_set data];

	// Parse back using RawMessageSet and check the contents.
	RawMessageSet* raw = [RawMessageSet parseFromData:data error:NULL];

	XCTAssertEqual([raw.unknownFields countOfFields], (NSUInteger)0);

	XCTAssertEqual(raw.itemArray.count, (NSUInteger)3);
	XCTAssertEqual((uint32_t)[raw.itemArray[0] typeId],
	[TestMessageSetExtension1 messageSetExtension].fieldNumber);
	XCTAssertEqual((uint32_t)[raw.itemArray[1] typeId],
	[TestMessageSetExtension2 messageSetExtension].fieldNumber);
	XCTAssertEqual([raw.itemArray[2] typeId], kUnknownTypeId);

	TestMessageSetExtension1* message1 =
	[TestMessageSetExtension1 parseFromData:[((RawMessageSet_Item*)raw.itemArray[0]) message]
	error:NULL];
	XCTAssertEqual(message1.i, 123);

	TestMessageSetExtension2* message2 =
	[TestMessageSetExtension2 parseFromData:[((RawMessageSet_Item*)raw.itemArray[1]) message]
	error:NULL];
	XCTAssertEqualObjects(message2.str, @"foo");

	XCTAssertEqualObjects([raw.itemArray[2] message],
	[NSData dataWithBytes:"bar" length:3]);
	}

	- (void)testParseMessageSet {
	// Set up a RawMessageSet with two known messages and an unknown one.
	RawMessageSet* raw = [RawMessageSet message];

	{
	RawMessageSet_Item* item = [RawMessageSet_Item message];
	item.typeId = [TestMessageSetExtension1 messageSetExtension].fieldNumber;
	TestMessageSetExtension1* message = [TestMessageSetExtension1 message];
	message.i = 123;
	item.message = [message data];
	[raw.itemArray addObject:item];
	}

	{
	RawMessageSet_Item* item = [RawMessageSet_Item message];
	item.typeId = [TestMessageSetExtension2 messageSetExtension].fieldNumber;
	TestMessageSetExtension2* message = [TestMessageSetExtension2 message];
	message.str = @"foo";
	item.message = [message data];
	[raw.itemArray addObject:item];
	}

	{
	RawMessageSet_Item* item = [RawMessageSet_Item message];
	item.typeId = kUnknownTypeId;
	item.message = [NSData dataWithBytes:"bar" length:3];
	[raw.itemArray addObject:item];
	}

	NSData* data = [raw data];

	// Parse as a TestMessageSet and check the contents.
	TestMessageSet* messageSet =
	[TestMessageSet parseFromData:data
	extensionRegistry:[UnittestMsetRoot extensionRegistry]
	error:NULL];

	XCTAssertEqual(
	[[messageSet
	getExtension:[TestMessageSetExtension1 messageSetExtension]] i],
	123);
	XCTAssertEqualObjects(
	[[messageSet
	getExtension:[TestMessageSetExtension2 messageSetExtension]] str],
	@"foo");

	XCTAssertEqual([messageSet.unknownFields countOfFields], (NSUInteger)1);
	GPBUnknownField* unknownField = [messageSet.unknownFields getField:kUnknownTypeId];
	XCTAssertNotNil(unknownField);
	XCTAssertEqual(unknownField.lengthDelimitedList.count, (NSUInteger)1);
	XCTAssertEqualObjects(unknownField.lengthDelimitedList[0],
	[NSData dataWithBytes:"bar" length:3]);
	}

	- (void)assertFieldsInOrder:(NSData*)data {
	GPBCodedInputStream* input = [GPBCodedInputStream streamWithData:data];
	int32_t previousTag = 0;

	while (YES) {
	int32_t tag = [input readTag];
	if (tag == 0) {
	break;
	}

	XCTAssertGreaterThan(tag, previousTag);
	[input skipField:tag];
	}
	}

	@end