使用依赖注入进行单元测试可提高可测试性和可维护性。使用 wire 框架设置依赖关系，定义接口并通过 newset 函数创建 injector。利用 newinvoker 函数测试依赖函数，并通过模拟依赖关系验证函数是否正确调用并返回预期结果。
Go 框架中使用依赖注入进行单元测试
简介
依赖注入是一种将依赖关系传递给对象的软件设计模式。它允许你创建松散耦合的代码，提高可测试性和可维护性。在 Go 框架中使用依赖注入进行单元测试可以使测试更简单、更可靠。
立即学习“go语言免费学习笔记（深入）”；
设置
为了使用依赖注入进行单元测试，需要使用一个依赖注入框架。推荐使用 [wire](https://github.com/google/wire)。它是一个由 Google 开发的轻量级、高性能的依赖注入框架。
使用 wire，首先需要定义依赖关系提供的接口：
type UserRepository interface {
Get(id int) (*User, error)
}然后，可以通过 wire.NewSet 函数创建一个 Injector，该函数传入依赖项以及提供它们的函数：
func InitializeUserRepository(db *sql.DB) UserRepository {
return NewUserRepository(db)
}
var Injector wire.Injector
func setup() {
// 使用 sqlmock 进行模拟
db, mock := sqlmock.New()
defer db.Close()
Injector = wire.NewSet(InitializeUserRepository, wire.Value(db), wire.Value(mock))
}实战案例
假设有一个 GetUser 函数，它需要一个 UserRepository：
func GetUser(id int) (*User, error) {
userRepo := Injector.MustInstance(new(UserRepository)).(UserRepository)
return userRepo.Get(id)
}要测试 GetUser 函数，可以使用 wire 提供的 wire.NewInvoker 函数，它接收 GetUser 函数和其依赖项作为参数：
func TestGetUser(t *testing.T) {
want := &User{ID: 1, Name: "John"}
setup()
mock.ExpectQuery("SELECT \\* FROM users WHERE id = \\?").WithArgs(1).WillReturnRows(
sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "John"),
)
got, err := wire.NewInvoker(GetUser, Injector)(1)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if !reflect.DeepEqual(got, want) {
t.Errorf("Expected user %v, got %v", want, got)
}
}在这个测试中，我们使用 sqlmock 模拟数据库交互。通过将期望的查询结果添加到 mock 中，我们可以验证 GetUser 函数是否正确调用了存储库并且返回了预期的结果。
