OCMock一个Class方法用自定义数据回调传递的块

时间:2013-12-19 21:51:51

标签: objective-c unit-testing mocking ocmock

我有一个类来处理我的所有网络通信:

typedef void (^ networkEndblock) (NSArray *, NSError *);

@interface NetworkAPI : NetworkBase

+ (void) doSomeNetworkAcion:(networkEndblock) endBlock;

@end

我使用上面这样的代码(我不想在这里讨论不相关的细节)

- (void) runMyProcess:(SomeEndBlock)proccesEnded{

  // Bussiness logic


  // Get Data from the web
  [NetworkAPI doSomeNetworkAcion:^(NSArray *resultArray, NSError *error){

   // Check for error. if we have error then do error handling
   // If no error, check for array.
   // If array is empty then go to code that handles empty array
   // else Process data                                 

   }];
}

在我的测试方法中,我想触发runMyProcess进行测试,我不希望它进入网络,我想控制它并设置案例以使其返回错误,空数组......等等。我知道如何使用SenTest,它是MACROS,但我不知道如何伪造我的网络API。

我查看存根并期待,但如果我能做我想做的事,我会感到困惑。

由于

1 个答案:

答案 0 :(得分:5)

类模拟[NetworkAPI doSomeNetworkAction:]并使用andDo块。

// This is in the OCMock project but is really useful
#import "NSInvocation+OCMAdditions.h"

// Whatever you want for these values
NSArray *fakeResultArray;
NSError *fakeError;

id networkAPIMock = [OCMockObject mockForClass:NetworkAPI.class];
[[[networkAPIMock expect] andDo:^(NSInvocation *invocation) {
    networkEndBlock endBlock = [invocation getArgumentAtIndexAsObject:2];
    endBlock(fakeResultArray, fakeError);
}] doSomeNetworkAction:OCMOCK_ANY];

另外,我会在我的typedef中大写NetworkEndBlock,因为它会让你的代码更易于阅读。