调用另一个模拟函数时更改模拟函数的返回值

时间:2020-07-17 05:43:23

标签: c++ googletest googlemock

假设我有以下模拟内容

enum class State
{
    IDLE,
    BUSY,
    ERROR1,
    ERROR2
};

class MockActuator : public ActuatorInterface
{
    public:
        MOCK_METHOD0(doAction, void());
        MOCK_METHOD0(getState, State());
};

我正在测试的模块假设,如果调用doAction()getState()应该返回BUSY

我该如何在GMock下编码此假设?我想将getState()保留为模拟函数,因为还需要测试其他返回值。

我的第一次尝试是:

    EXPECT_CALL(actuator, doAction()).Times(1).WillOnce(InvokeWithoutArgs(
                [&](){
                    ON_CALL(actuator, getState()).WillByDefault(Return(State::BUSY));
                }));

但这会导致以下隐式错误:

/usr/src/googletest/googlemock/include/gmock/gmock-actions.h:861:64: error: void value not ignored as it ought to be
   Result Perform(const ArgumentTuple&) { return function_impl_(); }

1 个答案:

答案 0 :(得分:2)

您可以在docs中看到有关此操作的示例。

上课之后。

首先是模仿原始类的假类:

enum class State
{
    IDLE,
    BUSY,
    ERROR1,
    ERROR2
};

class FakeActuator : public ActuatorInterface
{
    public:
        virtual void doAction () { _state = BUSY; }
        virtual State getState () { return _state; }
    private:
        State _state;
};

之后,模拟类:

class MockActuator : public ActuatorInterface {
 public:
  // Normal mock method definitions using gMock.
  MOCK_METHOD(void, doAction, (), (override));
  MOCK_METHOD(State, getState, (), (override));

  // Delegates the default actions of the methods to a FakeActuator object.
  // This must be called *before* the custom ON_CALL() statements.
  void DelegateToFake() {
    ON_CALL(*this, doAction).WillByDefault([this]() {
      fake_.doAction();
    });
    ON_CALL(*this, getState).WillByDefault([this]() {
      return fake_.getState();
    });
  }

 private:
  FakeActuator fake_;  // Keeps an instance of the fake in the mock.
};

此“伪造”对象可以保持两次调用之间的状态。您还可以根据需要创建尽可能多的“伪”类。

测试:

TEST(AbcTest, Xyz) {
  MockFoo foo;

  foo.DelegateToFake();  // Enables the fake for delegation.

  // Put your ON_CALL(foo, ...)s here, if any.

  // No action specified, meaning to use the default action.
  EXPECT_CALL(foo, doAction());
  EXPECT_CALL(foo, getState());

  foo.doAction();  // FakeActuator::doAction() is invoked.

  EXPECT_EQ(BUSY, foo.getState());  // FakeActuator::getState() is invoked.
}