期望来自使用 OCMock 的非实例化变量的方法调用

Question

我不确定这是否是最好的标题，但基本上我知道变量将在运行函数时实例化。我想对这个变量进行部分模拟，并期望在实例化变量之前进行某些方法调用。这是我正在尝试做的一个例子。

-(void)testMethod {
    id mockVar = [OCMock partialMockForObject:self.controller.variable];
    [[mockVar expect] someMethod];
    [self.controller method];
    [mockVar verify];
}

控制器内的方法如下所示：

-(void)method {
    self.variable = [[Class alloc]init];
    [self.variable someMethod];
}

我收到一条类似"不识别选择器"的消息。这可能吗？

Answer 1

如果可以存根属性访问器，我会这样做。 但是，如果您需要部分模拟由真实代码实际创建的实例，只要它存储在属性中(setter 方法成为钩子(，这是可能的，但它肯定是丑陋的。

基本思想是使用 OCMArg 的 checkWithBlock： 功能来拦截 set 方法的参数，并在此时创建部分模拟。 您可以将其分配给块外的变量，以便稍后进行验证。 它看起来像这样：

-(void)testMethod
{
    id mockController = [OCMockObject partialMockForObject:self.controller];
    __block id mockVar;
    [[[mockController stub] andForwardToRealObject] setVariable:[OCMArg checkWithBlock:^BOOL(id param) {
        mockVar = [OCMockObject partialMockForObject:param];
        [[mockVar expect] someMethod];
        return YES;
    }]];
    [self.controller method];  // mockVar will be created and assigned during this call
    [mockVar verify];
}

如果你的真实代码创建了一个对象，并且只将其临时存储在局部变量(不是实例变量(中，除了重组真实代码以使其更易于测试(例如将创建对象的代码移动到新方法中，这使得存根更容易(之外，您实际上无能为力。

Answer 2

这感觉像是一个奇怪的测试，但如果你模拟访问器，这是可能的：

-(void)testMethod {
    id mockVar = [OCMock mockForClass:[Class class]];
    id mockController = [OCMock partialMockForObject:controller];
    [[[mockController stub] andReturn:mockVar] variable];
    [[mockVar expect] someMethod];
    [self.controller method];
    [mockVar verify];
}

如果你提供一个更具体的例子来说明你想要实现的目标，我可以给你一个更好的答案。