OCMock3中无法Stub init方法，原OCMock2用户如何绕过该限制？

我之前在从OCMock2迁移到OCMock3时也碰到过一模一样的问题，大量依赖init stub的测试突然失效确实头疼。虽然官方明确禁用了init方法的stub，但有几个实用的办法能解决这个问题，分享给你：

1. 抽离初始化逻辑到独立方法 #

这是最推荐的方案，既符合测试最佳实践，也能适配OCMock3的规则。把init方法里的核心业务逻辑抽成一个单独的实例方法，比如setupInitialStateWithParam:，然后在测试中用**部分模拟（Partial Mock）**来stub这个抽离后的方法。

举个重构的例子：

生产代码重构前 #

- (instancetype)initWithUserID:(NSString *)userID {
    self = [super init];
    if (self) {
        // 直接在init里处理复杂逻辑
        self.userProfile = [APIClient fetchProfileForUserID:userID];
        self.settings = [SettingsManager loadUserSettings:userID];
    }
    return self;
}

生产代码重构后 #

- (instancetype)initWithUserID:(NSString *)userID {
    self = [super init];
    if (self) {
        [self setupUserContextWithID:userID];
    }
    return self;
}

// 抽离后的初始化逻辑方法
- (void)setupUserContextWithID:(NSString *)userID {
    self.userProfile = [APIClient fetchProfileForUserID:userID];
    self.settings = [SettingsManager loadUserSettings:userID];
}

测试代码写法 #

// 创建真实实例，再包装成Partial Mock
UserManager *realManager = [[UserManager alloc] init];
id mockManager = OCMPartialMock(realManager);

// Stub掉依赖的外部方法，返回测试数据
OCMStub([APIClient fetchProfileForUserID:OCMOCK_ANY]).andReturn(testProfile);
OCMStub([SettingsManager loadUserSettings:OCMOCK_ANY]).andReturn(testSettings);

// 手动调用抽离后的初始化方法
[mockManager setupUserContextWithID:@"test_user_123"];

// 验证初始化后的状态是否符合预期
XCTAssertEqualObjects(mockManager.userProfile, testProfile);

2. 封装初始化调用为类方法 #

如果你的业务代码里是直接调用[[MyClass alloc] initWithXXX:]，可以把这个初始化逻辑封装成一个类方法，比如+ (instancetype)instanceWithXXX:，然后在测试中直接stub这个类方法返回预设的Mock对象。

生产代码 #

@implementation UserManager
+ (instancetype)instanceWithUserID:(NSString *)userID {
    return [[self alloc] initWithUserID:userID];
}

// 原init方法保持不变
- (instancetype)initWithUserID:(NSString *)userID {
    // ...原有逻辑
}
@end

测试代码 #

// 创建类Mock
id mockUserManager = OCMClassMock([UserManager class]);

// Stub类方法，返回我们的Mock实例
OCMStub([UserManager instanceWithUserID:OCMOCK_ANY]).andReturn(mockUserManager);

// 此时业务代码调用[UserManager instanceWithUserID:@"xxx"]时，会直接返回Mock
// 你可以继续Stub这个Mock的其他方法来验证逻辑
OCMStub([mockUserManager doSomething]).andReturn(YES);

3. 用Alloc Stub + Init Stub的Trick（不推荐但可用） #

这个方法属于“曲线救国”，虽然能实现类似OCMock2的效果，但官方不推荐，因为init的语义比较特殊，可能会带来测试隔离问题。思路是先Stub类的alloc方法返回一个Mock，再Stub这个Mock的init方法返回自身。

测试代码示例 #

id mock = OCMClassMock([UserManager class]);
// Stub alloc返回Mock实例
OCMStub([mock alloc]).andReturn(mock);
// Stub init方法返回Mock自身
OCMStub([mock initWithUserID:OCMOCK_ANY]).andReturn(mock);

// 当业务代码调用[[UserManager alloc] initWithUserID:@"xxx"]时，会返回这个Mock

⚠️ 注意：使用这个方法后，一定要在测试结束后调用[mock stopMocking]清理状态，避免影响其他测试用例。

总的来说，优先推荐前两种方案，尤其是第一种，既能让代码结构更清晰，也能让测试更稳定。如果实在来不及大规模重构，第三种可以作为临时过渡方案。

内容的提问来源于stack exchange，提问作者sudeepdino008