iOS中异步回调的单元测试方法

在XCode6以上版本中，苹果添加了用于异步回调测试的api，因此不用像旧版本那样，发起异步调用后通过循环查询标志位，来检查异步回调函数的调用了。

在新版本中直接使用

XCTestExpectation

的API即可实现这一功能。

首先来看一下官方文档中的代码片段：

- (void)testDocumentOpening
{
// 创建一个expectation对象
XCTestExpectation *documentOpenExpectation = [self expectationWithDescription:@"document open"];

NSURL *URL = [[NSBundle bundleForClass:[self class]]
URLForResource:@"TestDocument" withExtension:@"mydoc"];
UIDocument *doc = [[UIDocument alloc] initWithFileURL:URL];
[doc openWithCompletionHandler:^(BOOL success) {
XCTAssert(success);

// assert成功后 便会调用expectation的fulfill方法，来触发下面的handler
[documentOpenExpectation fulfill];
}];

// 在case最后使用这一方法，此时单测程序会阻塞到这里；除非达到了超时时间（秒单位）或者是回调函数中调用了fulfill，单测程序才会结束
// 若是超时情况，则认为case失败；若通过expectation的fulfill触发，则case通过
[self waitForExpectationsWithTimeout:1 handler:^(NSError *error) {
[doc closeWithCompletionHandler:nil];
}];
}

步骤就是：

在单测开始位置声明需要使用的Expectation对象，在回调中触发

fulfill

函数，单测的末尾调用api进行等待。

官方文档使用的block方式回调。对于代理方式的回调，同样适用。不过由于回调函数在单测函数外侧，需要把变量声明到类中，这里通过

NSURLConnection

异步GET回调来说明：

#import <XCTest/XCTest.h>

@interface HelloWorldXCodeTests : XCTestCase

@end

@implementation HelloWorldXCodeTests

// 回调对应的Expectation对象
XCTestExpectation *networkSuccessExpectation;

- (void)setUp {
[super setUp];
}

- (void)tearDown {
[super tearDown];
}

- (void)testNetworkOpening
{
networkSuccessExpectation = [self expectationWithDescription:@"document open"];

// 创建url
NSURL *url = [NSURL URLWithString:@"https://api.douban.com/v2/book/1220562"];
// 创建请求
NSURLRequest *request = [[NSURLRequest alloc]initWithURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy timeoutInterval:10];
// 连接服务器
NSURLConnection *connection = [[NSURLConnection alloc] initWithRequest:request delegate:self];
[connection start];

// 等待回调方法，10秒超时
[self waitForExpectationsWithTimeout:10 handler:^(NSError *error) {
NSLog(@"test case over");
}];

}

// 异步GET请求回调
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response
{
NSHTTPURLResponse *res = (NSHTTPURLResponse *)response;
NSLog(@"http connection: %@",[res allHeaderFields]);
// 触发fulfill方法，跳出单测程序等待状态
[networkSuccessExpectation fulfill];
}

@end

如果网络不通，那么10s超时后就会导致case失败；反之若成功请求则会在回调函数中打印出返回头，并触发单测成功。