c++ - GoogleMock:如何验证输入参数 w.r.t 调用次数？

Question

我正在使用 GoogleTest 为 C++ 类编写单元测试和 GoogleMock .我当前的代码如下所示:

MockNetConnector* connector = new MockNetConnector();
NetClient* client = new NetClient(connector);

TEST_F(NetClientTest, connect)
{
    EXPECT_CALL(*connector, attempt_connect(_,_)).Times(3)
        .WillOnce(Return(false))
        .WillOnce(Return(false))
        .WillOnce(Return(true));

    std::string srv_list = "127.0.0.1:30001,127.0.0.2:30002,127.0.0.3:30003";
    bool is_connected = client->connect(srv_list);

    ASSERT_TRUE(is_connected);
}

如您所见，(*connector).attempt_connect(_,_) 当前不验证输入参数。现在我希望它在每次调用时验证输入，即对于调用 x-th<，输入应该是 (127.0.0.x, 3000x)/em>.

我知道如何验证固定值的参数，例如attempt_connect(StrEq("127.0.0.1"),Eq(30001))，但不知道如何验证因调用而异的参数。

Answer 1

我找到了问题的解决方案，但可能不是最好的。

首先，定义一个名为postfix的全局变量:

static int postfix = 0;

写自定义matchers :

MATCHER(ValidateIP, "")
{
    std::string expected_ip = "127.0.0." + to_string(::ip_postfix);

    return (expected_ip == arg);
}

MATCHER(ValidatePort, "")
{
    int expected_port = 30000 + ::port_postfix;

    return (expected_port == arg);
}

写一个自定义action在每次调用后增加 postfix。

ACTION(IncreasePostfix)
{
    ::postfix++;
}

将匹配器应用于期望:

EXPECT_CALL(*connector, attempt_connect(ValidateIP(),ValidatePort())).Times(3)
    .WillOnce(DoAll(IncreasePostfix(), Return(false)))
    .WillOnce(DoAll(IncreasePostfix(), Return(false)))
    .WillOnce(DoAll(IncreasePostfix(), Return(true)));

要使 postfix 对每个测试单独有效，请在测试开始前重置它们(即在 NetClientTest::SetUp() 内)

class NetClientTest : public testing::Test
{
    virtual void SetUp() {
        ::postfix = 1;
    }

    // other methods
};