如何更改 OCMock 存根的返回值？

Question

似乎我第一次在 OCMock 存根上添加 andReturnValue 时，该返回值就已确定。例如：

id physics = [OCMockObject niceMockForClass:[DynamicPhysicsComponent class]
Entity *testEntity = [Entity entityWithPhysicsComponent:physics];
CGPoint velocity1 = CGPointMake(100, 100);
CGPoint velocity2 = CGPointZero;
[[[physics stub] andReturnValue:OCMOCK_VALUE(velocity1)] getCurrentVelocity];
[testEntity update:0.1];
[[[physics stub] andReturnValue:OCMOCK_VALUE(velocity2)] getCurrentVelocity];
[testEntity update:0.1];

在[testEntity update]中调用存根方法。但每次存根方法返回速度 1 值，所以我猜想设置方法返回值的第二次尝试不会被执行。

有没有办法在 OCMock 中做到这一点？

Answer 1

当您存根一个方法时，您是说它应该始终以指定的方式运行，无论它被调用多少次。解决此问题的最简单方法是将 stub 更改为 expect：

CGPoint velocity1 = CGPointMake(100, 100);
CGPoint velocity2 = CGPointZero;
[[[physics expect] andReturnValue:OCMOCK_VALUE(velocity1)] getCurrentVelocity];
[testEntity update:0.1];
[[[physics expect] andReturnValue:OCMOCK_VALUE(velocity2)] getCurrentVelocity];
[testEntity update:0.1];

或者，如果您需要 stub（例如，如果该方法可能不是调用），你可以重新创建模拟：

CGPoint velocity1 = CGPointMake(100, 100);
CGPoint velocity2 = CGPointZero;
[[[physics stub] andReturnValue:OCMOCK_VALUE(velocity1)] getCurrentVelocity];
[testEntity update:0.1];
[physics verify];

physics = [OCMockObject mockForClass:[Physics class]];
[[[physics stub] andReturnValue:OCMOCK_VALUE(velocity2)] getCurrentVelocity];
[testEntity update:0.1];
[physics verify];

Answer 2

实际上，当您stub时，如果您使用andReturn或andReturnValue，您只是在设置返回值。您可以随时使用方法andDo 更改返回值。这是对 expect 的改进，您需要知道一个方法将被调用多少次。这是完成此操作的代码片段：

__weak TestClass *weakSelf = self;
[[[physics stub] andDo:^(NSInvocation *invocation) {
    NSValue *result = [NSValue valueWithCGPoint:weakSelf.currentVelocity];
    [invocation setReturnValue:&result];
}] getCurrentVelocity];

Answer 3

虽然我认为 CipherCom 有正确的答案，但我发现自己更喜欢创建一个帮助器类来返回各种值。我过去曾遇到过 NSInspiration 的问题。

@interface TestHelper : NSObject
@property (nonatomic, assign) CGPoint velocity;
- (CGPoint)getCurrentVelocity;
@end

@implementation TestHelper
- (CGPoint)getCurrentVelocity
{
    return self.velocity;
}
@end

然后在我的测试类中，我将有一个 TestHelper 的私有成员变量，并在 setUp 方法中这样做：

self.testHelper = [TestHelper new];

[[[physics stub] andCall:@selector(getCurrentVelocity) onObject:self.testHelper]
                 getCurrentVelocity]; 

这样在每个测试中我都可以设置速度到我想要的测试。

self.testHelper.velocity = CGPointMake(100, 200);

Answer 4

看起来 andReturn/andReturnValue/andDo 在多次调用时都不会被覆盖。我的解决方法是向测试类添加一个属性，并使用它来控制模拟对象应返回的内容。例如，如果模拟对象上有 isAvailable 属性，我的代码将如下所示：

@interface MyTest: XCTestCase 
@property BOOL stubbedIsAvailable;
@end

@implementation MyTest

- (void)setUp {
    [OCMStub([myMockedObject isAvailable]) andDo:^(NSInvocation invocation) {
        BOOL retVal = self.stubbedIsAvailable;
        [invocation setReturnValue:&retVal];
    }
}

- (void)testBehaviourWhenIsAvailable {
    self.stubbedIsAvailable = YES;
    // test the unit
}

- (void)testBehaviourWhenIsNotAvailable {
    self.stubbedIsAvailable = NOT;
    // test the unit
}