[C/C++]gtest+mockcpp,修改mockcpp以支持C++中类成员函数的mock
① 修改mockcpp的接口文件(追加一个宏),在mokc.h中追加MOCKER_CPP宏:#define MOCKER_CPP(api, TT) MOCKCPP_NS::mockAPI(#api, reinterpret_cast<TT>(api))② 使用例子:#ifndef SAMPLE_H#define SAMPLE_H#include <gtest/gtest.h>
·
① 修改mockcpp的接口文件(追加一个宏),在mokc.h中追加MOCKER_CPP宏:
#define MOCKER_CPP(api, TT) MOCKCPP_NS::mockAPI(#api, reinterpret_cast<TT>(api))
② 使用例子:
#ifndef SAMPLE_H
#define SAMPLE_H
#include <gtest/gtest.h>
#include <mockcpp/mockcpp.hpp>
#include <memory>
using namespace std;
class ICalc
{
public:
virtual ~ICalc() {};
virtual int add(int a, int b) = 0;
virtual int add4(int a, int b) = 0;
};
class CalcImpl
: public ICalc
{
public:
virtual int add(int a, int b) {
return a + b;
}
virtual int add4(int a, int b) {
return a + b;
}
static int add2(int a, int b) {
return a + b;
}
int add3(int a, int b) {
return a + b;
}
};
class CalcImpl2
: public ICalc
{
public:
virtual int add(int a, int b) {
return a + b;
}
virtual int add4(int a, int b) {
return a + b;
}
static int add2(int a, int b) {
return a + b;
}
int add3(int a, int b) {
return a + b;
}
int add3(int a, int b, int c) {
return a + b + c;
}
};
class TestUse
{
public:
int add(int a, int b) {
calc.add3(a, b);
}
char* getData() {
char* sd = (char *)malloc(sizeof(char) * 100);
return sd;
}
CalcImpl calc;
CalcImpl calc2;
};
class CalcTest : public testing::Test
{
protected:
CalcTest() = default;
~CalcTest() = default;
virtual void SetUp() override
{
}
virtual void TearDown() override
{
}
};
// 测试纯虚类中的纯虚函数
TEST_F(CalcTest, test_xxx_001)
{
MockObject<ICalc> calc;
MOCK_METHOD(calc, add)
.stubs()
.will(returnValue(11));
EXPECT_EQ(11, calc->add(1, 1));
GlobalMockObject::verify();
}
// 测试纯虚类中的纯虚函数
TEST_F(CalcTest, test_xxx_002)
{
MockObject<CalcImpl> calc;
MOCK_METHOD(calc, add)
.stubs()
.will(returnValue(11));
EXPECT_EQ(11, calc->add(1, 1));
GlobalMockObject::verify();
}
// 测试类中的static函数
TEST_F(CalcTest, test_xxx_003)
{
MOCKER(&CalcImpl::add2)
.stubs()
.will(returnValue(11));
EXPECT_EQ(11, CalcImpl::add2(1, 1));
GlobalMockObject::verify();
}
// 测试已存在对象中的虚函数函数mock
TEST_F(CalcTest, test_xxx_004)
{
// add4和add为虚函数,第一个为this指针
CalcImpl impl;
MOCKER_CPP(&CalcImpl::add4, int (*)(CalcImpl*, int, int))
.stubs()
.will(returnValue(11));
MOCKER_CPP(&CalcImpl::add, int (*)(CalcImpl*, int, int))
.expects(once())
.with(any(), eq(1), eq(1))
.will(returnValue(222));
EXPECT_EQ(11, impl.add4(1, 1));
EXPECT_EQ(222, impl.add(1, 1));
GlobalMockObject::verify();
}
// 测试已存在对象中的成员函数mock
TEST_F(CalcTest, test_xxx_005)
{
CalcImpl impl;
MOCKER_CPP(&CalcImpl::add3, int (*)(CalcImpl*, int, int))
.expects(once())
.with(any(), eq(1), eq(1))
.will(returnValue(11));
EXPECT_EQ(11, impl.add3(1, 1));
GlobalMockObject::verify();
}
// 测试嵌套调用对象中的成员函数mock(开启mock)
TEST_F(CalcTest, test_xxx_006)
{
TestUse tu;
MOCKER_CPP(&CalcImpl::add3, int (*)(CalcImpl*, int, int))
.expects(once())
.with(eq(&tu.calc), eq(1), eq(1))
.will(returnValue(11));
EXPECT_EQ(11, tu.add(1, 1));
GlobalMockObject::verify();
}
// 测试嵌套调用对象中的成员函数mock(关闭mock)
TEST_F(CalcTest, test_xxx_007)
{
TestUse tu;
EXPECT_EQ(2, tu.add(1, 1));
GlobalMockObject::verify();
}
// 测试malloc的hook
//TEST_F(CalcTest, test_xxx_008)
//{
// TestUse tu;
// MOCKER(malloc)
// .expects(once())
// .will(returnValue(NULL));
// GlobalMockObject::verify();
//}
// 测试嵌套调用对象中的成员函数mock(关闭mock)
TEST_F(CalcTest, test_xxx_009)
{
TestUse tu;
EXPECT_EQ(2, tu.add(1, 1));
GlobalMockObject::verify();
}
// 测试只针对某个对象自己的虚函数mock
TEST_F(CalcTest, test_xxx_010)
{
CalcImpl tu;
CalcImpl2 tu2;
int (CalcImpl::*pfn)(int, int) = &CalcImpl::add4;
int (CalcImpl2::*pfn2)(int, int) = &CalcImpl2::add4;
MOCKER_CPP(tu.*pfn, int (*)(CalcImpl*, int, int))
.expects(once())
.with(any(), eq(1), eq(1))
.will(returnValue(11));
printf("1: %p\n", (void*)(tu.*pfn));
printf("2: %p\n", (void*)(tu2.*pfn2));
EXPECT_EQ(11, tu.add4(1, 1));
EXPECT_EQ(2, tu2.add4(1, 1));
GlobalMockObject::verify();
}
// 测试重载的情况
TEST_F(CalcTest, test_xxx_011)
{
CalcImpl2 tu2;
int (CalcImpl2::*pfn)(int, int, int) = &CalcImpl2::add3;
int (CalcImpl2::*pfn2)(int, int) = &CalcImpl2::add3;
MOCKER_CPP(tu2.*pfn, int (*)(CalcImpl2*, int, int, int))
.expects(once())
.with(any(), eq(1), eq(1), eq(1))
.will(returnValue(222));
MOCKER_CPP(tu2.*pfn2, int (*)(CalcImpl2*, int, int))
.expects(once())
.with(any(), eq(1), eq(1))
.will(returnValue(111));
EXPECT_EQ(222, tu2.add3(1, 1, 1));
EXPECT_EQ(111, tu2.add3(1, 1));
GlobalMockObject::verify();
}
#endif
更多推荐



所有评论(0)