MessageMock

     

优雅的模拟 Objective-C 方法，对参数、返回值进行任意修改和检查。

原理分析文章：MessageMock : 优雅的模拟 Objective-C 方法

特性

通过任意[target selector]调用命中目标方法。

功能：

• 修改目标方法返回值、参数
• 验证目标方法返回值、参数
• 跳过目标方法调用
• 获取目标方法命中次数

TODO

• 支持 x86
• 支持大于指针长度的返回值修改
• 支持大于指针长度的参数修改

安装

CocoaPods

1. 在 Podfile 中添加：

pod 'MessageMocker'

2. 执行 pod install 或 pod update。
3. 导入 #import <MessageMocker.h>。

手动导入

1. 把 MessageMock 源码文件夹整个拖入工程。
2. 导入文件#import "MessageMocker.h"

基本用法

通过MessageMocker.h的接口进行链式配置（参考MessageMockerTests.m的单元测试代码）。

修改返回值

    NSObject *value = [NSObject new];
    
    MessageMocker.build(NSNotificationCenter.self, @selector(defaultCenter)).mockReturn(value).start();

    XCTAssertTrue(value == [NSNotificationCenter defaultCenter]);

修改参数

    UIView *view = [UIView new];
    NSInteger value = 999;
    
    MessageMocker.build(view, @selector(setTag:)).mockArgument(0, value).start();
    
    view.tag = 666;
    XCTAssertTrue(view.tag == value);

检查返回值

    __block unsigned long value = 0;
    
    NSObject *obj = [NSObject new];
    
    MessageMocker.build(obj, @selector(hash))
    .checkReturn(^(unsigned long arg) {
        value = arg;
    })
    .start();
    
    unsigned long res = [obj hash];
    XCTAssertTrue(value == res);

检查参数

    Class anyCls = NSString.self;
    
    MessageMocker.build(NSObject.self, @selector(isSubclassOfClass:))
    .checkArgument(0, ^(Class cls){
        XCTAssertTrue(anyCls == cls);
    })
    .start();
    
    [NSObject isSubclassOfClass:anyCls];

跳过方法调用

    UIView *view = [UIView new];
    NSInteger tag = 666;
    view.tag = tag;
    
    MessageMocker.build(view, @selector(setTag:)).skip(YES).start();
    
    view.tag = 999;
    XCTAssertTrue(view.tag = tag);

设置/读取命中次数

    NSObject *value = [NSObject new];
    
    MessageMocker *mocker = MessageMocker.build(TestObject.self, @selector(createObj))
    .mockReturn(value)
    .limitHitCount(2);
    mocker.start();
    
    XCTAssertTrue(value == [TestObject createObj]);
    XCTAssertTrue(mocker.hitCount == 1);
    XCTAssertTrue(value == [TestObject createObj]);
    XCTAssertTrue(mocker.hitCount == 2);
    
    XCTAssertTrue(value != [TestObject createObj]);