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

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 release 2.1开始，我们可以存根类方法。更多信息请参考此链接:http://www.ocmock.org/features/

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

id keyChainUtilMock = [OCMockObject mockForClass:[KeyChainUtil class]];
[[[keyChainUtilMock stub] andReturn:@"aasdf"] loadValueForKey:USER_ID];
NSString* userId = [KeyChainUtil loadValueForKey:USER_ID];
NSLog(@" stubbed value-->%@", userId);

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