GMock：在模拟函数中捕获引用参数

Posted 2021-04-03

我有一个带有引用参数的模拟函数：

MOCK_METHOD(func, void(MyObj &obj);

我想将obj捕获到一个变量中，以便我可以在单元测试中调用它上面的方法。所以类似于：

MyObj *capturedObj;
EXPECT_CALL(foo, func(testing::_)).WillOnce(<capture arg into capturedObj>);
capturedObj->bar();

我该如何实现这一目标？

答案

一种可能的方法是调用一个单独的测试函数，它将为您捕获参数。

这可以通过Invoke（）动作完成，如documentation中所述：

调用（f）使用传递给mock函数的参数调用f，其中f可以是全局/静态函数或函子。

Invoke（object_pointer，＆class :: method）使用传递给mock函数的参数调用对象上的方法。

在您的情况下，类似下面的代码应该工作：

// Globally, outside your test case
MyObj *capturedObj;

void captureObj(MyObj &obj) {
    captured = &obj;
}

// [..]

// In your test case
EXPECT_CALL(foo, func(testing::_)).WillOnce(testing::Invoke(captureObj));
capturedObj->bar();

如果使用测试夹具，capturedObj和captureObj可以是类成员，您需要更新语法如下：

  EXPECT_CALL(foo, func(testing::_)).WillOnce(testing::Invoke(this, &TestFixtureClass::captureObj));

另一答案

你可以为它写一个custom action：

ACTION_P(SaveArgAddress, p) {
  p = &arg;
}

然后使用该动作。

但请注意，capturedObj只有在运行测试代码时才会初始化，并且只有在EXPECT_CALL实际匹配调用时才会初始化 - 而不是在执行EXPECT_CALL宏之后。