Golang中如何对依赖无接口第三方库的组件进行单元测试

Go语言生态官方推荐的处理方式是自定义适配接口，利用Go语言接口隐式实现的特性，解耦业务代码和第三方具体实现，具体落地步骤如下：

1. 抽象依赖接口 #

把你业务中实际用到的第三方结构体的方法，提取为自定义接口，只需要包含你用到的方法即可，不需要完整复刻第三方库的所有能力：

package mypackage

import (
    "context"
    "go.mongodb.org/mongo-driver/mongo"
)

// 自定义Mongo操作接口，仅声明业务用到的FindOne方法
type MongoFinder interface {
    FindOne(ctx context.Context, filter interface{}) *mongo.SingleResult
}

2. 业务代码依赖接口而非具体实现 #

修改你的业务结构体，将依赖的类型改为上述自定义接口：

type myClientBeingTested struct {
    client MongoFinder // 依赖抽象接口，不直接绑定*mongo.Client
}

生产环境初始化时，原生的*mongo.Client天然实现了这个接口，直接传入即可，不需要做额外的包装改造：

// 生产环境初始化示例
func NewMyClient(mongoCli *mongo.Client) *myClientBeingTested {
    return &myClientBeingTested{
        client: mongoCli,
    }
}

3. 单元测试编写Mock实现 #

测试时只需要自己实现这个自定义接口，或者用mockgen等工具自动生成mock代码，就可以完全控制依赖的返回结果：

// 测试用Mock实现
type mockMongoFinder struct {
    mockFindOne func(ctx context.Context, filter interface{}) *mongo.SingleResult
}

func (m *mockMongoFinder) FindOne(ctx context.Context, filter interface{}) *mongo.SingleResult {
    return m.mockFindOne(ctx, filter)
}

测试用例示例：

func TestMyClientFindOne(t *testing.T) {
    // 构造mock实例，自定义FindOne返回逻辑
    mockCli := &mockMongoFinder{
        mockFindOne: func(ctx context.Context, filter interface{}) *mongo.SingleResult {
            // 这里可以根据测试场景返回你需要的结果，比如构造ErrNoDocuments、自定义错误等
            return &mongo.SingleResult{}
        },
    }
    testClient := &myClientBeingTested{client: mockCli}
    
    // 执行测试逻辑，断言结果符合预期
    res, err := testClient.FindOne(bson.M{"id": 1})
    // 后续断言逻辑...
}

方案优势 #

• 完全符合Go语言依赖倒置的设计理念，是生态公认的标准实践，不属于hack方案
• 没有全局变量污染，支持测试用例并行执行，不会出现多个用例修改全局变量互相影响的问题
• 扩展性强，如果后续需要替换Mongo驱动，只需要新的驱动实现你定义的接口即可，不需要修改业务代码
• 接口仅包含业务用到的方法，保持最小依赖，符合接口隔离原则

你之前使用全局变量替换的方案最大的问题是不支持并行测试，而且全局变量的修改如果忘记回滚，会影响其他测试用例的执行结果，仅适合小型临时场景，不是通用的标准方案。

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