| Index: third_party/ocmock/OCMock/OCProtocolMockObject.m
|
| diff --git a/third_party/ocmock/OCMock/OCProtocolMockObject.m b/third_party/ocmock/OCMock/OCProtocolMockObject.m
|
| index 19e25e8796801126d4bb61b8dffbd86a168ee73c..cd43a0cb8fb9cd3fe724da4a6b86591e90cf62e2 100644
|
| --- a/third_party/ocmock/OCMock/OCProtocolMockObject.m
|
| +++ b/third_party/ocmock/OCMock/OCProtocolMockObject.m
|
| @@ -1,7 +1,18 @@
|
| -//---------------------------------------------------------------------------------------
|
| -// $Id$
|
| -// Copyright (c) 2005-2008 by Mulle Kybernetik. See License file for details.
|
| -//---------------------------------------------------------------------------------------
|
| +/*
|
| + * Copyright (c) 2005-2015 Erik Doernenburg and contributors
|
| + *
|
| + * Licensed under the Apache License, Version 2.0 (the "License"); you may
|
| + * not use these files except in compliance with the License. You may obtain
|
| + * a copy of the License at
|
| + *
|
| + * http://www.apache.org/licenses/LICENSE-2.0
|
| + *
|
| + * Unless required by applicable law or agreed to in writing, software
|
| + * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
| + * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
| + * License for the specific language governing permissions and limitations
|
| + * under the License.
|
| + */
|
|
|
| #import <objc/runtime.h>
|
| #import "NSMethodSignature+OCMAdditions.h"
|
| @@ -13,31 +24,34 @@
|
|
|
| - (id)initWithProtocol:(Protocol *)aProtocol
|
| {
|
| - [super init];
|
| - mockedProtocol = aProtocol;
|
| - return self;
|
| + NSParameterAssert(aProtocol != nil);
|
| + [super init];
|
| + mockedProtocol = aProtocol;
|
| + return self;
|
| }
|
|
|
| - (NSString *)description
|
| {
|
| const char* name = protocol_getName(mockedProtocol);
|
| - return [NSString stringWithFormat:@"OCMockObject[%s]", name];
|
| + return [NSString stringWithFormat:@"OCMockObject(%s)", name];
|
| }
|
|
|
| #pragma mark Proxy API
|
|
|
| - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
|
| {
|
| - struct objc_method_description methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, YES, YES);
|
| - if(methodDescription.name == NULL)
|
| - {
|
| - methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, NO, YES);
|
| + struct {
|
| + BOOL isRequired;
|
| + BOOL isInstance;
|
| + } opts[4] = {{YES, YES}, {NO, YES}, {YES, NO}, {NO, NO}};
|
| + for (int i = 0; i < 4; i++) {
|
| + struct objc_method_description methodDescription =
|
| + protocol_getMethodDescription(mockedProtocol, aSelector,
|
| + opts[i].isRequired, opts[i].isInstance);
|
| + if (methodDescription.name != NULL)
|
| + return [NSMethodSignature signatureWithObjCTypes:methodDescription.types];
|
| }
|
| - if(methodDescription.name == NULL)
|
| - {
|
| - return nil;
|
| - }
|
| - return [NSMethodSignature signatureWithObjCTypes:methodDescription.types];
|
| + return nil;
|
| }
|
|
|
| - (BOOL)conformsToProtocol:(Protocol *)aProtocol
|
|
|