纯工具类APP已经沦为炮灰，移动APP几乎都是基于网络的，因此写单元测试，网络是一个绕不开的话题。实际iOS开发中，凡是基于http的网络连接， AFNetworking 几乎已成为一个标准库，比如发起一个post请求，会创建AFHTTPSessionManager 对象：

 AFHTTPSessionManager *manager = [AFHTTPSessionManager manager];
    manager.responseSerializer.acceptableContentTypes = [NSSet setWithObjects:@"text/plain",@"text/html",nil];

并调用以下方法：

- (NSURLSessionDataTask *)POST:(NSString *)URLString
                    parameters:(id)parameters
                      progress:(void (^)(NSProgress * _Nonnull))uploadProgress
                       success:(void (^)(NSURLSessionDataTask * _Nonnull, id _Nullable))success
                       failure:(void (^)(NSURLSessionDataTask * _Nullable, NSError * _Nonnull))failure

网络接口为什么难测？

还记得《没有单元测试，何谈重构》 中的测试循环吗？让我们再来复习一下：

测试循环

当我们为AFHTTPSessionManager编写单元测试时，会遇到以下问题：

1. 返回结果不可控

虽然我们可以准备我们想要的目标URLString、parameters等参数，但在实际网络中，progress、success、failure等block回调——也就是测试循环的“验证结果”部分——是不确定的。
即使能勉强控制测试的网络环境，保证服务器一定返回success，定制返回数据也困难重重：准备把目标返回数据都写在服务端？基于真实的网络，很难获得理想的数据。

鬼知道服务器会返回给你什么！
——某开发者云

2. 可能还有经济上的问题

假设要开发的是对接获取验证码接口的方法，难道运行一次就真的请求一个短信验证码？短信下发平台可是会¥扣钱¥的。更何况，万一第三方平台没有响应或超时，我们的测试就失败了，这种异步的、不确定的测试，无论从金钱还是时间上衡量，都不够经济，因此很难实现。

因此，我们需要有方法指定返回数据，并且不需要 实际访问网络，我们这里选择使用OCMock。

使用OCMock

OCMock是一个用于建立仿造对象的框架，它使用OC的运行时机制，可以自动的创建任何OC对象实例, 比如可以这样来仿制 AFHTTPSessionManager 对象：

id mockManager = [OCMockObject mockForClass:[AFHTTPSessionManager class]];

准备数据

接着我们就可以使用mockManager提供的 andDo 方法，“伪造”出一个success调用：

[[[mockManager expect] andDo:^(NSInvocation *invocation) {
        void (^successBlock)(NSURLSessionDataTask *task, id responseObject) = nil;
        [invocation getArgument:&successBlock atIndex:5];
        successBlock([[NSURLSessionDataTask alloc] init],
                     @{@"keyTest":@"valueTest"}
                     );
    }] POST:[OCMArg any]
     parameters:nil
     progress:[OCMArg any]
     success:[OCMArg any]
     failure:[OCMArg any]];

调用expect 方法用来告诉OCMock “接管” POST方法，附加的 andDo block方法中，提供了 NSInvocation 对象，这使得我们有机会指定调用的block和返回值：

1. 先声明 successBlock ，注意，格式要与调用的block声明一致；
2. 通过 getArgument 指定调用success方法，atIndex用于指明希望调用参数的位置，这里要从2开始，前两位分别是self（target）,selector(_cmd)）；
3. 调用successBlock，传入希望返回的数据。

注意：[OCMArg any] 用于告诉OCMock，可以接受任何参数，反之，如果expert时，POST参数传@"aTestPath"，调用时任何非@"aTestPath"的POST调用将被忽略。

调用接口

[mockManager POST:@"any path"
            parameters:nil
              progress:nil
               success:^(NSURLSessionDataTask *task, id responseObject) {
                   XCTAssertEqual(@"valueTest", responseObject[@"keyTest"]);
               } failure:^(NSURLSessionDataTask * _Nullable task,
                           NSError * _Nonnull error) {
               }];

这时，success block将会回调，并传回我们在andDo中指明的字典数据@{@"keyTest":@"valueTest"}

到此，我们已经有了一个测试网络应用的良好开端，基于这里介绍的技术，我们可以改造iOS开发中的网络接口，使之能更方便编写单元测试，欢迎关注溪石iOS，留言、私信分享你的感想和经验，共同探讨iOS单元测试中的奇技淫巧。