1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
-
-
- #import <OCMock/OCMRecorder.h>
- #import <OCMock/OCMFunctions.h>
- #import <objc/runtime.h>
-
- @interface OCMStubRecorder : OCMRecorder
-
- - (id)andReturn:(id)anObject;
- - (id)andReturnValue:(NSValue *)aValue;
- - (id)andThrow:(NSException *)anException;
- - (id)andPost:(NSNotification *)aNotification;
- - (id)andCall:(SEL)selector onObject:(id)anObject;
- - (id)andDo:(void (^)(NSInvocation *invocation))block;
- - (id)andForwardToRealObject;
-
- @end
-
-
- @interface OCMStubRecorder (Properties)
-
- #define andReturn(aValue) _andReturn(({ \
- __typeof__(aValue) _val = (aValue); \
- NSValue *_nsval = [NSValue value:&_val withObjCType:@encode(__typeof__(_val))]; \
- if (OCMIsObjectType(@encode(__typeof(_val)))) { \
- objc_setAssociatedObject(_nsval, "OCMAssociatedBoxedValue", *(__unsafe_unretained id *) (void *) &_val, OBJC_ASSOCIATION_RETAIN); \
- } \
- _nsval; \
- }))
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andReturn)(NSValue *);
-
- #define andThrow(anException) _andThrow(anException)
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andThrow)(NSException *);
-
- #define andPost(aNotification) _andPost(aNotification)
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andPost)(NSNotification *);
-
- #define andCall(anObject, aSelector) _andCall(anObject, aSelector)
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andCall)(id, SEL);
-
- #define andDo(aBlock) _andDo(aBlock)
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andDo)(void (^)(NSInvocation *));
-
- #define andForwardToRealObject() _andForwardToRealObject()
- @property (nonatomic, readonly) OCMStubRecorder *(^ _andForwardToRealObject)(void);
-
- @end
-
-
-
|