使用GMOCK嘲笑一個擁有私有構造函數的類

Question

我有一個帶私有Ctor，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

您可以使用factory pattern來創建您的對象。

#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(); 
} 

你一定要確保你之前設置的創造者訪問的實例。否則，將調用默認的創建者函數。