Golang测试辅助函数封装与示例

func mustCreateTempFile(t *testing.T, content string) string {
    t.Helper()
    file, err := os.CreateTemp("", "testfile-")
    if err != nil {
        t.Fatalf("无法创建临时文件: %v", err)
    }
    defer file.Close()
    if _, err := file.WriteString(content); err != nil {
        t.Fatalf("无法写入文件: %v", err)
    }
    return file.Name()
}

func TestReadConfig(t *testing.T) {
    filename := mustCreateTempFile(t, `{"port": 8080}`)
    config, err := ReadConfig(filename)
    if err != nil {
        t.Errorf("期望无错误，但得到: %v", err)
    }
    if config.Port != 8080 {
        t.Errorf("期望端口为8080，实际为%d", config.Port)
    }
}

func assertErrorContains(t *testing.T, err error, want string) {
    t.Helper()
    if err == nil {
        t.Fatal("期望有错误，但得到 nil")
    }
    if !strings.Contains(err.Error(), want) {
        t.Errorf("错误信息期望包含 %q，但得到 %q", want, err.Error())
    }
}

_, err := ParseURL("invalid://")
assertErrorContains(t, err, "不支持的协议")

type UserBuilder struct {
    user User
}

func NewUser() *UserBuilder {
    return &UserBuilder{
        user: User{Age: 18, Active: true}, // 默认值
    }
}

func (b *UserBuilder) Name(name string) *UserBuilder {
    b.user.Name = name
    return b
}

func (b *UserBuilder) Age(age int) *UserBuilder {
    b.user.Age = age
    return b
}

func (b *UserBuilder) Build() User {
    return b.user
}

func TestValidateUser(t *testing.T) {
    user := NewUser().Name("Alice").Age(25).Build()
    if err := ValidateUser(user); err != nil {
        t.Errorf("有效用户不应报错，但得到: %v", err)
    }
}

func startTestServer(t *testing.T, handler http.Handler) *httptest.Server {
    t.Helper()
    server := httptest.NewServer(handler)
    // 自动在测试结束时关闭
    t.Cleanup(server.Close)
    return server
}

server := startTestServer(t, mux)
resp, err := http.Get(server.URL + "/health")
if err != nil {
    t.Fatal(err)
}
defer resp.Body.Close()