Golang 1.19测试Mock对象：Mongo客户端接口匹配问题求助

方案1：包装原生Mongo类型适配接口 #

问题核心是mongo.Collection的FindOne返回*mongo.SingleResult，与你定义的ISingleResult接口类型不匹配，导致原生集合无法实现ICollection接口。可以通过适配层结构体包装原生类型，让它们符合接口要求：

1. 实现ISingleResult的适配结构体 #

把原生*mongo.SingleResult包装成符合接口的类型：

type MongoSingleResultAdapter struct {
    sr *mongo.SingleResult
}

func (a *MongoSingleResultAdapter) Decode(v interface{}) error {
    return a.sr.Decode(v)
}

这个适配器完全复用原生SingleResult的Decode逻辑，同时实现了ISingleResult接口。

2. 实现ICollection的适配结构体 #

包装原生mongo.Collection，让它的FindOne返回适配后的结果：

type MongoCollectionAdapter struct {
    coll *mongo.Collection
}

func (a *MongoCollectionAdapter) FindOne(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) ISingleResult {
    sr := a.coll.FindOne(ctx, filter, opts...)
    return &MongoSingleResultAdapter{sr: sr}
}

现在MongoCollectionAdapter实现了ICollection接口，内部调用原生集合方法，同时返回适配后的结果。

3. 业务代码与测试Mock #

业务客户端需依赖ICollection接口而非具体类型：生产环境传入MongoCollectionAdapter包装的实例；测试时使用自定义Mock：

// Mock Collection实现
type MockCollection struct {
    FindOneFunc func(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) ISingleResult
}

func (m *MockCollection) FindOne(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) ISingleResult {
    return m.FindOneFunc(ctx, filter, opts...)
}

// Mock SingleResult实现
type MockSingleResult struct {
    DecodeFunc func(v interface{}) error
}

func (m *MockSingleResult) Decode(v interface{}) error {
    return m.DecodeFunc(v)
}

测试时可自定义FindOneFunc和DecodeFunc的逻辑，轻松控制返回结果、模拟错误，还能跟踪调用次数和参数。

方案2：使用Mock库简化实现 #

如果不想手动编写适配器和Mock，可以用成熟的Go Mock库（如testify/mock或gomock），它们能自动生成符合接口的Mock实现，支持调用断言、参数校验等功能：

1. 确保接口定义正确
2. 用工具生成Mock代码
3. 测试中设置期望调用和返回值

这种方式省去重复代码，适合接口较多的场景。

关键注意事项 #

• 业务代码必须面向接口编程，所有Mongo操作依赖自定义接口而非原生类型
• 适配器层仅做类型转换，不添加额外业务逻辑，保证生产环境行为与原生客户端一致
• 测试时通过Mock接口完全隔离数据库依赖，无需启动真实Mongo实例

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