golang单元测试如何管理测试数据？

管理 Go 单元测试中的测试数据涉及使用 fixtures、内建函数和第三方库来初始化、清理和生成数据。最佳实践包括：1. 使用 fixtures 加载预定义数据集；2. 使用 TempDir 和 TempFile 创建临时文件和目录；3. 使用 TestDB 和 Gomock 等第三方库生成 mock 数据；4. 在每个测试完成后清理创建的数据。实战案例中演示了如何使用 testfixtures 和 go-sqlmock 为 UserRepository 的 GetUser 方法编写单元测试。

Go 单元测试中的测试数据管理

在编写 Go 单元测试时，管理测试数据至关重要，它有助于确保测试的准确性和可靠性。本文将指导您了解在 Go 中管理测试数据的最佳实践，并提供一个实战案例。

最佳实践

• 使用 fixtures: Fixtures 是预定义的数据集，用于初始化测试环境。它们通常存储在文件中，并在测试开始前加载。
• 使用内建函数: Go 标准库提供了 TempDir 和 TempFile 函数，用于创建临时目录和文件。
• 使用第三方库: 诸如 [TestDB](https://github.com/DATA-DOG/go-sqlmock) 和 [Gomock](https://github.com/golang/mock) 等第三方库可以帮助生成 mock 数据。
• 清理数据: 在每个测试结束时，请务必清理任何创建的数据，以防止泄漏和干扰后续测试。

实战案例

假设我们有一个 UserRepository，它负责处理数据库中的用户操作。我们想为 GetUser 方法编写一个单元测试。

fixtures.json

{
  "users": [
    {
      "id": 1,
      "name": "Alice",
      "email": "alice@example.com"
    },
    {
      "id": 2,
      "name": "Bob",
      "email": "bob@example.com"
    }
  ]
}

user_test.go

import (
    "context"
    "encoding/json"
    "io/ioutil"
    "os"
    "strings"
    "testing"

    "github.com/DATA-DOG/go-sqlmock"
    "github.com/go-testfixtures/testfixtures"
)

func TestGetUser(t *testing.T) {
    // 初始化 testfixtures
    fixtures, err := testfixtures.New(t, ...testfixtures.Database)
    if err != nil {
        t.Fatalf("testfixtures.New: %v", err)
    }

    // 从 fixtures 中读取数据
    data, err := ioutil.ReadFile("fixtures.json")
    if err != nil {
        t.Fatalf("ioutil.ReadFile: %v", err)
    }
    fixtures.AddFile(t, strings.NewReader(string(data)), "users")

    // 准备 mock 数据库连接
    db, mock, err := sqlmock.New()
    defer db.Close()

    ctx := context.Background()
    repo := NewUserRepository(db)

    // 预期 SQL 查询并返回 mock 数据
    rows := sqlmock.NewRows([]string{"id", "name", "email"}).
        .AddRow(1, "Alice", "alice@example.com")
    mock.ExpectQuery("SELECT").WillReturnRows(rows)

    // 执行测试
    user, err := repo.GetUser(ctx, 1)
    if err != nil {
        t.Fatalf("repo.GetUser: %v", err)
    }

    // 断言结果与预期相符
    if user.ID != 1 || user.Name != "Alice" || user.Email != "alice@example.com" {
        t.Errorf("unexpected user: %v", user)
    }

    // 检查 mock 期望是否都已实现
    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("there were unfulfilled expectations: %v", err)
    }
}