#import "GPBTestUtilities.h"
#import "GPBExtensionRegistry.h"
#import "objectivec/Tests/Unittest.pbobjc.h"
@interface GPBExtensionRegistryTest : GPBTestCase
@end
@implementation GPBExtensionRegistryTest
- (void)testBasics {
GPBExtensionRegistry *reg = [[[GPBExtensionRegistry alloc] init] autorelease];
XCTAssertNotNil(reg);
XCTAssertNil([reg extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1]);
XCTAssertNil([reg extensionForDescriptor:[TestAllTypes descriptor] fieldNumber:1]);
[reg addExtension:[UnittestRoot optionalInt32Extension]];
[reg addExtension:[UnittestRoot packedInt64Extension]];
XCTAssertTrue([reg extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1] ==
[UnittestRoot optionalInt32Extension]); XCTAssertNil([reg extensionForDescriptor:[TestAllTypes descriptor] fieldNumber:1]);
XCTAssertTrue([reg extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:91] ==
[UnittestRoot packedInt64Extension]); }
- (void)testCopy {
GPBExtensionRegistry *reg1 = [[[GPBExtensionRegistry alloc] init] autorelease];
[reg1 addExtension:[UnittestRoot optionalInt32Extension]];
GPBExtensionRegistry *reg2 = [[reg1 copy] autorelease];
XCTAssertNotNil(reg2);
XCTAssertTrue([reg1 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1] ==
[UnittestRoot optionalInt32Extension]); XCTAssertTrue([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1] ==
[UnittestRoot optionalInt32Extension]);
[reg1 addExtension:[UnittestRoot optionalBoolExtension]];
[reg2 addExtension:[UnittestRoot optionalStringExtension]];
XCTAssertTrue([reg1 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:13] ==
[UnittestRoot optionalBoolExtension]); XCTAssertNil([reg1 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:14]);
XCTAssertNil([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:13]);
XCTAssertTrue([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:14] ==
[UnittestRoot optionalStringExtension]);
[reg1 addExtension:[UnittestRoot packedInt64Extension]];
[reg2 addExtension:[UnittestRoot packedSint32Extension]];
XCTAssertTrue([reg1 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:91] ==
[UnittestRoot packedInt64Extension]); XCTAssertNil([reg1 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:94]);
XCTAssertNil([reg2 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:91]);
XCTAssertTrue([reg2 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:94] ==
[UnittestRoot packedSint32Extension]); }
- (void)testAddExtensions {
GPBExtensionRegistry *reg1 = [[[GPBExtensionRegistry alloc] init] autorelease];
[reg1 addExtension:[UnittestRoot optionalInt32Extension]];
GPBExtensionRegistry *reg2 = [[[GPBExtensionRegistry alloc] init] autorelease];
XCTAssertNil([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1]);
[reg2 addExtensions:reg1];
XCTAssertTrue([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:1] ==
[UnittestRoot optionalInt32Extension]);
[reg1 addExtension:[UnittestRoot optionalBoolExtension]];
[reg1 addExtension:[UnittestRoot packedInt64Extension]];
XCTAssertTrue([reg1 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:13] ==
[UnittestRoot optionalBoolExtension]); XCTAssertTrue([reg1 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:91] ==
[UnittestRoot packedInt64Extension]); XCTAssertNil([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:13]);
XCTAssertNil([reg2 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:91]);
[reg2 addExtension:[UnittestRoot optionalStringExtension]];
[reg2 addExtension:[UnittestRoot packedSint32Extension]];
XCTAssertNil([reg1 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:14]);
XCTAssertNil([reg1 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:94]);
XCTAssertTrue([reg2 extensionForDescriptor:[TestAllExtensions descriptor] fieldNumber:14] ==
[UnittestRoot optionalStringExtension]); XCTAssertTrue([reg2 extensionForDescriptor:[TestPackedExtensions descriptor] fieldNumber:94] ==
[UnittestRoot packedSint32Extension]); }
@end