protobuf/objectivec/Tests/GPBMessageTests.m
2019-08-30 11:43:08 -04:00

2211 lines
86 KiB
Objective-C

// 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 <objc/runtime.h>
#import "GPBArray_PackagePrivate.h"
#import "GPBDescriptor.h"
#import "GPBDictionary_PackagePrivate.h"
#import "GPBMessage_PackagePrivate.h"
#import "GPBUnknownField_PackagePrivate.h"
#import "GPBUnknownFieldSet_PackagePrivate.h"
#import "google/protobuf/Unittest.pbobjc.h"
#import "google/protobuf/UnittestObjc.pbobjc.h"
#import "google/protobuf/UnittestObjcOptions.pbobjc.h"
@interface MessageTests : GPBTestCase
@end
@implementation MessageTests
// TODO(thomasvl): this should get split into a few files of logic junks, it is
// a jumble of things at the moment (and the testutils have a bunch of the real
// assertions).
- (TestAllTypes *)mergeSource {
TestAllTypes *message = [TestAllTypes message];
[message setOptionalInt32:1];
[message setOptionalString:@"foo"];
[message setOptionalForeignMessage:[ForeignMessage message]];
[message.repeatedStringArray addObject:@"bar"];
return message;
}
- (TestAllTypes *)mergeDestination {
TestAllTypes *message = [TestAllTypes message];
[message setOptionalInt64:2];
[message setOptionalString:@"baz"];
ForeignMessage *foreignMessage = [ForeignMessage message];
[foreignMessage setC:3];
[message setOptionalForeignMessage:foreignMessage];
[message.repeatedStringArray addObject:@"qux"];
return message;
}
- (TestAllTypes *)mergeDestinationWithoutForeignMessageIvar {
TestAllTypes *message = [TestAllTypes message];
[message setOptionalInt64:2];
[message setOptionalString:@"baz"];
[message.repeatedStringArray addObject:@"qux"];
return message;
}
- (TestAllTypes *)mergeResult {
TestAllTypes *message = [TestAllTypes message];
[message setOptionalInt32:1];
[message setOptionalInt64:2];
[message setOptionalString:@"foo"];
ForeignMessage *foreignMessage = [ForeignMessage message];
[foreignMessage setC:3];
[message setOptionalForeignMessage:foreignMessage];
[message.repeatedStringArray addObject:@"qux"];
[message.repeatedStringArray addObject:@"bar"];
return message;
}
- (TestAllTypes *)mergeResultForDestinationWithoutForeignMessageIvar {
TestAllTypes *message = [TestAllTypes message];
[message setOptionalInt32:1];
[message setOptionalInt64:2];
[message setOptionalString:@"foo"];
ForeignMessage *foreignMessage = [ForeignMessage message];
[message setOptionalForeignMessage:foreignMessage];
[message.repeatedStringArray addObject:@"qux"];
[message.repeatedStringArray addObject:@"bar"];
return message;
}
- (TestAllExtensions *)mergeExtensionsDestination {
TestAllExtensions *message = [TestAllExtensions message];
[message setExtension:[UnittestRoot optionalInt32Extension] value:@5];
[message setExtension:[UnittestRoot optionalStringExtension] value:@"foo"];
ForeignMessage *foreignMessage = [ForeignMessage message];
foreignMessage.c = 4;
[message setExtension:[UnittestRoot optionalForeignMessageExtension]
value:foreignMessage];
TestAllTypes_NestedMessage *nestedMessage =
[TestAllTypes_NestedMessage message];
[message setExtension:[UnittestRoot optionalNestedMessageExtension]
value:nestedMessage];
return message;
}
- (TestAllExtensions *)mergeExtensionsSource {
TestAllExtensions *message = [TestAllExtensions message];
[message setExtension:[UnittestRoot optionalInt64Extension] value:@6];
[message setExtension:[UnittestRoot optionalStringExtension] value:@"bar"];
ForeignMessage *foreignMessage = [ForeignMessage message];
[message setExtension:[UnittestRoot optionalForeignMessageExtension]
value:foreignMessage];
TestAllTypes_NestedMessage *nestedMessage =
[TestAllTypes_NestedMessage message];
nestedMessage.bb = 7;
[message setExtension:[UnittestRoot optionalNestedMessageExtension]
value:nestedMessage];
return message;
}
- (TestAllExtensions *)mergeExtensionsResult {
TestAllExtensions *message = [TestAllExtensions message];
[message setExtension:[UnittestRoot optionalInt32Extension] value:@5];
[message setExtension:[UnittestRoot optionalInt64Extension] value:@6];
[message setExtension:[UnittestRoot optionalStringExtension] value:@"bar"];
ForeignMessage *foreignMessage = [ForeignMessage message];
foreignMessage.c = 4;
[message setExtension:[UnittestRoot optionalForeignMessageExtension]
value:foreignMessage];
TestAllTypes_NestedMessage *nestedMessage =
[TestAllTypes_NestedMessage message];
nestedMessage.bb = 7;
[message setExtension:[UnittestRoot optionalNestedMessageExtension]
value:nestedMessage];
return message;
}
- (void)testMergeFrom {
TestAllTypes *result = [[self.mergeDestination copy] autorelease];
[result mergeFrom:self.mergeSource];
NSData *resultData = [result data];
NSData *mergeResultData = [self.mergeResult data];
XCTAssertEqualObjects(resultData, mergeResultData);
XCTAssertEqualObjects(result, self.mergeResult);
// Test when destination does not have an Ivar (type is an object) but source
// has such Ivar.
// The result must has the Ivar which is same as the one in source.
result = [[self.mergeDestinationWithoutForeignMessageIvar copy] autorelease];
[result mergeFrom:self.mergeSource];
resultData = [result data];
mergeResultData =
[self.mergeResultForDestinationWithoutForeignMessageIvar data];
XCTAssertEqualObjects(resultData, mergeResultData);
XCTAssertEqualObjects(
result, self.mergeResultForDestinationWithoutForeignMessageIvar);
// Test when destination is empty.
// The result must is same as the source.
result = [TestAllTypes message];
[result mergeFrom:self.mergeSource];
resultData = [result data];
mergeResultData = [self.mergeSource data];
XCTAssertEqualObjects(resultData, mergeResultData);
XCTAssertEqualObjects(result, self.mergeSource);
}
- (void)testMergeFromWithExtensions {
TestAllExtensions *result = [self mergeExtensionsDestination];
[result mergeFrom:[self mergeExtensionsSource]];
NSData *resultData = [result data];
NSData *mergeResultData = [[self mergeExtensionsResult] data];
XCTAssertEqualObjects(resultData, mergeResultData);
XCTAssertEqualObjects(result, [self mergeExtensionsResult]);
// Test merging from data.
result = [self mergeExtensionsDestination];
NSData *data = [[self mergeExtensionsSource] data];
XCTAssertNotNil(data);
[result mergeFromData:data
extensionRegistry:[UnittestRoot extensionRegistry]];
resultData = [result data];
XCTAssertEqualObjects(resultData, mergeResultData);
XCTAssertEqualObjects(result, [self mergeExtensionsResult]);
}
- (void)testIsEquals {
TestAllTypes *result = [[self.mergeDestination copy] autorelease];
[result mergeFrom:self.mergeSource];
XCTAssertEqualObjects(result.data, self.mergeResult.data);
XCTAssertEqualObjects(result, self.mergeResult);
TestAllTypes *result2 = [[self.mergeDestination copy] autorelease];
XCTAssertNotEqualObjects(result2.data, self.mergeResult.data);
XCTAssertNotEqualObjects(result2, self.mergeResult);
}
// =================================================================
// Required-field-related tests.
- (TestRequired *)testRequiredInitialized {
TestRequired *message = [TestRequired message];
[message setA:1];
[message setB:2];
[message setC:3];
return message;
}
- (void)testRequired {
TestRequired *message = [TestRequired message];
XCTAssertFalse(message.initialized);
[message setA:1];
XCTAssertFalse(message.initialized);
[message setB:1];
XCTAssertFalse(message.initialized);
[message setC:1];
XCTAssertTrue(message.initialized);
}
- (void)testRequiredForeign {
TestRequiredForeign *message = [TestRequiredForeign message];
XCTAssertTrue(message.initialized);
[message setOptionalMessage:[TestRequired message]];
XCTAssertFalse(message.initialized);
[message setOptionalMessage:self.testRequiredInitialized];
XCTAssertTrue(message.initialized);
[message.repeatedMessageArray addObject:[TestRequired message]];
XCTAssertFalse(message.initialized);
[message.repeatedMessageArray removeAllObjects];
[message.repeatedMessageArray addObject:self.testRequiredInitialized];
XCTAssertTrue(message.initialized);
}
- (void)testRequiredExtension {
TestAllExtensions *message = [TestAllExtensions message];
XCTAssertTrue(message.initialized);
[message setExtension:[TestRequired single] value:[TestRequired message]];
XCTAssertFalse(message.initialized);
[message setExtension:[TestRequired single]
value:self.testRequiredInitialized];
XCTAssertTrue(message.initialized);
[message addExtension:[TestRequired multi] value:[TestRequired message]];
XCTAssertFalse(message.initialized);
[message setExtension:[TestRequired multi]
index:0
value:self.testRequiredInitialized];
XCTAssertTrue(message.initialized);
}
- (void)testDataFromUninitialized {
TestRequired *message = [TestRequired message];
NSData *data = [message data];
// In DEBUG, the data generation will fail, but in non DEBUG, it passes
// because the check isn't done (for speed).
#ifdef DEBUG
XCTAssertNil(data);
#else
XCTAssertNotNil(data);
XCTAssertFalse(message.initialized);
#endif // DEBUG
}
- (void)testInitialized {
// We're mostly testing that no exception is thrown.
TestRequired *message = [TestRequired message];
XCTAssertFalse(message.initialized);
}
- (void)testDataFromNestedUninitialized {
TestRequiredForeign *message = [TestRequiredForeign message];
[message setOptionalMessage:[TestRequired message]];
[message.repeatedMessageArray addObject:[TestRequired message]];
[message.repeatedMessageArray addObject:[TestRequired message]];
NSData *data = [message data];
// In DEBUG, the data generation will fail, but in non DEBUG, it passes
// because the check isn't done (for speed).
#ifdef DEBUG
XCTAssertNil(data);
#else
XCTAssertNotNil(data);
XCTAssertFalse(message.initialized);
#endif // DEBUG
}
- (void)testNestedInitialized {
// We're mostly testing that no exception is thrown.
TestRequiredForeign *message = [TestRequiredForeign message];
[message setOptionalMessage:[TestRequired message]];
[message.repeatedMessageArray addObject:[TestRequired message]];
[message.repeatedMessageArray addObject:[TestRequired message]];
XCTAssertFalse(message.initialized);
}
- (void)testParseUninitialized {
NSError *error = nil;
TestRequired *msg =
[TestRequired parseFromData:GPBEmptyNSData() error:&error];
// In DEBUG, the parse will fail, but in non DEBUG, it passes because
// the check isn't done (for speed).
#ifdef DEBUG
XCTAssertNil(msg);
XCTAssertNotNil(error);
XCTAssertEqualObjects(error.domain, GPBMessageErrorDomain);
XCTAssertEqual(error.code, GPBMessageErrorCodeMissingRequiredField);
#else
XCTAssertNotNil(msg);
XCTAssertNil(error);
XCTAssertFalse(msg.initialized);
#endif // DEBUG
}
- (void)testCoding {
GPBMessage *original = [self mergeResult];
NSData *data =
[NSKeyedArchiver archivedDataWithRootObject:original];
id unarchivedObject = [NSKeyedUnarchiver unarchiveObjectWithData:data];
XCTAssertEqualObjects(unarchivedObject, original);
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(unarchivedObject, original);
}
- (void)testSecureCoding {
GPBMessage *original = [self mergeResult];
NSString *key = @"testing123";
NSMutableData *data = [NSMutableData data];
NSKeyedArchiver *archiver = [[NSKeyedArchiver alloc] initForWritingWithMutableData:data];
[archiver setRequiresSecureCoding:YES];
[archiver encodeObject:original forKey:key];
[archiver finishEncoding];
NSKeyedUnarchiver *unarchiver = [[NSKeyedUnarchiver alloc] initForReadingWithData:data];
[unarchiver setRequiresSecureCoding:YES];
id unarchivedObject = [unarchiver decodeObjectOfClass:[GPBMessage class]
forKey:key];
[unarchiver finishDecoding];
XCTAssertEqualObjects(unarchivedObject, original);
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(unarchivedObject, original);
}
- (void)testObjectReset {
// Tests a failure where clearing out defaults values caused an over release.
TestAllTypes *message = [TestAllTypes message];
message.hasOptionalNestedMessage = NO;
[message setOptionalNestedMessage:[TestAllTypes_NestedMessage message]];
message.hasOptionalNestedMessage = NO;
[message setOptionalNestedMessage:[TestAllTypes_NestedMessage message]];
[message setOptionalNestedMessage:nil];
message.hasOptionalNestedMessage = NO;
}
- (void)testSettingHasToYes {
TestAllTypes *message = [TestAllTypes message];
XCTAssertThrows([message setHasOptionalNestedMessage:YES]);
}
- (void)testRoot {
XCTAssertNotNil([UnittestRoot extensionRegistry]);
}
- (void)testGPBMessageSize {
// See the note in GPBMessage_PackagePrivate.h about why we want to keep the
// base instance size pointer size aligned.
size_t messageSize = class_getInstanceSize([GPBMessage class]);
XCTAssertEqual((messageSize % sizeof(void *)), (size_t)0,
@"Base size isn't pointer size aligned");
// Since we add storage ourselves (see +allocWithZone: in GPBMessage), confirm
// that the size of some generated classes is still the same as the base for
// that logic to work as desired.
size_t testMessageSize = class_getInstanceSize([TestAllTypes class]);
XCTAssertEqual(testMessageSize, messageSize);
}
- (void)testInit {
TestAllTypes *message = [TestAllTypes message];
[self assertClear:message];
}
- (void)testAccessors {
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
[self assertAllFieldsSet:message repeatedCount:kGPBDefaultRepeatCount];
}
- (void)testKVC_ValueForKey {
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
[self assertAllFieldsKVCMatch:message];
}
- (void)testKVC_SetValue_ForKey {
TestAllTypes *message = [TestAllTypes message];
[self setAllFieldsViaKVC:message repeatedCount:kGPBDefaultRepeatCount];
[self assertAllFieldsKVCMatch:message];
[self assertAllFieldsSet:message repeatedCount:kGPBDefaultRepeatCount];
[self assertAllFieldsKVCMatch:message];
}
- (void)testDescription {
// No real test, just exercise code
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
GPBUnknownFieldSet *unknownFields =
[[[GPBUnknownFieldSet alloc] init] autorelease];
GPBUnknownField *field =
[[[GPBUnknownField alloc] initWithNumber:2] autorelease];
[field addVarint:2];
[unknownFields addField:field];
field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease];
[field addVarint:4];
[unknownFields addField:field];
[message setUnknownFields:unknownFields];
NSString *description = [message description];
XCTAssertGreaterThan([description length], 0U);
GPBMessage *message2 = [TestAllExtensions message];
[message2 setExtension:[UnittestRoot optionalInt32Extension] value:@1];
[message2 addExtension:[UnittestRoot repeatedInt32Extension] value:@2];
description = [message2 description];
XCTAssertGreaterThan([description length], 0U);
}
- (void)testSetter {
// Test to make sure that if we set a value that has a default value
// with the default, that the has is set, and the value gets put into the
// message correctly.
TestAllTypes *message = [TestAllTypes message];
GPBDescriptor *descriptor = [[message class] descriptor];
XCTAssertNotNil(descriptor);
GPBFieldDescriptor *fieldDescriptor =
[descriptor fieldWithName:@"defaultInt32"];
XCTAssertNotNil(fieldDescriptor);
GPBGenericValue defaultValue = [fieldDescriptor defaultValue];
[message setDefaultInt32:defaultValue.valueInt32];
XCTAssertTrue(message.hasDefaultInt32);
XCTAssertEqual(message.defaultInt32, defaultValue.valueInt32);
// Do the same thing with an object type.
message = [TestAllTypes message];
fieldDescriptor = [descriptor fieldWithName:@"defaultString"];
XCTAssertNotNil(fieldDescriptor);
defaultValue = [fieldDescriptor defaultValue];
[message setDefaultString:defaultValue.valueString];
XCTAssertTrue(message.hasDefaultString);
XCTAssertEqualObjects(message.defaultString, defaultValue.valueString);
// Test default string type.
message = [TestAllTypes message];
XCTAssertEqualObjects(message.defaultString, @"hello");
XCTAssertFalse(message.hasDefaultString);
fieldDescriptor = [descriptor fieldWithName:@"defaultString"];
XCTAssertNotNil(fieldDescriptor);
defaultValue = [fieldDescriptor defaultValue];
[message setDefaultString:defaultValue.valueString];
XCTAssertEqualObjects(message.defaultString, @"hello");
XCTAssertTrue(message.hasDefaultString);
[message setDefaultString:nil];
XCTAssertEqualObjects(message.defaultString, @"hello");
XCTAssertFalse(message.hasDefaultString);
message.hasDefaultString = NO;
XCTAssertFalse(message.hasDefaultString);
XCTAssertEqualObjects(message.defaultString, @"hello");
// Test default bytes type.
NSData *defaultBytes = [@"world" dataUsingEncoding:NSUTF8StringEncoding];
XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
XCTAssertFalse(message.hasDefaultString);
fieldDescriptor = [descriptor fieldWithName:@"defaultBytes"];
XCTAssertNotNil(fieldDescriptor);
defaultValue = [fieldDescriptor defaultValue];
[message setDefaultBytes:defaultValue.valueData];
XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
XCTAssertTrue(message.hasDefaultBytes);
[message setDefaultBytes:nil];
XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
XCTAssertFalse(message.hasDefaultBytes);
message.hasDefaultBytes = NO;
XCTAssertFalse(message.hasDefaultBytes);
XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
// Test optional string.
XCTAssertFalse(message.hasOptionalString);
XCTAssertEqualObjects(message.optionalString, @"");
XCTAssertFalse(message.hasOptionalString);
message.optionalString = nil;
XCTAssertFalse(message.hasOptionalString);
XCTAssertEqualObjects(message.optionalString, @"");
NSString *string = @"string";
message.optionalString = string;
XCTAssertEqualObjects(message.optionalString, string);
XCTAssertTrue(message.hasOptionalString);
message.optionalString = nil;
XCTAssertFalse(message.hasOptionalString);
XCTAssertEqualObjects(message.optionalString, @"");
// Test optional data.
XCTAssertFalse(message.hasOptionalBytes);
XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
XCTAssertFalse(message.hasOptionalBytes);
message.optionalBytes = nil;
XCTAssertFalse(message.hasOptionalBytes);
XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
NSData *data = [@"bytes" dataUsingEncoding:NSUTF8StringEncoding];
message.optionalBytes = data;
XCTAssertEqualObjects(message.optionalBytes, data);
XCTAssertTrue(message.hasOptionalBytes);
message.optionalBytes = nil;
XCTAssertFalse(message.hasOptionalBytes);
XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
// Test lazy message setting
XCTAssertFalse(message.hasOptionalLazyMessage);
XCTAssertNotNil(message.optionalLazyMessage);
XCTAssertFalse(message.hasOptionalLazyMessage);
message.hasOptionalLazyMessage = NO;
XCTAssertFalse(message.hasOptionalLazyMessage);
XCTAssertNotNil(message.optionalLazyMessage);
XCTAssertFalse(message.hasOptionalLazyMessage);
message.optionalLazyMessage = nil;
XCTAssertFalse(message.hasOptionalLazyMessage);
// Test nested messages
XCTAssertFalse(message.hasOptionalLazyMessage);
message.optionalLazyMessage.bb = 1;
XCTAssertTrue(message.hasOptionalLazyMessage);
XCTAssertEqual(message.optionalLazyMessage.bb, 1);
XCTAssertNotNil(message.optionalLazyMessage);
message.optionalLazyMessage = nil;
XCTAssertFalse(message.hasOptionalLazyMessage);
XCTAssertEqual(message.optionalLazyMessage.bb, 0);
XCTAssertFalse(message.hasOptionalLazyMessage);
XCTAssertNotNil(message.optionalLazyMessage);
// -testDefaultSubMessages tests the "defaulting" handling of fields
// containing messages.
}
- (void)testRepeatedSetters {
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
[self modifyRepeatedFields:message];
[self assertRepeatedFieldsModified:message
repeatedCount:kGPBDefaultRepeatCount];
}
- (void)testClear {
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
[self clearAllFields:message];
[self assertClear:message];
TestAllTypes *message2 = [TestAllTypes message];
XCTAssertEqualObjects(message, message2);
}
- (void)testClearKVC {
TestAllTypes *message = [TestAllTypes message];
[self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
[self clearAllFields:message];
[self assertClear:message];
[self assertClearKVC:message];
}
- (void)testClearExtension {
// clearExtension() is not actually used in TestUtil, so try it manually.
GPBMessage *message1 = [TestAllExtensions message];
[message1 setExtension:[UnittestRoot optionalInt32Extension] value:@1];
XCTAssertTrue([message1 hasExtension:[UnittestRoot optionalInt32Extension]]);
[message1 clearExtension:[UnittestRoot optionalInt32Extension]];
XCTAssertFalse([message1 hasExtension:[UnittestRoot optionalInt32Extension]]);
GPBMessage *message2 = [TestAllExtensions message];
[message2 addExtension:[UnittestRoot repeatedInt32Extension] value:@1];
XCTAssertEqual(
[[message2 getExtension:[UnittestRoot repeatedInt32Extension]] count],
(NSUInteger)1);
[message2 clearExtension:[UnittestRoot repeatedInt32Extension]];
XCTAssertEqual(
[[message2 getExtension:[UnittestRoot repeatedInt32Extension]] count],
(NSUInteger)0);
// Clearing an unset extension field shouldn't make the target message
// visible.
GPBMessage *message3 = [TestAllExtensions message];
GPBMessage *extension_msg =
[message3 getExtension:[UnittestObjcRoot recursiveExtension]];
XCTAssertFalse([message3 hasExtension:[UnittestObjcRoot recursiveExtension]]);
[extension_msg clearExtension:[UnittestRoot optionalInt32Extension]];
XCTAssertFalse([message3 hasExtension:[UnittestObjcRoot recursiveExtension]]);
}
- (void)testDefaultingSubMessages {
TestAllTypes *message = [TestAllTypes message];
// Initially they should all not have values.
XCTAssertFalse(message.hasOptionalGroup);
XCTAssertFalse(message.hasOptionalNestedMessage);
XCTAssertFalse(message.hasOptionalForeignMessage);
XCTAssertFalse(message.hasOptionalImportMessage);
XCTAssertFalse(message.hasOptionalPublicImportMessage);
XCTAssertFalse(message.hasOptionalLazyMessage);
// They should auto create something when fetched.
TestAllTypes_OptionalGroup *optionalGroup = [message.optionalGroup retain];
TestAllTypes_NestedMessage *optionalNestedMessage =
[message.optionalNestedMessage retain];
ForeignMessage *optionalForeignMessage =
[message.optionalForeignMessage retain];
ImportMessage *optionalImportMessage = [message.optionalImportMessage retain];
PublicImportMessage *optionalPublicImportMessage =
[message.optionalPublicImportMessage retain];
TestAllTypes_NestedMessage *optionalLazyMessage =
[message.optionalLazyMessage retain];
XCTAssertNotNil(optionalGroup);
XCTAssertNotNil(optionalNestedMessage);
XCTAssertNotNil(optionalForeignMessage);
XCTAssertNotNil(optionalImportMessage);
XCTAssertNotNil(optionalPublicImportMessage);
XCTAssertNotNil(optionalLazyMessage);
// Although they were created, they should not respond to hasValue until that
// submessage is mutated.
XCTAssertFalse(message.hasOptionalGroup);
XCTAssertFalse(message.hasOptionalNestedMessage);
XCTAssertFalse(message.hasOptionalForeignMessage);
XCTAssertFalse(message.hasOptionalImportMessage);
XCTAssertFalse(message.hasOptionalPublicImportMessage);
XCTAssertFalse(message.hasOptionalLazyMessage);
// And they set that value back in to the message since the value created was
// mutable (so a second fetch should give the same object).
XCTAssertEqual(message.optionalGroup, optionalGroup);
XCTAssertEqual(message.optionalNestedMessage, optionalNestedMessage);
XCTAssertEqual(message.optionalForeignMessage, optionalForeignMessage);
XCTAssertEqual(message.optionalImportMessage, optionalImportMessage);
XCTAssertEqual(message.optionalPublicImportMessage,
optionalPublicImportMessage);
XCTAssertEqual(message.optionalLazyMessage, optionalLazyMessage);
// And the default objects for a second message should be distinct (again,
// since they are mutable, each needs their own copy).
TestAllTypes *message2 = [TestAllTypes message];
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(message2.optionalGroup, optionalGroup);
XCTAssertNotEqual(message2.optionalNestedMessage, optionalNestedMessage);
XCTAssertNotEqual(message2.optionalForeignMessage, optionalForeignMessage);
XCTAssertNotEqual(message2.optionalImportMessage, optionalImportMessage);
XCTAssertNotEqual(message2.optionalPublicImportMessage,
optionalPublicImportMessage);
XCTAssertNotEqual(message2.optionalLazyMessage, optionalLazyMessage);
// Setting the values to nil will clear the has flag, and on next access you
// get back new submessages.
message.optionalGroup = nil;
message.optionalNestedMessage = nil;
message.optionalForeignMessage = nil;
message.optionalImportMessage = nil;
message.optionalPublicImportMessage = nil;
message.optionalLazyMessage = nil;
XCTAssertFalse(message.hasOptionalGroup);
XCTAssertFalse(message.hasOptionalNestedMessage);
XCTAssertFalse(message.hasOptionalForeignMessage);
XCTAssertFalse(message.hasOptionalImportMessage);
XCTAssertFalse(message.hasOptionalPublicImportMessage);
XCTAssertFalse(message.hasOptionalLazyMessage);
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(message.optionalGroup, optionalGroup);
XCTAssertNotEqual(message.optionalNestedMessage, optionalNestedMessage);
XCTAssertNotEqual(message.optionalForeignMessage, optionalForeignMessage);
XCTAssertNotEqual(message.optionalImportMessage, optionalImportMessage);
XCTAssertNotEqual(message.optionalPublicImportMessage,
optionalPublicImportMessage);
XCTAssertNotEqual(message.optionalLazyMessage, optionalLazyMessage);
[optionalGroup release];
[optionalNestedMessage release];
[optionalForeignMessage release];
[optionalImportMessage release];
[optionalPublicImportMessage release];
[optionalLazyMessage release];
}
- (void)testMultiplePointersToAutocreatedMessage {
// Multiple objects pointing to the same autocreated message.
TestAllTypes *message = [TestAllTypes message];
TestAllTypes *message2 = [TestAllTypes message];
message2.optionalGroup = message.optionalGroup;
XCTAssertTrue([message2 hasOptionalGroup]);
XCTAssertFalse([message hasOptionalGroup]);
message2.optionalGroup.a = 42;
XCTAssertTrue([message hasOptionalGroup]);
XCTAssertTrue([message2 hasOptionalGroup]);
}
- (void)testCopyWithAutocreatedMessage {
// Mutable copy should not copy autocreated messages.
TestAllTypes *message = [TestAllTypes message];
message.optionalGroup.a = 42;
XCTAssertNotNil(message.optionalNestedMessage);
TestAllTypes *message2 = [[message copy] autorelease];
XCTAssertTrue([message2 hasOptionalGroup]);
XCTAssertFalse([message2 hasOptionalNestedMessage]);
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(message.optionalNestedMessage,
message2.optionalNestedMessage);
}
- (void)testClearAutocreatedSubmessage {
// Call clear on an intermediate submessage should cause it to get recreated
// on the next call.
TestRecursiveMessage *message = [TestRecursiveMessage message];
TestRecursiveMessage *message_inner = [message.a.a.a retain];
XCTAssertNotNil(message_inner);
XCTAssertTrue(GPBWasMessageAutocreatedBy(message_inner, message.a.a));
[message.a.a clear];
XCTAssertFalse(GPBWasMessageAutocreatedBy(message_inner, message.a.a));
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(message.a.a.a, message_inner);
[message_inner release];
}
- (void)testRetainAutocreatedSubmessage {
// Should be able to retain autocreated submessage while the creator is
// dealloced.
TestAllTypes *message = [TestAllTypes message];
ForeignMessage *subMessage;
@autoreleasepool {
TestAllTypes *message2 = [TestAllTypes message];
subMessage = message2.optionalForeignMessage; // Autocreated
message.optionalForeignMessage = subMessage;
XCTAssertTrue(GPBWasMessageAutocreatedBy(message.optionalForeignMessage,
message2));
}
// Should be the same object, and should still be live.
XCTAssertEqual(message.optionalForeignMessage, subMessage);
XCTAssertNotNil([subMessage description]);
}
- (void)testSetNilAutocreatedSubmessage {
TestRecursiveMessage *message = [TestRecursiveMessage message];
TestRecursiveMessage *message_inner = [message.a.a retain];
XCTAssertFalse([message hasA]);
XCTAssertFalse([message.a hasA]);
message.a.a = nil;
// |message.a| has to be made visible, but |message.a.a| was set to nil so
// shouldn't be.
XCTAssertTrue([message hasA]);
XCTAssertFalse([message.a hasA]);
// Setting submessage to nil should cause it to lose its creator.
XCTAssertFalse(GPBWasMessageAutocreatedBy(message_inner, message.a));
// After setting to nil, getting it again should create a new autocreated
// message.
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(message.a.a, message_inner);
[message_inner release];
}
- (void)testSetDoesntHaveAutocreatedSubmessage {
// Clearing submessage (set has == NO) should NOT cause it to lose its
// creator.
TestAllTypes *message = [TestAllTypes message];
TestAllTypes_NestedMessage *nestedMessage = message.optionalNestedMessage;
XCTAssertFalse([message hasOptionalNestedMessage]);
[message setHasOptionalNestedMessage:NO];
XCTAssertFalse([message hasOptionalNestedMessage]);
XCTAssertEqual(message.optionalNestedMessage, nestedMessage);
}
- (void)testSetAutocreatedMessageBecomesVisible {
// Setting a value should cause the submessage to appear to its creator.
// Test this several levels deep.
TestRecursiveMessage *message = [TestRecursiveMessage message];
message.a.a.a.a.i = 42;
XCTAssertTrue([message hasA]);
XCTAssertTrue([message.a hasA]);
XCTAssertTrue([message.a.a hasA]);
XCTAssertTrue([message.a.a.a hasA]);
XCTAssertFalse([message.a.a.a.a hasA]);
XCTAssertEqual(message.a.a.a.a.i, 42);
}
- (void)testClearUnsetFieldOfAutocreatedMessage {
// Clearing an unset field should not cause the submessage to appear to its
// creator.
TestRecursiveMessage *message = [TestRecursiveMessage message];
message.a.a.a.a.hasI = NO;
XCTAssertFalse([message hasA]);
XCTAssertFalse([message.a hasA]);
XCTAssertFalse([message.a.a hasA]);
XCTAssertFalse([message.a.a.a hasA]);
}
- (void)testAutocreatedSubmessageAssignSkip {
TestRecursiveMessage *message = [TestRecursiveMessage message];
TestRecursiveMessage *messageLevel1 = [message.a retain];
TestRecursiveMessage *messageLevel2 = [message.a.a retain];
TestRecursiveMessage *messageLevel3 = [message.a.a.a retain];
TestRecursiveMessage *messageLevel4 = [message.a.a.a.a retain];
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel1, message));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel4, messageLevel3));
// Test skipping over an autocreated submessage and ensure it gets unset.
message.a = message.a.a;
XCTAssertEqual(message.a, messageLevel2);
XCTAssertTrue([message hasA]);
XCTAssertEqual(message.a.a, messageLevel3);
XCTAssertFalse([message.a hasA]);
XCTAssertEqual(message.a.a.a, messageLevel4);
XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel1,
message)); // Because it was orphaned.
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
[messageLevel1 release];
[messageLevel2 release];
[messageLevel3 release];
[messageLevel4 release];
}
- (void)testAutocreatedSubmessageAssignLoop {
TestRecursiveMessage *message = [TestRecursiveMessage message];
TestRecursiveMessage *messageLevel1 = [message.a retain];
TestRecursiveMessage *messageLevel2 = [message.a.a retain];
TestRecursiveMessage *messageLevel3 = [message.a.a.a retain];
TestRecursiveMessage *messageLevel4 = [message.a.a.a.a retain];
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel1, message));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel4, messageLevel3));
// Test a property with a loop. You'd never do this but at least ensure the
// autocreated submessages behave sanely.
message.a.a = message.a;
XCTAssertTrue([message hasA]);
XCTAssertEqual(message.a, messageLevel1);
XCTAssertTrue([message.a hasA]);
XCTAssertEqual(message.a.a, messageLevel1);
XCTAssertTrue([message.a.a hasA]);
XCTAssertEqual(message.a.a.a, messageLevel1);
XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel1,
message)); // Because it was assigned.
XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel2,
messageLevel1)); // Because it was orphaned.
XCTAssertFalse([messageLevel2 hasA]);
// Break the retain loop.
message.a.a = nil;
XCTAssertTrue([message hasA]);
XCTAssertFalse([message.a hasA]);
[messageLevel1 release];
[messageLevel2 release];
[messageLevel3 release];
[messageLevel4 release];
}
- (void)testSetAutocreatedSubmessage {
// Setting autocreated submessage to another value should cause the old one to
// lose its creator.
TestAllTypes *message = [TestAllTypes message];
TestAllTypes_NestedMessage *nestedMessage =
[message.optionalNestedMessage retain];
message.optionalNestedMessage = [TestAllTypes_NestedMessage message];
XCTAssertTrue([message hasOptionalNestedMessage]);
XCTAssertTrue(message.optionalNestedMessage != nestedMessage);
XCTAssertFalse(GPBWasMessageAutocreatedBy(nestedMessage, message));
[nestedMessage release];
}
- (void)testAutocreatedUnknownFields {
// Doing anything with (except reading) unknown fields should cause the
// submessage to become visible.
TestAllTypes *message = [TestAllTypes message];
XCTAssertNotNil(message.optionalNestedMessage);
XCTAssertFalse([message hasOptionalNestedMessage]);
XCTAssertNil(message.optionalNestedMessage.unknownFields);
XCTAssertFalse([message hasOptionalNestedMessage]);
GPBUnknownFieldSet *unknownFields =
[[[GPBUnknownFieldSet alloc] init] autorelease];
message.optionalNestedMessage.unknownFields = unknownFields;
XCTAssertTrue([message hasOptionalNestedMessage]);
message.optionalNestedMessage = nil;
XCTAssertFalse([message hasOptionalNestedMessage]);
[message.optionalNestedMessage setUnknownFields:unknownFields];
XCTAssertTrue([message hasOptionalNestedMessage]);
}
- (void)testSetAutocreatedSubmessageToSelf {
// Setting submessage to itself should cause it to become visible.
TestAllTypes *message = [TestAllTypes message];
XCTAssertNotNil(message.optionalNestedMessage);
XCTAssertFalse([message hasOptionalNestedMessage]);
message.optionalNestedMessage = message.optionalNestedMessage;
XCTAssertTrue([message hasOptionalNestedMessage]);
}
- (void)testAutocreatedSubmessageMemoryLeaks {
// Test for memory leaks with autocreated submessages.
TestRecursiveMessage *message;
TestRecursiveMessage *messageLevel1;
TestRecursiveMessage *messageLevel2;
TestRecursiveMessage *messageLevel3;
TestRecursiveMessage *messageLevel4;
@autoreleasepool {
message = [[TestRecursiveMessage alloc] init];
messageLevel1 = [message.a retain];
messageLevel2 = [message.a.a retain];
messageLevel3 = [message.a.a.a retain];
messageLevel4 = [message.a.a.a.a retain];
message.a.i = 1;
}
XCTAssertEqual(message.retainCount, (NSUInteger)1);
[message release];
XCTAssertEqual(messageLevel1.retainCount, (NSUInteger)1);
[messageLevel1 release];
XCTAssertEqual(messageLevel2.retainCount, (NSUInteger)1);
[messageLevel2 release];
XCTAssertEqual(messageLevel3.retainCount, (NSUInteger)1);
[messageLevel3 release];
XCTAssertEqual(messageLevel4.retainCount, (NSUInteger)1);
[messageLevel4 release];
}
- (void)testDefaultingArrays {
// Basic tests for default creation of arrays in a message.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message2 =
[TestRecursiveMessageWithRepeatedField message];
// Simply accessing the array should not make any fields visible.
XCTAssertNotNil(message.a.a.iArray);
XCTAssertFalse([message hasA]);
XCTAssertFalse([message.a hasA]);
XCTAssertNotNil(message2.a.a.strArray);
XCTAssertFalse([message2 hasA]);
XCTAssertFalse([message2.a hasA]);
// But adding an element to the array should.
[message.a.a.iArray addValue:42];
XCTAssertTrue([message hasA]);
XCTAssertTrue([message.a hasA]);
XCTAssertEqual([message.a.a.iArray count], (NSUInteger)1);
[message2.a.a.strArray addObject:@"foo"];
XCTAssertTrue([message2 hasA]);
XCTAssertTrue([message2.a hasA]);
XCTAssertEqual([message2.a.a.strArray count], (NSUInteger)1);
}
- (void)testAutocreatedArrayShared {
// Multiple objects pointing to the same array.
TestRecursiveMessageWithRepeatedField *message1a =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message1b =
[TestRecursiveMessageWithRepeatedField message];
message1a.a.iArray = message1b.a.iArray;
XCTAssertTrue([message1a hasA]);
XCTAssertFalse([message1b hasA]);
[message1a.a.iArray addValue:1];
XCTAssertTrue([message1a hasA]);
XCTAssertTrue([message1b hasA]);
XCTAssertEqual(message1a.a.iArray, message1b.a.iArray);
TestRecursiveMessageWithRepeatedField *message2a =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message2b =
[TestRecursiveMessageWithRepeatedField message];
message2a.a.strArray = message2b.a.strArray;
XCTAssertTrue([message2a hasA]);
XCTAssertFalse([message2b hasA]);
[message2a.a.strArray addObject:@"bar"];
XCTAssertTrue([message2a hasA]);
XCTAssertTrue([message2b hasA]);
XCTAssertEqual(message2a.a.strArray, message2b.a.strArray);
}
- (void)testAutocreatedArrayCopy {
// Copy should not copy autocreated arrays.
TestAllTypes *message = [TestAllTypes message];
XCTAssertNotNil(message.repeatedStringArray);
XCTAssertNotNil(message.repeatedInt32Array);
TestAllTypes *message2 = [[message copy] autorelease];
// Pointer conparisions.
XCTAssertNotEqual(message.repeatedStringArray, message2.repeatedStringArray);
XCTAssertNotEqual(message.repeatedInt32Array, message2.repeatedInt32Array);
// Mutable copy should copy empty arrays that were explicitly set (end up
// with different objects that are equal).
TestAllTypes *message3 = [TestAllTypes message];
message3.repeatedInt32Array = [GPBInt32Array arrayWithValue:42];
message3.repeatedStringArray = [NSMutableArray arrayWithObject:@"wee"];
XCTAssertNotNil(message.repeatedInt32Array);
XCTAssertNotNil(message.repeatedStringArray);
TestAllTypes *message4 = [[message3 copy] autorelease];
XCTAssertNotEqual(message3.repeatedInt32Array, message4.repeatedInt32Array);
XCTAssertEqualObjects(message3.repeatedInt32Array,
message4.repeatedInt32Array);
XCTAssertNotEqual(message3.repeatedStringArray, message4.repeatedStringArray);
XCTAssertEqualObjects(message3.repeatedStringArray,
message4.repeatedStringArray);
}
- (void)testAutocreatedArrayRetain {
// Should be able to retain autocreated array while the creator is dealloced.
TestAllTypes *message = [TestAllTypes message];
@autoreleasepool {
TestAllTypes *message2 = [TestAllTypes message];
message.repeatedInt32Array = message2.repeatedInt32Array;
message.repeatedStringArray = message2.repeatedStringArray;
// Pointer conparision
XCTAssertEqual(message.repeatedInt32Array->_autocreator, message2);
XCTAssertTrue([message.repeatedStringArray
isKindOfClass:[GPBAutocreatedArray class]]);
XCTAssertEqual(
((GPBAutocreatedArray *)message.repeatedStringArray)->_autocreator,
message2);
}
XCTAssertNil(message.repeatedInt32Array->_autocreator);
XCTAssertTrue(
[message.repeatedStringArray isKindOfClass:[GPBAutocreatedArray class]]);
XCTAssertNil(
((GPBAutocreatedArray *)message.repeatedStringArray)->_autocreator);
}
- (void)testSetNilAutocreatedArray {
// Setting array to nil should cause it to lose its delegate.
TestAllTypes *message = [TestAllTypes message];
GPBInt32Array *repeatedInt32Array = [message.repeatedInt32Array retain];
GPBAutocreatedArray *repeatedStringArray =
(GPBAutocreatedArray *)[message.repeatedStringArray retain];
XCTAssertTrue([repeatedStringArray isKindOfClass:[GPBAutocreatedArray class]]);
XCTAssertEqual(repeatedInt32Array->_autocreator, message);
XCTAssertEqual(repeatedStringArray->_autocreator, message);
message.repeatedInt32Array = nil;
message.repeatedStringArray = nil;
XCTAssertNil(repeatedInt32Array->_autocreator);
XCTAssertNil(repeatedStringArray->_autocreator);
[repeatedInt32Array release];
[repeatedStringArray release];
}
- (void)testSetOverAutocreatedArrayAndSetAgain {
// Ensure when dealing with replacing an array it is handled being either
// an autocreated one or a straight NSArray.
// The real test here is that nothing crashes while doing the work.
TestAllTypes *message = [TestAllTypes message];
[message.repeatedStringArray addObject:@"foo"];
XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)1);
message.repeatedStringArray = [NSMutableArray arrayWithObjects:@"bar", @"bar2", nil];
XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)2);
message.repeatedStringArray = [NSMutableArray arrayWithObject:@"baz"];
XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)1);
}
- (void)testReplaceAutocreatedArray {
// Replacing array should orphan the old one and cause its creator to become
// visible.
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.iArray);
XCTAssertFalse([message hasA]);
GPBInt32Array *iArray = [message.a.iArray retain];
XCTAssertEqual(iArray->_autocreator, message.a); // Pointer comparision
message.a.iArray = [GPBInt32Array arrayWithValue:1];
XCTAssertTrue([message hasA]);
XCTAssertNotEqual(message.a.iArray, iArray); // Pointer comparision
XCTAssertNil(iArray->_autocreator);
[iArray release];
}
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.strArray);
XCTAssertFalse([message hasA]);
GPBAutocreatedArray *strArray =
(GPBAutocreatedArray *)[message.a.strArray retain];
XCTAssertTrue([strArray isKindOfClass:[GPBAutocreatedArray class]]);
XCTAssertEqual(strArray->_autocreator, message.a); // Pointer comparision
message.a.strArray = [NSMutableArray arrayWithObject:@"foo"];
XCTAssertTrue([message hasA]);
XCTAssertNotEqual(message.a.strArray, strArray); // Pointer comparision
XCTAssertNil(strArray->_autocreator);
[strArray release];
}
}
- (void)testSetAutocreatedArrayToSelf {
// Setting array to itself should cause it to become visible.
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.iArray);
XCTAssertFalse([message hasA]);
message.a.iArray = message.a.iArray;
XCTAssertTrue([message hasA]);
XCTAssertNil(message.a.iArray->_autocreator);
}
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.strArray);
XCTAssertFalse([message hasA]);
message.a.strArray = message.a.strArray;
XCTAssertTrue([message hasA]);
XCTAssertTrue([message.a.strArray isKindOfClass:[GPBAutocreatedArray class]]);
XCTAssertNil(((GPBAutocreatedArray *)message.a.strArray)->_autocreator);
}
}
- (void)testAutocreatedArrayRemoveAllValues {
// Calling removeAllValues on autocreated array should not cause it to be
// visible.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
[message.a.iArray removeAll];
XCTAssertFalse([message hasA]);
[message.a.strArray removeAllObjects];
XCTAssertFalse([message hasA]);
}
- (void)testDefaultingMaps {
// Basic tests for default creation of maps in a message.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message2 =
[TestRecursiveMessageWithRepeatedField message];
// Simply accessing the map should not make any fields visible.
XCTAssertNotNil(message.a.a.iToI);
XCTAssertFalse([message hasA]);
XCTAssertFalse([message.a hasA]);
XCTAssertNotNil(message2.a.a.strToStr);
XCTAssertFalse([message2 hasA]);
XCTAssertFalse([message2.a hasA]);
// But adding an element to the map should.
[message.a.a.iToI setInt32:100 forKey:200];
XCTAssertTrue([message hasA]);
XCTAssertTrue([message.a hasA]);
XCTAssertEqual([message.a.a.iToI count], (NSUInteger)1);
[message2.a.a.strToStr setObject:@"foo" forKey:@"bar"];
XCTAssertTrue([message2 hasA]);
XCTAssertTrue([message2.a hasA]);
XCTAssertEqual([message2.a.a.strToStr count], (NSUInteger)1);
}
- (void)testAutocreatedMapShared {
// Multiple objects pointing to the same map.
TestRecursiveMessageWithRepeatedField *message1a =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message1b =
[TestRecursiveMessageWithRepeatedField message];
message1a.a.iToI = message1b.a.iToI;
XCTAssertTrue([message1a hasA]);
XCTAssertFalse([message1b hasA]);
[message1a.a.iToI setInt32:1 forKey:2];
XCTAssertTrue([message1a hasA]);
XCTAssertTrue([message1b hasA]);
XCTAssertEqual(message1a.a.iToI, message1b.a.iToI);
TestRecursiveMessageWithRepeatedField *message2a =
[TestRecursiveMessageWithRepeatedField message];
TestRecursiveMessageWithRepeatedField *message2b =
[TestRecursiveMessageWithRepeatedField message];
message2a.a.strToStr = message2b.a.strToStr;
XCTAssertTrue([message2a hasA]);
XCTAssertFalse([message2b hasA]);
[message2a.a.strToStr setObject:@"bar" forKey:@"foo"];
XCTAssertTrue([message2a hasA]);
XCTAssertTrue([message2b hasA]);
XCTAssertEqual(message2a.a.strToStr, message2b.a.strToStr);
}
- (void)testAutocreatedMapCopy {
// Copy should not copy autocreated maps.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.strToStr);
XCTAssertNotNil(message.iToI);
TestRecursiveMessageWithRepeatedField *message2 =
[[message copy] autorelease];
// Pointer conparisions.
XCTAssertNotEqual(message.strToStr, message2.strToStr);
XCTAssertNotEqual(message.iToI, message2.iToI);
// Mutable copy should copy empty arrays that were explicitly set (end up
// with different objects that are equal).
TestRecursiveMessageWithRepeatedField *message3 =
[TestRecursiveMessageWithRepeatedField message];
message3.iToI = [[[GPBInt32Int32Dictionary alloc] init] autorelease];
[message3.iToI setInt32:10 forKey:20];
message3.strToStr =
[NSMutableDictionary dictionaryWithObject:@"abc" forKey:@"123"];
XCTAssertNotNil(message.iToI);
XCTAssertNotNil(message.iToI);
TestRecursiveMessageWithRepeatedField *message4 =
[[message3 copy] autorelease];
XCTAssertNotEqual(message3.iToI, message4.iToI);
XCTAssertEqualObjects(message3.iToI, message4.iToI);
XCTAssertNotEqual(message3.strToStr, message4.strToStr);
XCTAssertEqualObjects(message3.strToStr, message4.strToStr);
}
- (void)testAutocreatedMapRetain {
// Should be able to retain autocreated map while the creator is dealloced.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
@autoreleasepool {
TestRecursiveMessageWithRepeatedField *message2 =
[TestRecursiveMessageWithRepeatedField message];
message.iToI = message2.iToI;
message.strToStr = message2.strToStr;
// Pointer conparision
XCTAssertEqual(message.iToI->_autocreator, message2);
XCTAssertTrue([message.strToStr
isKindOfClass:[GPBAutocreatedDictionary class]]);
XCTAssertEqual(
((GPBAutocreatedDictionary *)message.strToStr)->_autocreator,
message2);
}
XCTAssertNil(message.iToI->_autocreator);
XCTAssertTrue(
[message.strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
XCTAssertNil(
((GPBAutocreatedDictionary *)message.strToStr)->_autocreator);
}
- (void)testSetNilAutocreatedMap {
// Setting map to nil should cause it to lose its delegate.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
GPBInt32Int32Dictionary *iToI = [message.iToI retain];
GPBAutocreatedDictionary *strToStr =
(GPBAutocreatedDictionary *)[message.strToStr retain];
XCTAssertTrue([strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
XCTAssertEqual(iToI->_autocreator, message);
XCTAssertEqual(strToStr->_autocreator, message);
message.iToI = nil;
message.strToStr = nil;
XCTAssertNil(iToI->_autocreator);
XCTAssertNil(strToStr->_autocreator);
[iToI release];
[strToStr release];
}
- (void)testSetOverAutocreatedMapAndSetAgain {
// Ensure when dealing with replacing a map it is handled being either
// an autocreated one or a straight NSDictionary.
// The real test here is that nothing crashes while doing the work.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
message.strToStr[@"foo"] = @"bar";
XCTAssertEqual(message.strToStr_Count, (NSUInteger)1);
message.strToStr =
[NSMutableDictionary dictionaryWithObjectsAndKeys:@"bar", @"key1", @"baz", @"key2", nil];
XCTAssertEqual(message.strToStr_Count, (NSUInteger)2);
message.strToStr =
[NSMutableDictionary dictionaryWithObject:@"baz" forKey:@"mumble"];
XCTAssertEqual(message.strToStr_Count, (NSUInteger)1);
}
- (void)testReplaceAutocreatedMap {
// Replacing map should orphan the old one and cause its creator to become
// visible.
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.iToI);
XCTAssertFalse([message hasA]);
GPBInt32Int32Dictionary *iToI = [message.a.iToI retain];
XCTAssertEqual(iToI->_autocreator, message.a); // Pointer comparision
message.a.iToI = [[[GPBInt32Int32Dictionary alloc] init] autorelease];
[message.a.iToI setInt32:6 forKey:7];
XCTAssertTrue([message hasA]);
XCTAssertNotEqual(message.a.iToI, iToI); // Pointer comparision
XCTAssertNil(iToI->_autocreator);
[iToI release];
}
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.strToStr);
XCTAssertFalse([message hasA]);
GPBAutocreatedDictionary *strToStr =
(GPBAutocreatedDictionary *)[message.a.strToStr retain];
XCTAssertTrue([strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
XCTAssertEqual(strToStr->_autocreator, message.a); // Pointer comparision
message.a.strToStr =
[NSMutableDictionary dictionaryWithObject:@"abc" forKey:@"def"];
XCTAssertTrue([message hasA]);
XCTAssertNotEqual(message.a.strToStr, strToStr); // Pointer comparision
XCTAssertNil(strToStr->_autocreator);
[strToStr release];
}
}
- (void)testSetAutocreatedMapToSelf {
// Setting map to itself should cause it to become visible.
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.iToI);
XCTAssertFalse([message hasA]);
message.a.iToI = message.a.iToI;
XCTAssertTrue([message hasA]);
XCTAssertNil(message.a.iToI->_autocreator);
}
{
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.strToStr);
XCTAssertFalse([message hasA]);
message.a.strToStr = message.a.strToStr;
XCTAssertTrue([message hasA]);
XCTAssertTrue([message.a.strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
XCTAssertNil(((GPBAutocreatedDictionary *)message.a.strToStr)->_autocreator);
}
}
- (void)testAutocreatedMapRemoveAllValues {
// Calling removeAll on autocreated map should not cause it to be visible.
TestRecursiveMessageWithRepeatedField *message =
[TestRecursiveMessageWithRepeatedField message];
[message.a.iToI removeAll];
XCTAssertFalse([message hasA]);
[message.a.strToStr removeAllObjects];
XCTAssertFalse([message hasA]);
}
- (void)testExtensionAccessors {
TestAllExtensions *message = [TestAllExtensions message];
[self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
[self assertAllExtensionsSet:message repeatedCount:kGPBDefaultRepeatCount];
}
- (void)testExtensionRepeatedSetters {
TestAllExtensions *message = [TestAllExtensions message];
[self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
[self modifyRepeatedExtensions:message];
[self assertRepeatedExtensionsModified:message
repeatedCount:kGPBDefaultRepeatCount];
}
- (void)testExtensionDefaults {
[self assertExtensionsClear:[TestAllExtensions message]];
}
- (void)testExtensionIsEquals {
TestAllExtensions *message = [TestAllExtensions message];
[self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
[self modifyRepeatedExtensions:message];
TestAllExtensions *message2 = [TestAllExtensions message];
[self setAllExtensions:message2 repeatedCount:kGPBDefaultRepeatCount];
XCTAssertFalse([message isEqual:message2]);
message2 = [TestAllExtensions message];
[self setAllExtensions:message2 repeatedCount:kGPBDefaultRepeatCount];
[self modifyRepeatedExtensions:message2];
XCTAssertEqualObjects(message, message2);
}
- (void)testExtensionsMergeFrom {
TestAllExtensions *message = [TestAllExtensions message];
[self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
[self modifyRepeatedExtensions:message];
message = [TestAllExtensions message];
[self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
TestAllExtensions *message2 = [TestAllExtensions message];
[self modifyRepeatedExtensions:message2];
[message2 mergeFrom:message];
XCTAssertEqualObjects(message, message2);
}
- (void)testDefaultingExtensionMessages {
TestAllExtensions *message = [TestAllExtensions message];
// Initially they should all not have values.
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
XCTAssertFalse([message
hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
// They should auto create something when fetched.
TestAllTypes_OptionalGroup *optionalGroup =
[message getExtension:[UnittestRoot optionalGroupExtension]];
TestAllTypes_NestedMessage *optionalNestedMessage =
[message getExtension:[UnittestRoot optionalNestedMessageExtension]];
ForeignMessage *optionalForeignMessage =
[message getExtension:[UnittestRoot optionalForeignMessageExtension]];
ImportMessage *optionalImportMessage =
[message getExtension:[UnittestRoot optionalImportMessageExtension]];
PublicImportMessage *optionalPublicImportMessage = [message
getExtension:[UnittestRoot optionalPublicImportMessageExtension]];
TestAllTypes_NestedMessage *optionalLazyMessage =
[message getExtension:[UnittestRoot optionalLazyMessageExtension]];
XCTAssertNotNil(optionalGroup);
XCTAssertNotNil(optionalNestedMessage);
XCTAssertNotNil(optionalForeignMessage);
XCTAssertNotNil(optionalImportMessage);
XCTAssertNotNil(optionalPublicImportMessage);
XCTAssertNotNil(optionalLazyMessage);
// Although it auto-created empty messages, it should not show that it has
// them.
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
// And they set that value back in to the message since the value created was
// mutable (so a second fetch should give the same object).
XCTAssertEqual([message getExtension:[UnittestRoot optionalGroupExtension]],
optionalGroup);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalNestedMessageExtension]],
optionalNestedMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalForeignMessageExtension]],
optionalForeignMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalImportMessageExtension]],
optionalImportMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
optionalPublicImportMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalLazyMessageExtension]],
optionalLazyMessage);
// And the default objects for a second message should be distinct (again,
// since they are mutable, each needs their own copy).
TestAllExtensions *message2 = [TestAllExtensions message];
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalGroupExtension]],
optionalGroup);
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalNestedMessageExtension]],
optionalNestedMessage);
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalForeignMessageExtension]],
optionalForeignMessage);
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalImportMessageExtension]],
optionalImportMessage);
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
optionalPublicImportMessage);
XCTAssertNotEqual(
[message2 getExtension:[UnittestRoot optionalLazyMessageExtension]],
optionalLazyMessage);
// Clear values, and on next access you get back new submessages.
[message setExtension:[UnittestRoot optionalGroupExtension] value:nil];
[message setExtension:[UnittestRoot optionalGroupExtension] value:nil];
[message setExtension:[UnittestRoot optionalNestedMessageExtension]
value:nil];
[message setExtension:[UnittestRoot optionalForeignMessageExtension]
value:nil];
[message setExtension:[UnittestRoot optionalImportMessageExtension]
value:nil];
[message setExtension:[UnittestRoot optionalPublicImportMessageExtension]
value:nil];
[message setExtension:[UnittestRoot optionalLazyMessageExtension] value:nil];
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
XCTAssertFalse([message
hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
XCTAssertFalse(
[message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
XCTAssertEqual([message getExtension:[UnittestRoot optionalGroupExtension]],
optionalGroup);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalNestedMessageExtension]],
optionalNestedMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalForeignMessageExtension]],
optionalForeignMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalImportMessageExtension]],
optionalImportMessage);
XCTAssertEqual(
[message
getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
optionalPublicImportMessage);
XCTAssertEqual(
[message getExtension:[UnittestRoot optionalLazyMessageExtension]],
optionalLazyMessage);
}
- (void)testMultiplePointersToAutocreatedExtension {
// 2 objects point to the same auto-created extension. One should "has" it.
// The other should not.
TestAllExtensions *message = [TestAllExtensions message];
TestAllExtensions *message2 = [TestAllExtensions message];
GPBExtensionDescriptor *extension = [UnittestRoot optionalGroupExtension];
[message setExtension:extension value:[message2 getExtension:extension]];
XCTAssertEqual([message getExtension:extension],
[message2 getExtension:extension]);
XCTAssertFalse([message2 hasExtension:extension]);
XCTAssertTrue([message hasExtension:extension]);
TestAllTypes_OptionalGroup *extensionValue =
[message2 getExtension:extension];
extensionValue.a = 1;
XCTAssertTrue([message2 hasExtension:extension]);
XCTAssertTrue([message hasExtension:extension]);
}
- (void)testCopyWithAutocreatedExtension {
// Mutable copy shouldn't copy autocreated extensions.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *optionalGroupExtension =
[UnittestRoot optionalGroupExtension];
GPBExtensionDescriptor *optionalNestedMessageExtesion =
[UnittestRoot optionalNestedMessageExtension];
TestAllTypes_OptionalGroup *optionalGroup =
[message getExtension:optionalGroupExtension];
optionalGroup.a = 42;
XCTAssertNotNil(optionalGroup);
XCTAssertNotNil([message getExtension:optionalNestedMessageExtesion]);
XCTAssertTrue([message hasExtension:optionalGroupExtension]);
XCTAssertFalse([message hasExtension:optionalNestedMessageExtesion]);
TestAllExtensions *message2 = [[message copy] autorelease];
// message2 should end up with its own copy of the optional group.
XCTAssertTrue([message2 hasExtension:optionalGroupExtension]);
XCTAssertEqualObjects([message getExtension:optionalGroupExtension],
[message2 getExtension:optionalGroupExtension]);
// Intentionally doing a pointer comparison.
XCTAssertNotEqual([message getExtension:optionalGroupExtension],
[message2 getExtension:optionalGroupExtension]);
XCTAssertFalse([message2 hasExtension:optionalNestedMessageExtesion]);
// Intentionally doing a pointer comparison (auto creation should be
// different)
XCTAssertNotEqual([message getExtension:optionalNestedMessageExtesion],
[message2 getExtension:optionalNestedMessageExtesion]);
}
- (void)testClearMessageAutocreatedExtension {
// Call clear should cause it to recreate its autocreated extensions.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *optionalGroupExtension =
[UnittestRoot optionalGroupExtension];
TestAllTypes_OptionalGroup *optionalGroup =
[[message getExtension:optionalGroupExtension] retain];
[message clear];
TestAllTypes_OptionalGroup *optionalGroupNew =
[message getExtension:optionalGroupExtension];
// Intentionally doing a pointer comparison.
XCTAssertNotEqual(optionalGroup, optionalGroupNew);
[optionalGroup release];
}
- (void)testRetainAutocreatedExtension {
// Should be able to retain autocreated extension while the creator is
// dealloced.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *optionalGroupExtension =
[UnittestRoot optionalGroupExtension];
@autoreleasepool {
TestAllExtensions *message2 = [TestAllExtensions message];
[message setExtension:optionalGroupExtension
value:[message2 getExtension:optionalGroupExtension]];
XCTAssertTrue(GPBWasMessageAutocreatedBy(
[message getExtension:optionalGroupExtension], message2));
}
XCTAssertFalse(GPBWasMessageAutocreatedBy(
[message getExtension:optionalGroupExtension], message));
}
- (void)testClearAutocreatedExtension {
// Clearing autocreated extension should NOT cause it to lose its creator.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *optionalGroupExtension =
[UnittestRoot optionalGroupExtension];
TestAllTypes_OptionalGroup *optionalGroup =
[[message getExtension:optionalGroupExtension] retain];
[message clearExtension:optionalGroupExtension];
TestAllTypes_OptionalGroup *optionalGroupNew =
[message getExtension:optionalGroupExtension];
XCTAssertEqual(optionalGroup, optionalGroupNew);
XCTAssertFalse([message hasExtension:optionalGroupExtension]);
[optionalGroup release];
// Clearing autocreated extension should not cause its creator to become
// visible
GPBExtensionDescriptor *recursiveExtension =
[UnittestObjcRoot recursiveExtension];
TestAllExtensions *message_lvl2 = [message getExtension:recursiveExtension];
TestAllExtensions *message_lvl3 =
[message_lvl2 getExtension:recursiveExtension];
[message_lvl3 clearExtension:recursiveExtension];
XCTAssertFalse([message hasExtension:recursiveExtension]);
}
- (void)testSetAutocreatedExtensionBecomesVisible {
// Setting an extension should cause the extension to appear to its creator.
// Test this several levels deep.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *recursiveExtension =
[UnittestObjcRoot recursiveExtension];
TestAllExtensions *message_lvl2 = [message getExtension:recursiveExtension];
TestAllExtensions *message_lvl3 =
[message_lvl2 getExtension:recursiveExtension];
TestAllExtensions *message_lvl4 =
[message_lvl3 getExtension:recursiveExtension];
XCTAssertFalse([message hasExtension:recursiveExtension]);
XCTAssertFalse([message_lvl2 hasExtension:recursiveExtension]);
XCTAssertFalse([message_lvl3 hasExtension:recursiveExtension]);
XCTAssertFalse([message_lvl4 hasExtension:recursiveExtension]);
[message_lvl4 setExtension:[UnittestRoot optionalInt32Extension] value:@(1)];
XCTAssertTrue([message hasExtension:recursiveExtension]);
XCTAssertTrue([message_lvl2 hasExtension:recursiveExtension]);
XCTAssertTrue([message_lvl3 hasExtension:recursiveExtension]);
XCTAssertFalse([message_lvl4 hasExtension:recursiveExtension]);
XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl4, message_lvl3));
XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl3, message_lvl2));
XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl2, message));
}
- (void)testSetAutocreatedExtensionToSelf {
// Setting extension to itself should cause it to become visible.
TestAllExtensions *message = [TestAllExtensions message];
GPBExtensionDescriptor *optionalGroupExtension =
[UnittestRoot optionalGroupExtension];
XCTAssertNotNil([message getExtension:optionalGroupExtension]);
XCTAssertFalse([message hasExtension:optionalGroupExtension]);
[message setExtension:optionalGroupExtension
value:[message getExtension:optionalGroupExtension]];
XCTAssertTrue([message hasExtension:optionalGroupExtension]);
}
- (void)testAutocreatedExtensionMemoryLeaks {
GPBExtensionDescriptor *recursiveExtension =
[UnittestObjcRoot recursiveExtension];
// Test for memory leaks with autocreated extensions.
TestAllExtensions *message;
TestAllExtensions *message_lvl2;
TestAllExtensions *message_lvl3;
TestAllExtensions *message_lvl4;
@autoreleasepool {
message = [[TestAllExtensions alloc] init];
message_lvl2 = [[message getExtension:recursiveExtension] retain];
message_lvl3 = [[message_lvl2 getExtension:recursiveExtension] retain];
message_lvl4 = [[message_lvl3 getExtension:recursiveExtension] retain];
[message_lvl2 setExtension:[UnittestRoot optionalInt32Extension]
value:@(1)];
}
XCTAssertEqual(message.retainCount, (NSUInteger)1);
@autoreleasepool {
[message release];
}
XCTAssertEqual(message_lvl2.retainCount, (NSUInteger)1);
@autoreleasepool {
[message_lvl2 release];
}
XCTAssertEqual(message_lvl3.retainCount, (NSUInteger)1);
@autoreleasepool {
[message_lvl3 release];
}
XCTAssertEqual(message_lvl4.retainCount, (NSUInteger)1);
[message_lvl4 release];
}
- (void)testSetExtensionWithAutocreatedValue {
GPBExtensionDescriptor *recursiveExtension =
[UnittestObjcRoot recursiveExtension];
TestAllExtensions *message;
@autoreleasepool {
message = [[TestAllExtensions alloc] init];
[message getExtension:recursiveExtension];
}
// This statements checks that the extension value isn't accidentally
// dealloced when removing it from the autocreated map.
[message setExtension:recursiveExtension
value:[message getExtension:recursiveExtension]];
XCTAssertTrue([message hasExtension:recursiveExtension]);
[message release];
}
- (void)testRecursion {
TestRecursiveMessage *message = [TestRecursiveMessage message];
XCTAssertNotNil(message.a);
XCTAssertNotNil(message.a.a);
XCTAssertEqual(message.a.a.i, 0);
}
- (void)testGenerateAndParseUnknownMessage {
GPBUnknownFieldSet *unknowns =
[[[GPBUnknownFieldSet alloc] init] autorelease];
[unknowns mergeVarintField:123 value:456];
GPBMessage *message = [GPBMessage message];
[message setUnknownFields:unknowns];
NSData *data = [message data];
GPBMessage *message2 =
[GPBMessage parseFromData:data extensionRegistry:nil error:NULL];
XCTAssertEqualObjects(message, message2);
}
- (void)testDelimitedWriteAndParseMultipleMessages {
GPBUnknownFieldSet *unknowns1 =
[[[GPBUnknownFieldSet alloc] init] autorelease];
[unknowns1 mergeVarintField:123 value:456];
GPBMessage *message1 = [GPBMessage message];
[message1 setUnknownFields:unknowns1];
GPBUnknownFieldSet *unknowns2 =
[[[GPBUnknownFieldSet alloc] init] autorelease];
[unknowns2 mergeVarintField:789 value:987];
[unknowns2 mergeVarintField:654 value:321];
GPBMessage *message2 = [GPBMessage message];
[message2 setUnknownFields:unknowns2];
NSMutableData *delimitedData = [NSMutableData data];
[delimitedData appendData:[message1 delimitedData]];
[delimitedData appendData:[message2 delimitedData]];
GPBCodedInputStream *input =
[GPBCodedInputStream streamWithData:delimitedData];
GPBMessage *message3 = [GPBMessage parseDelimitedFromCodedInputStream:input
extensionRegistry:nil
error:NULL];
GPBMessage *message4 = [GPBMessage parseDelimitedFromCodedInputStream:input
extensionRegistry:nil
error:NULL];
XCTAssertEqualObjects(message1, message3);
XCTAssertEqualObjects(message2, message4);
}
- (void)testDuplicateEnums {
XCTAssertEqual(TestEnumWithDupValue_Foo1, TestEnumWithDupValue_Foo2);
}
- (void)testWeirdDefaults {
ObjcWeirdDefaults *message = [ObjcWeirdDefaults message];
GPBDescriptor *descriptor = [[message class] descriptor];
GPBFieldDescriptor *fieldDesc = [descriptor fieldWithName:@"foo"];
XCTAssertNotNil(fieldDesc);
XCTAssertTrue(fieldDesc.hasDefaultValue);
XCTAssertFalse(message.hasFoo);
XCTAssertEqualObjects(message.foo, @"");
fieldDesc = [descriptor fieldWithName:@"bar"];
XCTAssertNotNil(fieldDesc);
XCTAssertTrue(fieldDesc.hasDefaultValue);
XCTAssertFalse(message.hasBar);
XCTAssertEqualObjects(message.bar, GPBEmptyNSData());
}
- (void)testEnumDescriptorFromExtensionDescriptor {
GPBExtensionDescriptor *extDescriptor =
[UnittestRoot optionalForeignEnumExtension];
XCTAssertEqual(extDescriptor.dataType, GPBDataTypeEnum);
GPBEnumDescriptor *enumDescriptor = extDescriptor.enumDescriptor;
GPBEnumDescriptor *expectedDescriptor = ForeignEnum_EnumDescriptor();
XCTAssertEqualObjects(enumDescriptor, expectedDescriptor);
}
- (void)testPropertyNaming {
// objectivec_helpers.cc has some special handing to get proper all caps
// for a few cases to meet objc developer expectations.
//
// This "test" confirms that the expected names are generated, otherwise the
// test itself will fail to compile.
ObjCPropertyNaming *msg = [ObjCPropertyNaming message];
// On their own, at the end, in the middle.
msg.URL = @"good";
msg.thumbnailURL = @"good";
msg.URLFoo = @"good";
msg.someURLBlah = @"good";
msg.HTTP = @"good";
msg.HTTPS = @"good";
// No caps since it was "urls".
[msg.urlsArray addObject:@"good"];
}
- (void)testEnumNaming {
// objectivec_helpers.cc has some interesting cases to deal with in
// EnumValueName/EnumValueShortName. Confirm that things generated as
// expected.
// This block just has to compile to confirm we got the expected types/names.
// The *_IsValidValue() calls are just there to keep the projects warnings
// flags happy by providing use of the variables/values.
Foo aFoo = Foo_SerializedSize;
Foo_IsValidValue(aFoo);
aFoo = Foo_Size;
Foo_IsValidValue(aFoo);
Category_Enum aCat = Category_Enum_Red;
Category_Enum_IsValidValue(aCat);
Time aTime = Time_Base;
Time_IsValidValue(aTime);
aTime = Time_SomethingElse;
Time_IsValidValue(aTime);
// This block confirms the names in the decriptors is what we wanted.
GPBEnumDescriptor *descriptor;
NSString *valueName;
descriptor = Foo_EnumDescriptor();
XCTAssertNotNil(descriptor);
XCTAssertEqualObjects(@"Foo", descriptor.name);
valueName = [descriptor enumNameForValue:Foo_SerializedSize];
XCTAssertEqualObjects(@"Foo_SerializedSize", valueName);
valueName = [descriptor enumNameForValue:Foo_Size];
XCTAssertEqualObjects(@"Foo_Size", valueName);
descriptor = Category_Enum_EnumDescriptor();
XCTAssertNotNil(descriptor);
XCTAssertEqualObjects(@"Category_Enum", descriptor.name);
valueName = [descriptor enumNameForValue:Category_Enum_Red];
XCTAssertEqualObjects(@"Category_Enum_Red", valueName);
descriptor = Time_EnumDescriptor();
XCTAssertNotNil(descriptor);
XCTAssertEqualObjects(@"Time", descriptor.name);
valueName = [descriptor enumNameForValue:Time_Base];
XCTAssertEqualObjects(@"Time_Base", valueName);
valueName = [descriptor enumNameForValue:Time_SomethingElse];
XCTAssertEqualObjects(@"Time_SomethingElse", valueName);
}
- (void)testNegativeEnums {
EnumTestMsg *msg = [EnumTestMsg message];
// Defaults
XCTAssertEqual(msg.foo, EnumTestMsg_MyEnum_Zero);
XCTAssertEqual(msg.bar, EnumTestMsg_MyEnum_One);
XCTAssertEqual(msg.baz, EnumTestMsg_MyEnum_NegOne);
// Bounce to wire and back.
NSData *data = [msg data];
XCTAssertNotNil(data);
EnumTestMsg *msgPrime = [EnumTestMsg parseFromData:data error:NULL];
XCTAssertEqualObjects(msgPrime, msg);
XCTAssertEqual(msgPrime.foo, EnumTestMsg_MyEnum_Zero);
XCTAssertEqual(msgPrime.bar, EnumTestMsg_MyEnum_One);
XCTAssertEqual(msgPrime.baz, EnumTestMsg_MyEnum_NegOne);
// Other values
msg.bar = EnumTestMsg_MyEnum_Two;
msg.baz = EnumTestMsg_MyEnum_NegTwo;
XCTAssertEqual(msg.bar, EnumTestMsg_MyEnum_Two);
XCTAssertEqual(msg.baz, EnumTestMsg_MyEnum_NegTwo);
// Bounce to wire and back.
data = [msg data];
XCTAssertNotNil(data);
msgPrime = [EnumTestMsg parseFromData:data error:NULL];
XCTAssertEqualObjects(msgPrime, msg);
XCTAssertEqual(msgPrime.foo, EnumTestMsg_MyEnum_Zero);
XCTAssertEqual(msgPrime.bar, EnumTestMsg_MyEnum_Two);
XCTAssertEqual(msgPrime.baz, EnumTestMsg_MyEnum_NegTwo);
// Repeated field (shouldn't ever be an issue since developer has to use the
// right GPBArray methods themselves).
msg.mumbleArray = [GPBEnumArray
arrayWithValidationFunction:EnumTestMsg_MyEnum_IsValidValue];
[msg.mumbleArray addValue:EnumTestMsg_MyEnum_Zero];
[msg.mumbleArray addValue:EnumTestMsg_MyEnum_One];
[msg.mumbleArray addValue:EnumTestMsg_MyEnum_Two];
[msg.mumbleArray addValue:EnumTestMsg_MyEnum_NegOne];
[msg.mumbleArray addValue:EnumTestMsg_MyEnum_NegTwo];
XCTAssertEqual([msg.mumbleArray valueAtIndex:0], EnumTestMsg_MyEnum_Zero);
XCTAssertEqual([msg.mumbleArray valueAtIndex:1], EnumTestMsg_MyEnum_One);
XCTAssertEqual([msg.mumbleArray valueAtIndex:2], EnumTestMsg_MyEnum_Two);
XCTAssertEqual([msg.mumbleArray valueAtIndex:3], EnumTestMsg_MyEnum_NegOne);
XCTAssertEqual([msg.mumbleArray valueAtIndex:4], EnumTestMsg_MyEnum_NegTwo);
// Bounce to wire and back.
data = [msg data];
XCTAssertNotNil(data);
msgPrime = [EnumTestMsg parseFromData:data error:NULL];
XCTAssertEqualObjects(msgPrime, msg);
XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:0],
EnumTestMsg_MyEnum_Zero);
XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:1], EnumTestMsg_MyEnum_One);
XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:2], EnumTestMsg_MyEnum_Two);
XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:3],
EnumTestMsg_MyEnum_NegOne);
XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:4],
EnumTestMsg_MyEnum_NegTwo);
}
- (void)testReservedWordNaming {
// objectivec_helpers.cc has some special handing to make sure that
// some "reserved" objc names get renamed in a way so they
// don't conflict.
//
// This "test" confirms that the expected names are generated,
// otherwise the test itself will fail to compile.
self_Class *msg = [self_Class message];
// Some ObjC/C/C++ keywords.
msg.className_p = msg.hasClassName_p;
msg.cmd = msg.hasCmd;
msg.nullable_p = msg.hasNullable_p;
msg.typeof_p = msg.hasTypeof_p;
msg.instancetype_p = msg.hasInstancetype_p;
msg.nil_p = msg.hasNil_p;
msg.instancetype_p = msg.hasInstancetype_p;
msg.public_p = msg.hasPublic_p;
// Some that would override NSObject methods
msg.camltype = msg.hasCamltype;
msg.isNsdictionary = msg.hasIsNsdictionary;
msg.dealloc_p = msg.hasDealloc_p;
msg.zone_p = msg.hasZone_p;
msg.accessibilityLabel_p = msg.hasAccessibilityLabel_p;
// Some that we shouldn't need to handle.
msg.atomic = msg.hasAtomic;
msg.nonatomic = msg.hasNonatomic;
msg.strong = msg.hasStrong;
msg.nullResettable = msg.hasNullResettable;
// Some that would override GPBMessage methods
msg.clear_p = msg.hasClear_p;
msg.data_p = msg.hasData_p;
// Some MacTypes
msg.fixed = msg.hasFixed;
msg.style = msg.hasStyle;
// Some C Identifiers
msg.generic = msg.hasGeneric;
msg.block = msg.hasBlock;
}
- (void)testOneBasedEnumHolder {
// Test case for https://github.com/protocolbuffers/protobuf/issues/1453
// Message with no explicit defaults, but a non zero default for an enum.
MessageWithOneBasedEnum *enumMsg = [MessageWithOneBasedEnum message];
XCTAssertEqual(enumMsg.enumField, MessageWithOneBasedEnum_OneBasedEnum_One);
}
- (void)testBoolOffsetUsage {
// Bools use storage within has_bits; this test ensures that this is honored
// in all places where things should crash or fail based on reading out of
// field storage instead.
BoolOnlyMessage *msg1 = [BoolOnlyMessage message];
BoolOnlyMessage *msg2 = [BoolOnlyMessage message];
msg1.boolField1 = YES;
msg2.boolField1 = YES;
msg1.boolField3 = YES;
msg2.boolField3 = YES;
msg1.boolField5 = YES;
msg2.boolField5 = YES;
msg1.boolField7 = YES;
msg2.boolField7 = YES;
msg1.boolField9 = YES;
msg2.boolField9 = YES;
msg1.boolField11 = YES;
msg2.boolField11 = YES;
msg1.boolField13 = YES;
msg2.boolField13 = YES;
msg1.boolField15 = YES;
msg2.boolField15 = YES;
msg1.boolField17 = YES;
msg2.boolField17 = YES;
msg1.boolField19 = YES;
msg2.boolField19 = YES;
msg1.boolField21 = YES;
msg2.boolField21 = YES;
msg1.boolField23 = YES;
msg2.boolField23 = YES;
msg1.boolField25 = YES;
msg2.boolField25 = YES;
msg1.boolField27 = YES;
msg2.boolField27 = YES;
msg1.boolField29 = YES;
msg2.boolField29 = YES;
msg1.boolField31 = YES;
msg2.boolField31 = YES;
msg1.boolField32 = YES;
msg2.boolField32 = YES;
XCTAssertTrue(msg1 != msg2); // Different pointers.
XCTAssertEqual([msg1 hash], [msg2 hash]);
XCTAssertEqualObjects(msg1, msg2);
BoolOnlyMessage *msg1Prime = [[msg1 copy] autorelease];
XCTAssertTrue(msg1Prime != msg1); // Different pointers.
XCTAssertEqual([msg1 hash], [msg1Prime hash]);
XCTAssertEqualObjects(msg1, msg1Prime);
// Field set in one, but not the other means they don't match (even if
// set to default value).
msg1Prime.boolField2 = NO;
XCTAssertNotEqualObjects(msg1Prime, msg1);
// And when set to different values.
msg1.boolField2 = YES;
XCTAssertNotEqualObjects(msg1Prime, msg1);
// And then they match again.
msg1.boolField2 = NO;
XCTAssertEqualObjects(msg1Prime, msg1);
XCTAssertEqual([msg1 hash], [msg1Prime hash]);
}
- (void)testCopyingMapFields {
TestMessageOfMaps *msg = [TestMessageOfMaps message];
msg.strToStr[@"foo"] = @"bar";
[msg.strToInt setInt32:1 forKey:@"mumble"];
[msg.intToStr setObject:@"wee" forKey:42];
[msg.intToInt setInt32:123 forKey:321];
[msg.strToBool setBool:YES forKey:@"one"];
[msg.boolToStr setObject:@"something" forKey:YES];
[msg.boolToBool setBool:YES forKey:NO];
[msg.intToBool setBool:YES forKey:13];
[msg.boolToInt setInt32:111 forKey:NO];
TestAllTypes *subMsg1 = [TestAllTypes message];
subMsg1.optionalInt32 = 1;
TestAllTypes *subMsg2 = [TestAllTypes message];
subMsg1.optionalInt32 = 2;
TestAllTypes *subMsg3 = [TestAllTypes message];
subMsg1.optionalInt32 = 3;
msg.strToMsg[@"baz"] = subMsg1;
[msg.intToMsg setObject:subMsg2 forKey:222];
[msg.boolToMsg setObject:subMsg3 forKey:YES];
TestMessageOfMaps *msg2 = [[msg copy] autorelease];
XCTAssertNotNil(msg2);
XCTAssertEqualObjects(msg2, msg);
XCTAssertTrue(msg2 != msg); // ptr compare
XCTAssertTrue(msg.strToStr != msg2.strToStr); // ptr compare
XCTAssertTrue(msg.intToStr != msg2.intToStr); // ptr compare
XCTAssertTrue(msg.intToInt != msg2.intToInt); // ptr compare
XCTAssertTrue(msg.strToBool != msg2.strToBool); // ptr compare
XCTAssertTrue(msg.boolToStr != msg2.boolToStr); // ptr compare
XCTAssertTrue(msg.boolToBool != msg2.boolToBool); // ptr compare
XCTAssertTrue(msg.intToBool != msg2.intToBool); // ptr compare
XCTAssertTrue(msg.boolToInt != msg2.boolToInt); // ptr compare
XCTAssertTrue(msg.strToMsg != msg2.strToMsg); // ptr compare
XCTAssertTrue(msg.intToMsg != msg2.intToMsg); // ptr compare
XCTAssertTrue(msg.boolToMsg != msg2.boolToMsg); // ptr compare
XCTAssertTrue(msg.strToMsg[@"baz"] != msg2.strToMsg[@"baz"]); // ptr compare
XCTAssertEqualObjects(msg.strToMsg[@"baz"], msg2.strToMsg[@"baz"]);
XCTAssertTrue([msg.intToMsg objectForKey:222] != [msg2.intToMsg objectForKey:222]); // ptr compare
XCTAssertEqualObjects([msg.intToMsg objectForKey:222], [msg2.intToMsg objectForKey:222]);
XCTAssertTrue([msg.boolToMsg objectForKey:YES] != [msg2.boolToMsg objectForKey:YES]); // ptr compare
XCTAssertEqualObjects([msg.boolToMsg objectForKey:YES], [msg2.boolToMsg objectForKey:YES]);
}
- (void)testPrefixedNames {
// The fact that this compiles is sufficient as a test.
// The assertions are just there to avoid "not-used" warnings.
// Verify that enum types and values get the prefix.
GPBTESTTestObjcProtoPrefixEnum value = GPBTESTTestObjcProtoPrefixEnum_Value;
XCTAssertNotEqual(value, 0);
// Verify that roots get the prefix.
GPBTESTUnittestObjcOptionsRoot *root = nil;
XCTAssertNil(root);
// Verify that messages that don't already have the prefix get a prefix.
GPBTESTTestObjcProtoPrefixMessage *prefixedMessage = nil;
XCTAssertNil(prefixedMessage);
// Verify that messages that already have a prefix aren't prefixed twice.
GPBTESTTestHasAPrefixMessage *alreadyPrefixedMessage = nil;
XCTAssertNil(alreadyPrefixedMessage);
// Verify that enums that already have a prefix aren't prefixed twice.
GPBTESTTestHasAPrefixEnum prefixedValue = GPBTESTTestHasAPrefixEnum_ValueB;
XCTAssertNotEqual(prefixedValue, 0);
// Verify that classes named the same as prefixes are prefixed.
GPBTESTGPBTEST *prefixMessage = nil;
XCTAssertNil(prefixMessage);
// Verify that classes that have the prefix followed by a lowercase
// letter DO get the prefix.
GPBTESTGPBTESTshouldGetAPrefixMessage *shouldGetAPrefixMessage = nil;
XCTAssertNil(shouldGetAPrefixMessage);
}
@end