如何使用 OCMock 验证方法调用的数量

Question

有没有办法验证一个方法是否已被调用“x”次？

Answer 1

查看 OCMock 的测试文件，似乎您需要拥有与呼叫数量相同的 expect 数量。因此，如果你调用 someMethod 三次，你需要做...

[[mock expect] someMethod];
[[mock expect] someMethod];
[[mock expect] someMethod];

...test code...

[mock verify];

虽然这看起来很难看，也许你可以把它们放在一个循环中？

Answer 2

我通过利用委托给块的能力取得了成功：

OCMStub([mock someMethod]).andDo(^(NSInvocation *invocation)
{ /* block that handles the method invocation */ });

在块内，我只需增加一个 callCount 变量，然后断言它与预期的调用数量匹配。例如：

- (void)testDoingSomething_shouldCallSomeMethodTwice {
    id mock = OCMClassMock([MyClass class]);

    __block int callCount = 0;
    OCMStub([mock someMethod]).andDo(^(NSInvocation *invocation) {
        ++callCount;
    });

    // ...exercise code...

    int expectedNumberOfCalls = 2;
    XCTAssertEqual(callCount, expectedNumberOfCalls);
}

每次调用 someMethod 时都应调用该块，因此 callCount 应始终与实际调用该方法的次数相同。

Answer 3

如果您需要检查一个方法是否只被调用一次，您可以这样做

[self.subject doSomething];
OCMVerify([self.mock method]);

OCMReject([self.mock method]);
[self.subject doSomething];