在循环中添加预期的调用

时间:2016-08-08 09:47:39

标签: c++ unit-testing googletest gmock

这就是我想要实现的目标。在我的测试夹具中,我想调用带有参数n的辅助函数来告诉测试夹具应该预期多少初始化序列。序列中使用的一些参数存储在三个std::vector容器中; fileDescriptor, handle, selectionObject

我写的是:

void MyTest::init_Ok(uint32_t n)
{
  for (uint32_t i = 0; i < n; ++i)
  {
    fileDescriptor.push_back(i); // FDs starting at 0
    handle.push_back(reinterpret_cast<void*>(18 + i)); // handles starting at 18
    selectionObject.push_back(555 + i); // SOs starting at 555

    EXPECT_CALL(MyMockApi::getApi(), initialize(Pointee(nullptr), StrEq("InitString"), MyMatcher()))
      .WillOnce(DoAll(SetArgPointee<0>(handle[i]),
                      Return(INIT_OK)));
    EXPECT_CALL(MyMockApi::getApi(), selectionObjectGet(handle[i], Pointee(nullptr)))
      .WillOnce(DoAll(SetArgPointee<1>(selectionObject[i]),
                      Return(SELECTION_OK)));
    EXPECT_CALL(MyMockApi::getApi(), finalize(handle[i]))
      .WillOnce(Return(FINAL_OK));
  }
}

我知道它为什么不起作用。预计对initialize的所有调用都是相同的,但我想对第一,第二,第三,......,第n个调用执行不同的操作(参数取决于循环计数器i)。无论参数initialize如何,当前实现只需要调用n。是否有可能解决此问题并以某种方式保持循环,或者我是否必须为每个WillOnce添加i行的操作?这意味着我必须检查n并为WillOnce的不同可能值添加不同数量的n行,这是我真正想要避免的。

1 个答案:

答案 0 :(得分:2)

一种方法是使用Invoke。您可以编写一个可以访问handle容器的函数和一个运行的成员/静态变量(比如说counterVar),它将指示函数被命中的次数。根据counterVar的值,您可以决定逻辑。

.WillRepeatedly(Invoke(<your function>))

类似的东西:

EXPECT_CALL(MyMockApi::getApi(), initialize(Pointee(nullptr), StrEq("InitString"), MyMatcher()))
      .WillRepeatedly(Invoke(successfulInitialize));

ReturnCode successfulInitialize(void* op, std::string msg)
{
   static int counterVar = 0;
   *op = handles[counterVar++];
   return INIT_OK;
}