c++ 如何在gmock expect_call中对结构参数进行部分匹配

ktca8awb  于 2022-12-01  发布在  其他
关注(0)|答案(1)|浏览(325)
struct obj
{
  int a;
  string str;
  string str2;
  bool operator==(const obj& o) const
  {
     if(a == o.a && str == o.str && str2 == o.str2) return true;
     return false;
   } 
}

然后在一个函数类中,它使用struct对象作为输入参数:

bool functionNeedsToBeMocked(obj& input)
{
  //do something
}

现在我想做的是

EXPECT_CALL(*mockedPointer, functionNeedsToBeMocked( /* if input.a == 1 && input.str == "test" && input.str2.contains("first")*/  )).Times(1).WillOnce(Return(true));

输入值为

inputFirst.a = 1;
inputFirst.str = "test";
inputFirst.str2 = "something first";

我希望inputFirst可以与我的EXPECT_CALL匹配。
我如何使用EXPECT_CALL匹配器来做到这一点呢?
我确实看到了

EXPECT_CALL(foo, DoThat(Not(HasSubstr("blah")),
                      NULL));

但是我不知道如何为一个结构参数做HasSubStr。

k3bvogb1

k3bvogb11#

您可以为obj结构实现自己的匹配器。
当您键入:

EXPECT_CALL(*mockedPointer, functionNeedsToBeMocked(some_obj)).Times(1).WillOnce(Return(true));

那么gmock将使用默认的匹配器Eq,使用some_obj作为它的预期参数,使用实际的functionNeedsToBeMocked作为匹配器中的argEq匹配器将默认为预期对象和实际对象调用bool operator==

EXPECT_CALL(*mockedPointer, functionNeedsToBeMocked(Eq(some_obj))).Times(1).WillOnce(Return(true));

但是,如果您不想使用bool operator==,可以编写一个自定义匹配器(删除Times(1),因为它也是默认匹配器):

// arg is passed to the matcher implicitly
// arg is the actual argument that the function was called with
MATCHER_P3(CustomObjMatcher, a, str, str2, "") {
  return arg.a == a and arg.str == str and (arg.str2.find(str2) != std::string::npos); 
}
[...]
EXPECT_CALL(*mockedPointer, functionNeedsToBeMocked(CustomObjMatcher(1, "test", "first"))).WillOnce(Return(true));

有一种可能性,可以使用Field匹配器和内置匹配器(如HasString)编写一个自定义匹配器,但让我们“把它作为一个练习留给读者”:P
更新:完整的Field匹配器代码:

struct obj {
    int a;
    std::string str;
    std::string str2;
};

struct Mock {
    MOCK_METHOD(bool, functionNeedsToBeMocked, (obj&));
};

// creates a matcher for `struct obj` that matches field-by-field
auto Match(int expected_a, std::string expected_str1, std::string expected_substr2) {
    return testing::AllOf(
            testing::Field(&obj::a, expected_a),
            testing::Field(&obj::str, expected_str1),
            testing::Field(&obj::str2, testing::HasSubstr(expected_substr2))
    );
}

TEST(MyTest, test) {
    Mock mock{};

    obj inputFirst;
    inputFirst.a = 1;
    inputFirst.str = "test";
    inputFirst.str2 = "something first";

    EXPECT_CALL(mock, functionNeedsToBeMocked(Match(1, "test", "first"))).Times(1).WillOnce(Return(true));

    mock.functionNeedsToBeMocked(inputFirst);
}

相关问题