| //--------------------------------------------------------------------------------------- |
| // $Id$ |
| // Copyright (c) 2009-2010 by Mulle Kybernetik. See License file for details. |
| //--------------------------------------------------------------------------------------- |
| |
| #import "OCPartialMockObject.h" |
| #import "OCMRealObjectForwarder.h" |
| #import "OCPartialMockRecorder.h" |
| |
| |
| @implementation OCPartialMockRecorder |
| |
| - (id)andForwardToRealObject |
| { |
| [invocationHandlers addObject:[[[OCMRealObjectForwarder alloc] init] autorelease]]; |
| return self; |
| } |
| |
| |
| - (void)forwardInvocation:(NSInvocation *)anInvocation |
| { |
| [super forwardInvocation:anInvocation]; |
| // not as clean as I'd wish... |
| [(OCPartialMockObject *)signatureResolver setupForwarderForSelector:[anInvocation selector]]; |
| } |
| |
| @end |