Golang单元测试Mock数据库依赖注入失败求助

核心问题分析 #

你的问题本质是user包的Handler实例没有正确关联到测试中替换的Mock DB实例。虽然在setup_test.go中修改了main.app.DB，但user.New创建Handler时可能直接绑定了初始化时的真实DB，或者Handler内部的DB引用和main.app.DB不是同一个变量，导致测试时Handler持有的DB仍然是nil。

具体修复方案 #

1. 让Handler通过构造函数接收DB实例 #

修改user包的New函数，不要依赖全局的main.app.DB，而是将DB实例作为参数传入：

// user包中的New函数
func New(db dbrepo.DBRepo) *Handler {
    return &Handler{
        DB: db,
    }
}

main包初始化时传入真实DB：

// main包初始化路由逻辑
func main() {
    realDB := dbrepo.NewRealDBRepo(...)
    userHandler := user.New(realDB)
    // 注册路由...
}

2. 测试时直接传入Mock DB到Handler #

在setup_test.go中无需修改main.app.DB，直接创建Mock实例并传入user.New：

// setup_test.go
func setupTest() *user.Handler {
    mockDB := &dbrepo.TestDBRepo{/* 初始化Mock数据 */}
    return user.New(mockDB)
}

测试用例中直接使用这个Handler：

func TestSignUp(t *testing.T) {
    handler := setupTest()
    // 执行测试逻辑，调用handler.SignUp...
}

3. 移除跨包全局变量依赖 #

如果user包的Handler依赖了全局的main.app变量，必须彻底移除这种依赖。所有资源（DB、配置等）都要通过构造函数注入，这是多包结构下保证依赖可替换的核心原则。

4. 验证Mock实现是否符合接口要求 #

确保dbrepo.TestDBRepo完全实现了dbrepo.DBRepo接口，避免因接口不匹配导致注入失效。可以在测试文件中添加编译期检查：

var _ dbrepo.DBRepo = &dbrepo.TestDBRepo{}

这行代码会在编译时验证TestDBRepo是否满足DBRepo接口的所有方法定义。

关键注意事项 #

• 多包结构下，跨包依赖全局变量是依赖注入失效的常见诱因，必须杜绝。
• 单元测试的核心是隔离依赖，每个测试用例应独立创建专属的Mock实例，而非依赖修改全局状态。

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