如何将gmock函数分配给特定的函数指针?

时间:2019-07-17 08:55:10

标签: c++ c function-pointers googlemock gmock

我正在使用C ++中的Gtest和Gmock进行单元dll的单元测试:
A.dll和B.dll都是用C编写的,我无法对其进行修改。

A.dll的init函数通过函数指针将B.dll的函数用作参数。我想模拟B的功能(因为它们是与硬件相关的)。

我为A.dll创建了一个测试夹具类,该类可动态加载功能initcalc。以下代码提供了所需功能的快速概述:

class TestFixture : public ::testing::Test {
    // dynamically loads functions from A.dll and assigns
    // them to function pointers init_ptr and calc_ptr.
};

// function pointer typedef for use with B.dll's functions
typedef int (*funcPtr)(int, int);

// Loaded from A.dll    
void init(funcPtr func1, funcPtr func2) {
    // internally set functions in A.dll to be used in calculate
}

// Loaded from A.dll
int calculate(int a, int b) {
    // returns a+b + a+b
    return func1(func2(a,b), func2(a,b));
}

// Inside B.dll, should be mocked
int add(int a, int b) { return a+b; }

// Dummy class for B.dll
class B {
    virtual ~B() {}
    virtual int add(int a, int b) = 0;    
};

class MockB : public B {
virtual ~MockB() {}
    MOCK_METHOD(int, add, (int a, int b));
};

// Following example test run is the goal:
TEST_F(TestFixture, AddTest) {
    MockB b;

    // want to use mocked add function here
    init_ptr(mockedAdd, mockedAdd);

    EXPECT_CALL(b, add(_,_)).Times(3);
    EXPECT_EQ(calc_ptr(2,3), 10);
}

当我尝试创建虚拟B和MockB类时,我没有设法将模拟方法分配给init_ptr(funcPtr, funcPtr)所需的函数指针。有没有办法用Gmock(或类似的框架)实现这一目标?

1 个答案:

答案 0 :(得分:5)

最简单的解决方案是简单地声明一个静态(或免费)的函数来调用模拟程序。

class Fixture : public ::testing::Test
{
public:
    static int add(int a, int b)
    {
        return mock_.add(a, b);
    }

    static MockB mock_;
};

MockB Fixture::mock_; // static declaration somewhere in .cpp

TEST_F(Fixture, MyTest)
{
    EXPECT_CALL(mock_, add(_, _));
    init(Fixture::add, Fixture::add);
}