单例模式的mock类注入单元测试与友元类解决方案

问题背景：单例模式的测试挑战

在单元测试中，我们经常需要对单例模式进行测试，特别是当单例类依赖其他组件时。传统的单例模式通过静态方法获取实例，这使得依赖注入和mock替换变得困难。

核心困境

我们需要在测试时：

1. 注入Mock对象替换真实依赖

2. 访问私有构造函数来创建带有mock依赖的实例

3. 避免影响生产代码的单例行为

这是一个很常见的单例模式测试问题。你需要使用前向声明和友元声明的正确组合来解决。以下是具体的解决方案：

解决方案步骤

1. 在单例类中前向声明测试类并声明为友元

在你的单例类头文件中（比如 singleton.h）：

// singleton.h
#pragma once// 前向声明测试类
class SingletonTest;class Singleton {
public:static Singleton& GetInstance();// 公有方法void SomePublicMethod();private:// 私有构造函数Singleton();~Singleton();// 声明测试类为友元friend class SingletonTest;// 私有方法（测试需要访问的）void SomePrivateMethod();// 禁用拷贝和赋值Singleton(const Singleton&) = delete;Singleton& operator=(const Singleton&) = delete;// 成员变量int some_data_;
};

2. 在测试文件中包含单例类头文件

在你的测试文件中（比如 singleton_test.cc）：

// singleton_test.cc
#include "gtest/gtest.h"
#include "singleton.h"  // 必须包含单例类的完整定义class SingletonTest : public ::testing::Test {
protected:void SetUp() override {// 测试设置}void TearDown() override {// 测试清理}
};TEST_F(SingletonTest, TestPrivateMethod) {// 可以直接访问私有方法Singleton::GetInstance().SomePrivateMethod();// 或者通过其他方式测试// ...
}TEST_F(SingletonTest, TestConstructor) {// 如果需要测试构造函数逻辑，可以这样// 注意：单例模式通常不需要直接测试构造函数
}

更复杂的场景：如果测试类在不同的命名空间中

如果测试类在不同的命名空间，友元声明需要更精确：

单例类中：

// singleton.h
#pragma oncenamespace mynamespace {// 前向声明带命名空间的测试类namespace test {class SingletonTest;}class Singleton {public:static Singleton& GetInstance();private:Singleton();~Singleton();// 声明带命名空间的测试类为友元friend class test::SingletonTest;void SomePrivateMethod();};
}

测试类中：

// singleton_test.cc
#include "gtest/gtest.h"
#include "singleton.h"namespace mynamespace::test {class SingletonTest : public ::testing::Test {
protected:void SetUp() override {}void TearDown() override {}
};TEST_F(SingletonTest, TestPrivateAccess) {// 现在可以访问私有成员Singleton::GetInstance().SomePrivateMethod();
}} // namespace mynamespace::test