Go 单测提效指南：盘点 GitHub 上实用的 Testing 与 Mock 开源工具

# 安装
go install github.com/cweill/gotests/gotests@latest

# 为整个 user.go 文件生成测试
gotests -all -w user.go

with-expecter: true
dir: "mocks/{{.PackagePath}}"
mockname: "Mock{{.InterfaceName}}"
outpkg: "mocks"
packages:
  github.com/yourname/project/service:
    interfaces:
      - UserRepository

func TestGetUser(t *testing.T) {
    mockRepo := mocks.NewMockUserRepository(t)
    // 链式调用，类型安全
    mockRepo.EXPECT().FindByID(1).Return(&model.User{Name: "小周"}, nil)

    svc := service.NewUserService(mockRepo)
    user, err := svc.GetUser(1)

    assert.NoError(t, err)
    assert.Equal(t, "小周", user.Name)
}

package test

import (
    "io"
    "net/http"
    "testing"
    "github.com/h2non/gock"
    "github.com/stretchr/testify/assert"
)

func TestExternalApi(t *testing.T) {
    defer gock.Off() // 测试完务必清理拦截器

    // 声明拦截规则
    gock.New("https://api.github.com").
        Get("/users/octocat").
        Reply(200).
        JSON(map[string]string{"login": "octocat", "name": "The Octocat"})

    // 实际业务调用
    res, err := http.Get("https://api.github.com/users/octocat")
    assert.NoError(t, err)
    assert.Equal(t, 200, res.StatusCode)

    body, _ := io.ReadAll(res.Body)
    assert.Contains(t, string(body), "The Octocat")
    assert.True(t, gock.IsDone()) // 确保 Mock 命中
}

package test

import (
    "testing"
    "github.com/agiledragon/gomonkey/v2"
    "github.com/stretchr/testify/assert"
)

// 假设这是不可改造的底层包函数
func GetClusterConfig() string {
    return "production"
}

func TestLegacyCode(t *testing.T) {
    // 运行时给函数打桩
    patches := gomonkey.ApplyFunc(GetClusterConfig, func() string {
        return "mock-test-env"
    })
    defer patches.Reset() // 还原补丁

    assert.Equal(t, "mock-test-env", GetClusterConfig())
}

# 安装
go install gotest.tools/gotestsum@latest

# 用好看的格式运行所有测试
gotestsum --format pkgname

# 监听文件变动并自动跑单测 (超爽的开发循环体验)
gotestsum --watch