golang框架中如何使用依赖注入进行单元测试- 技术经验 -卓越飞翔博客

使用依赖注入进行单元测试可提高可测试性和可维护性。使用 wire 框架设置依赖关系，定义接口并通过 newset 函数创建 injector。利用 newinvoker 函数测试依赖函数，并通过模拟依赖关系验证函数是否正确调用并返回预期结果。

golang框架中如何使用依赖注入进行单元测试

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 函数是否正确调用了存储库并且返回了预期的结果。

相关推荐