third_party/ocmock/OCMock/OCPartialMockRecorder.m

   1 //---------------------------------------------------------------------------------------
   2 //  $Id$
   3 //  Copyright (c) 2009-2010 by Mulle Kybernetik. See License file for details.
   4 //---------------------------------------------------------------------------------------
   5
   6 #import "OCPartialMockObject.h"
   7 #import "OCMRealObjectForwarder.h"
   8 #import "OCPartialMockRecorder.h"
   9
  10
  11 @implementation OCPartialMockRecorder
  12
  13 - (id)andForwardToRealObject
  14 {
  15         [invocationHandlers addObject:[[[OCMRealObjectForwarder alloc] init] autorelease]];
  16         return self;
  17 }
  18
  19
  20 - (void)forwardInvocation:(NSInvocation *)anInvocation
  21 {
  22         [super forwardInvocation:anInvocation];
  23         // not as clean as I'd wish...
  24         [(OCPartialMockObject *)signatureResolver setupForwarderForSelector:[anInvocation selector]];
  25 }
  26
  27 @end