如何测试块是否被调用

Question

我有一个方法的单元测试，该方法应该调用完成块或失败块。现在我知道每种情况应该调用哪一个，因此我在不应调用的块中使用 STFail 。

我现在如何测试应该调用的块是否确实被调用？

这是我的设置：

NSString *parameter = @"foo";
[controller doSomethingWithParameter:parameter withcompletionBlock: 
^(NSString *result)
{
    // This block should be invoked
    // Check if the result is correct
    STAssertEquals(result, kSomeOKConstant, @"Result shout be 'kSomeOKConstant'");
} failedBlock:
^(NSString *errorMessage) {
    STFail(@"No error should happen with parameter '%@'",parameter);
}];

Answer 1

您需要添加块变量，并从块内部设置它们：

BOOL __block successBlockInvoked = NO;
BOOL __block failureBlockInvoked = NO;
NSString *parameter = @"foo";
[controller doSomethingWithParameter:parameter withcompletionBlock: 
^(NSString *result) {
     successBlockInvoked = YES;
     STAssertEquals(result, kSomeOKConstant, @"Result shout be 'kSomeOKConstant'");
} failedBlock:
^(NSString *errorMessage) {
    failureBlockInvoked = YES;
    STFail(@"No error should happen with parameter '%@'",parameter);
}];

此时，您可以对 successBlockInvoked 和 failureBlockInvoked 的值进行断言：如果预期的值不是设置，你的测试失败了。