如何使用 OCMock 初始化带有存根值的对象

Question

如何对 init 方法中使用的方法进行存根？

我的类中的相关方法：

- (id)init
{
    self = [super init];
    if (self) {
        if (self.adConfigurationType == AdConfigurationTypeDouble) {
             [self configureForDoubleConfiguration];
        }
        else {
            [self configureForSingleConfiguration];
        }
    }
    return self;
}

- (AdConfigurationType)adConfigurationType
{
    if (adConfigurationType == NSNotFound) {
        if ((random()%2)==1) {
            adConfigurationType = AdConfigurationTypeSingle;
        }
        else {
            adConfigurationType = AdConfigurationTypeDouble;
        }
    }
    return adConfigurationType;
}

我的测试：

- (void)testDoubleConfigurationLayout
{
    id mockController = [OCMockObject mockForClass:[AdViewController class]];
    AdConfigurationType type = AdConfigurationTypeDouble;
    [[[mockController stub] andReturnValue:OCMOCK_VALUE(type)] adConfigurationType];

    id controller = [mockController init];

    STAssertNotNil([controller smallAdRight], @"Expected a value here");
    STAssertNotNil([controller smallAdRight], @"Expected a value here");
    STAssertNil([controller largeAd], @"Expected nil here");
}

我的结果：

由于未捕获的异常'NSInternalInconsistencyException'而终止应用程序，原因：'OCMockObject [AdViewController]：调用了意外的方法：smallAdRight'

那么我将如何访问OCMockObject中的AdViewController？

Answer 1

如果您使用 mockForClass: 方法，您将需要为模拟类中调用的每个方法提供存根实现。在您的第一次测试中包括使用 [controllersmallAdRight] 进行的调用。

相反，您可以使用 niceMockForClass: 方法，该方法将忽略任何未模拟的消息。

另一种替代方法是实例化您的 AdViewController，然后使用 partialMockForObject: 方法为其创建部分模拟。这样控制器类的内部将完成主要部分的工作。

只是，您是否正在尝试测试 AdViewController 或使用它的类？看来您正在尝试模拟整个班级，然后测试它是否仍然表现正常。如果您想测试 AdViewController 在注入某些值时是否按预期运行，那么您最好的选择很可能是 partialMockForObject: 方法：

- (void)testDoubleConfigurationLayout {     
  AdViewController *controller = [AdViewController alloc];
  id mock = [OCMockObject partialMockForObject:controller];
  AdConfigurationType type = AdConfigurationTypeDouble;
  [[[mock stub] andReturnValue:OCMOCK_VALUE(type)] adConfigurationType];

  // You'll want to call init after the object have been stubbed
  [controller init]

  STAssertNotNil([controller smallAdRight], @"Expected a value here");
  STAssertNotNil([controller smallAdRight], @"Expected a value here");
  STAssertNil([controller largeAd], @"Expected nil here");
}