兩塊完成的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存根方法

回答

相關問題