OCMock 可以运行块参数吗

OCMock是一个Objective-C的单元测试框架，它主要用于模拟对象并进行单元测试。在进行单元测试时，有时候我们需要对一个方法的块参数进行测试，这时候可以通过OCMock来实现。

objective-c
// 假设我们有一个Calculator类，其中有一个方法add:withCompletion:，用于计算两个数的和，并在计算完成后执行一个块参数
// Calculator.h
@interface Calculator : NSObject
- (void)add:(NSInteger)a b:(NSInteger)b withCompletion:(void (^)(NSInteger))completion;
@end
// Calculator.m
@implementation Calculator
- (void)add:(NSInteger)a b:(NSInteger)b withCompletion:(void (^)(NSInteger))completion {
    NSInteger result = a + b;
    completion(result);
}
@end
// 测试代码
- (void)testAddWithCompletion {
    // 创建模拟对象
    Calculator *mockCalculator = OCMClassMock([Calculator class]);
    
    // 设置模拟对象方法的行为
    OCMStub([mockCalculator add:2 b:3 withCompletion:[OCMArg any]])
        .andDo(^(NSInvocation *invocation) {
            // 获取块参数
            void (^completion)(NSInteger);
            [invocation getArgument:&completion atIndex:3];
            
            // 执行块参数
            completion(5);
        });
    
    // 调用被测试的方法
    [mockCalculator add:2 b:3 withCompletion:^(NSInteger result) {
        // 断言
        XCTAssertEqual(result, 5);
    }];
}