两块完成的OCMock存根方法

我试图模拟一个有两块的方法的类（AFHTTPSessionManager），我怎样才能手动触发它们，例如，做两个测试，一个将使用第一个块实现和另一个测试将使用第二个块实现。两块完成的OCMock存根方法

[manager GET:path parameters:parameters success:^(NSURLSessionDataTask *task, id responseObject) { 
    //First block to trigger. 
    //logic code here 
} failure:^(NSURLSessionDataTask *task, NSError *error) { 
    //Second block to trigger. 
    //logic code here 
}];

目前，它会一直触发第二个块。

我嘲笑类是这样的：

NSError *error = [NSError errorWithDomain:NSLocalizedFailureReasonErrorKey code:-1009 userInfo:nil]; 
NSDictionary *responseObject = [NSDictionary dictionaryWithObjectsAndKeys:@"object", @"key", nil]; 

id mockAFHTTPSessionManager = OCMClassMock([AFHTTPSessionManager class]); 

OCMStub([mockAFHTTPSessionManager alloc]).andReturn(mockAFHTTPSessionManager); 
OCMStub([mockAFHTTPSessionManager initWithBaseURL:OCMOCK_ANY]).andReturn(mockAFHTTPSessionManager); 
OCMStub([mockAFHTTPSessionManager GET:OCMOCK_ANY parameters:OCMOCK_ANY success:([OCMArg invokeBlockWithArgs:OCMOCK_ANY, responseObject, nil]) failure:([OCMArg invokeBlockWithArgs:OCMOCK_ANY, error, nil])]);

来源

2017-02-28 Scar

我不知道是否有你磕碰分配/ initWithBaseURL一个具体的理由，但在这里是如何我模拟AFHTTPSessionManager：

id mockSessionManager = OCMClassMock([AFHTTPSessionManager class]); 
OCMStub([mockSessionManager manager]).andReturn(mockSessionManager);

现在，为了测试成功/失败块的调用，我将提出一个建议，为我节省了一堆难题：不要内嵌块。相反，从单独的方法返回块并单独测试这些方法。然后，在测试实际的GET:parameters:success:failure调用的测试中，您只需要期望返回完成块的方法被调用。

下面是一个例子：

MyClass.m

- (void)doGetRequest 
{ 
    AFHTTPSessionManager* manager = [AFHTTPSessionManager manager]; 
    [manager GET:@"http://something.whatever.com" 
      parameters:nil 
       success:[self getSuccessBlock] 
       failure:[self getFailureBlock]]; 
} 

- (void (^)(NSURLSessionDataTask*, id))getSuccessBlock 
{ 
    return ^(NSURLSessionDataTask* task, id responseObject) 
     { 
      // handle success 
     }; 
} 

- (void (^)(NSURLSessionDataTask*, NSError*))getFailure 
{ 
    return ^(NSURLSessionDataTask* task, NSError* error) 
     { 
      // handle failure 
     } 
}

MyClassTests.m

- (void)testDoGetRequest 
{ 
    MyClass* myClass = [[MyClass alloc] init]; 
    id mockMyClass = OCMPartialMock(myClass); 
    id mockSessionManager = OCMClassMock([AFHTTPSessionManager class]); 
    OCMStub([mockSessionManager manager]).andReturn(mockSessionManager); 

    id successBlock = ^(NSURLSessionDataTask* task, id o){}; 
    id failureBlock = ^(NSURLSessionDataTask* task, NSError* error){}; 
    OCMExpect([mockMyClass getSuccess]).andReturn(successBlock); 
    OCMExpect([mockMyClass getFailure]).andReturn(failureBlock); 

    OCMExpect([mockSessionManager GET:OCMOCK_ANY 
          parameters:OCMOCK_ANY 
           success:successBlock 
           failure:failureBlock]); 

    [myClass doGetRequest]; 

    OCMVerifyAll(mockMyClass); 
    OCMVerifyAll(mockSessionManager); 
} 

- (void)testGetSuccessBlock 
{ 
    MyClass* myClass = [[MyClass alloc] init]; 
    // setup expectations 
    NSURLSessionDataTaks* fakeTask = // whatever you need here 
    id fakeResponseObject = // whatever you need here 
    [myClass getSuccessBlock](fakeTask, fakeResponseObject); 
    // evaluate expectations 
} 

- (void)testGetFailureBlock 
{ 
    MyClass* myClass = [[MyClass alloc] init]; 
    // setup expectations 
    NSURLSessionDataTaks* fakeTask = // whatever you need here 
    NSError* fakeError = // whatever you need here 
    [myClass getFailureBlock](fakeTask, fakeError); 
    // evaluate expectations 
}

（我写这基本上把我的头的顶部，因此某些的语法可能不是100％，但你应该明白）

来源

2017-03-01 18:09:24

两块完成的OCMock存根方法

回答

相关问题