测试中如何Mock嵌套调用的静态方法？求低侵入实现方案

问题背景 #

在测试类A的foo方法时，需要Mock其中嵌套调用的类B的静态get方法，消除测试对B::get功能有效性的依赖。已尝试依赖注入方案，但该方案需要修改生产代码，现询问是否存在对类A、类B生产代码改动极小甚至无需改动的Mock方案，同时对自行设想的一段代码示例的合理性提出疑问。

原始生产代码 #

class B
{
    public:
    static int get()
    {
        return 10;
    }
};

class A
{
    public:
    void foo()    // test foo()
    {
        auto val = B::get();  // 需要Mock的调用
        // 其他业务逻辑
    }
};

已尝试的依赖注入方案 #

class BInterface
{
    public:
    virtual int get() = 0;
};

class B
{
    public:
    static int get()
    {
        cout << "10\n";
        return 10;
    }
};

class A
{
    BInterface* ptr;

    public:
    void set(BInterface* p)
    {
        ptr = p;
    }
    
    void foo() 
    {
        if (ptr) 
        {
            ptr->get();
            return;
        }
        // 默认逻辑
    }
};

class BMock : public BInterface
{
    MOCK_METHOD(int, get, (), override);
};

int main()
{
    BMock bmock;
    A a;
    a.set(&bmock);
    a.foo();
}

自行设想的代码示例 #

class AMock 
{
  public:
  void foo()
  {
     BMock bMock; 
     bMock.get();
  }
};

// 给A打桩
void A::foo()
{
   AMock aMock;
   aMock.foo();
   // 复制粘贴生产代码 (??)
}

低改动Mock静态方法的可行方案 #

1. 预编译宏切换实现 #

这个方案仅需对生产代码做极少量修改，完全不影响正常编译逻辑：
在类B的头文件中添加预编译判断：

class B
{
public:
#ifdef TEST_MODE
    static int get(); // 测试模式下仅声明，Mock实现放在测试代码中
#else
    static int get()
    {
        return 10;
    }
#endif
};

测试代码中实现Mock版本的B::get：

#define TEST_MODE
#include "B.h"
#include <gmock/gmock.h>

// 全局Mock对象控制返回值
class BMock {
public:
    MOCK_METHOD(int, get, ());
};
BMock* global_bmock;

int B::get() {
    return global_bmock->get();
}

// 测试用例
TEST(A_Foo_Test, MockBGet) {
    BMock bmock;
    global_bmock = &bmock;
    EXPECT_CALL(bmock, get()).WillOnce(testing::Return(20));
    
    A a;
    a.foo();
    // 业务逻辑断言
}

编译测试代码时添加-DTEST_MODE宏即可切换到Mock实现，生产编译时完全不受影响。

2. 链接器替换（Link-Time Replacement） #

如果使用GCC/Clang编译器，可利用链接器--wrap选项，完全不修改生产代码，直接替换B::get的实现：
测试代码中编写替换函数：

#include <gmock/gmock.h>
#include "A.h"

// 声明原始函数（需根据编译器名字修饰规则生成，可通过nm命令查看）
extern "C" int __real__ZN1B3getEv();
// 替换用的Mock函数
class BMock {
public:
    MOCK_METHOD(int, get, ());
};
BMock* global_bmock;

extern "C" int __wrap__ZN1B3getEv() {
    return global_bmock->get();
}

// 测试用例
TEST(A_Foo_Test, LinkTimeMock) {
    BMock bmock;
    global_bmock = &bmock;
    EXPECT_CALL(bmock, get()).WillOnce(testing::Return(30));
    
    A a;
    a.foo();
    // 业务逻辑断言
}

编译测试代码时添加链接选项：

g++ test.cpp A.cpp B.cpp -o test -lgmock -lgtest --wrap=_ZN1B3getEv

该方案完全不需要修改生产代码，靠链接器完成函数替换。

对您设想代码的分析 #

您的方案存在核心问题，不具备合理性：

• 直接重写A::foo()的实现会完全覆盖生产代码逻辑，测试时跑的是Mock代码，根本没有验证真实的A::foo()业务逻辑，失去了单元测试的意义。
• 复制粘贴生产代码到Mock中会导致维护灾难，生产代码更新后Mock代码无法同步，测试结果会失真。
• 这种方式本质是替换了被测试类本身，而非Mock依赖，完全不符合单元测试“验证目标类逻辑，隔离外部依赖”的核心目标。

内容的提问来源于stack exchange，提问作者xyf