ios - stub [[SomeClazz alloc] init] 不起作用，但接受的答案说它应该起作用

我的被测函数很简单:

@implementation MyHandler
...
-(void) processData {
  DataService *service = [[DataService alloc] init];
  NSDictionary *data = [service getData];
  [self handleData:data];
}

@end

我使用 OCMock 3对其进行单元测试。

我需要 stub [[DataService alloc] init] 来返回一个模拟实例，我尝试了answer from this question (这是一个公认的答案) stub [[SomeClazz alloc] init]:

// Stub 'alloc init' to return mocked DataService instance,
// exactly the same way as the accepted answer told
id DataServiceMock = OCMClassMock([DataService class]);
OCMStub([DataServiceMock alloc]).andReturn(DataServiceMock);
OCMStub([DataServiceMock init]).andReturn(DataServiceMock);

// run function under test
[MyHandlerPartialMock processData];

// verify [service getData] is invoked
OCMVerify([dataServiceMock getData]);

我在被测函数中设置了断点，我确定在运行单元测试时会调用 [service getData]，但是我上面的测试代码(OCMVerify)失败了。为什么？

是不是因为被测函数没有使用我的mocked DataService？但是该问题中接受的答案表明它应该有效。我现在糊涂了……

我想知道如何使用 OCMock stub [[SomeClazz alloc] init] 以返回模拟实例？

---

Best Answer-推荐答案

您不能模拟 init，因为它是由模拟对象本身实现的。模拟 init 在您链接的答案中起作用的原因是因为它是一个自定义初始化方法。如果您不想使用依赖注入(inject)，则必须为 DataService 编写一个自定义的 init 方法，您可以模拟该方法。

在您的实现中添加自定义 init 方法:

// DataService.m
...
- (id) initForTest
{
    self = [super init];
    if (self) {
        // custom initialization here if necessary, otherwise leave blank
    }

    return self;
}
...

然后更新MyHandler实现来调用这个initForTest:

@implementation MyHandler
...
-(void) processData {
  DataService *service = [[DataService alloc] initForTest];
  NSDictionary *data = [service getData];
  [self handleData:data];
}

@end

最后将您的测试更新为 stub initForTest:

id DataServiceMock = OCMClassMock([DataService class]);
OCMStub([DataServiceMock alloc]).andReturn(DataServiceMock);
OCMStub([DataServiceMock initForTest]).andReturn(DataServiceMock);

// run function under test
[MyHandlerPartialMock processData];

// verify [service getData] is invoked
OCMVerify([dataServiceMock getData]);

随意重命名 initForTest，只要不叫 init。

关于ios - stub [[SomeClazz alloc] init] 不起作用，但接受的答案说它应该起作用，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/37654226/