1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef THIRD_PARTY_OCMOCK_OCMOCK_EXTENSIONS_H_ 6 #define THIRD_PARTY_OCMOCK_OCMOCK_EXTENSIONS_H_ 7 8 #import <Foundation/Foundation.h> 9 10 #import "third_party/ocmock/OCMock/OCMock.h" 11 12 // Some enhancements to OCMock to make it easier to write mocks. 13 // Pointers to objects still have to be handled with 14 // - (id)andReturnValue:OCMOCK_VALUE(blah) 15 // to keep the types working correctly. 16 @interface OCMockRecorder(CrExtensions) 17 - (id)andReturnChar:(char)value; 18 - (id)andReturnUnsignedChar:(unsigned char)value; 19 - (id)andReturnShort:(short)value; 20 - (id)andReturnUnsignedShort:(unsigned short)value; 21 - (id)andReturnInt:(int)value; 22 - (id)andReturnUnsignedInt:(unsigned int)value; 23 - (id)andReturnLong:(long)value; 24 - (id)andReturnUnsignedLong:(unsigned long)value; 25 - (id)andReturnLongLong:(long long)value; 26 - (id)andReturnUnsignedLongLong:(unsigned long long)value; 27 - (id)andReturnFloat:(float)value; 28 - (id)andReturnDouble:(double)value; 29 - (id)andReturnBool:(BOOL)value; 30 - (id)andReturnInteger:(NSInteger)value; 31 - (id)andReturnUnsignedInteger:(NSUInteger)value; 32 #if !TARGET_OS_IPHONE 33 - (id)andReturnCGFloat:(CGFloat)value; 34 - (id)andReturnNSRect:(NSRect)rect; 35 - (id)andReturnCGRect:(CGRect)rect; 36 - (id)andReturnNSPoint:(NSPoint)point; 37 - (id)andReturnCGPoint:(CGPoint)point; 38 #endif 39 @end 40 41 // A constraint for verifying that something conforms to a protocol. 42 @interface cr_OCMConformToProtocolConstraint : OCMConstraint { 43 @private 44 Protocol* protocol_; 45 } 46 - (id)initWithProtocol:(Protocol*)protocol; 47 @end 48 49 @interface OCMArg(CrExtensions) 50 + (id)conformsToProtocol:(Protocol*)protocol; 51 @end 52 53 #endif // THIRD_PARTY_OCMOCK_OCMOCK_EXTENSIONS_H_ 54
