c++ - 使用 GMOCK 模拟具有私有构造函数的类

Question

我有一个带有私有 Tor、Dtor 和一个 getInstance() 方法的 Singleton 类。

class Single {
public:
virtual void* alloc(size_t size, uint line){}
Single* getInstance() {
if(!m_Instance)
        m_Instance = __OSAL_NEW OSAL_Memory;
    return m_Instance;
}
private:
Single();
~Single();
static Single* m_Instance;
};


#define Allocate(size_t size)\
(Single::getInstance())->alloc(size, __LINE__)

我需要使用 GMOCK 模拟这个类。有什么办法可以模拟它。

Answer 1

您可以使用工厂模式来创建您的对象。

#include <iostream>
#include <functional>

struct A
{
    virtual ~A(){}

    virtual void foo() = 0;
};

struct Areal : A
{
    virtual void foo(){
        std::cout<<"real"<<std::endl;
    }
};

struct Amock : A
{
    virtual void foo(){
        std::cout<<"mock"<<std::endl;
    }
};

struct Single
{
    typedef std::function< A*() > CreatorFn;

    static A* getInstance(){
        if ( ! instance() )
            instance() = (create())();
        return instance();
    }

    static void setCreator( CreatorFn newFn ){
        create() = newFn;
    }

private:

    static CreatorFn& create(){
        static CreatorFn f( [](){return new Areal;} );
        return f;
    }
    static A*& instance(){
        static A* p=nullptr;
        return p;
    }
};


bool useMock = true;

int main()
{
    if ( useMock )
    {
        Single::CreatorFn mockFn( [](){ return new Amock; } );
        Single::setCreator( mockFn );
    }

    Single::getInstance()->foo();
}

您只需确保在访问实例之前设置了创建者。否则，将调用默认的创建者函数。