objective-c - 模拟另一个类方法的内部调用

Question

我正在尝试使用ocmock学习单元测试。我发现很难从我正在进行单元测试的类中模拟另一个类的调用。

有人可以建议如何模拟调用 KeyChainUtils 类和 HttpRequest 类吗:

使用 OCMock 进行单元测试的代码:

@implementation UserProfileService {
+(BOOL) isValidUser
{
    NSString* userId = [KeyChainUtil loadValueForKey:USER_ID]; //mock this call
    bool isValidUser = NO;
    if(userId && userId.length > 0){
        NSDictionary* response = [HTTPDataService getJSONForURL:@"http://xtest.com/checkuserid" forRequestData:@{@"userid": userId}];

        if(response && response[@"valid"]){
            isValidUser = [response[@"valid"] boolValue];             
        }else{
            NSLog(@"error in connecting to server. response => %@", response);
        }
    }
    return isValidUser;
 }
}

Answer 1

从 OCMock 2.1 版开始，我们可以对类方法进行 stub 处理。请参阅此链接了解更多信息:http://www.ocmock.org/features/

因此，我们可以像这样 stub 类方法:

id keyChainUtilMock = [OCMockObject mockForClass:[KeyChainUtil class]];
[[[keyChainUtilMock stub] andReturn:@"aasdf"] loadValueForKey:USER_ID];

NSString* userId = [KeyChainUtil loadValueForKey:USER_ID];
NSLog(@" stubbed value-->%@", userId);

所以，在运行这段特定的代码之后。这里不调用实际的类方法，而是返回 stub 值。我希望这对您有帮助。