探索 Go 框架中常见的测试模式和实践

import "testing"

func TestAdd(t *testing.T) {
    tests := []struct {
        a, b int
        want int
    }{
        {1, 2, 3},
        {-5, 10, 5},
    }

    for _, test := range tests {
        got := Add(test.a, test.b)
        if got != test.want {
            t.Errorf("Add(%d, %d) = %d; want %d", test.a, test.b, got, test.want)
        }
    }
}

package main

import (
    "context"
    "net/http/httptest"
    "testing"

    "github.com/gorilla/mux"
)

func TestHomeHandler(t *testing.T) {
    // 设置路由器和 HTTP 处理程序
    router := mux.NewRouter()
    router.HandleFunc("/", HomeHandler)

    // 创建测试服务器
    rr := httptest.NewRecorder()
    req, err := http.NewRequest("GET", "/", nil)
    if err != nil {
        t.Fatalf("Could not create request: %v", err)
    }

    // 执行 HTTP 请求
    router.ServeHTTP(rr, req)

    // 检查响应状态
    if status := rr.Code; status != 200 {
        t.Errorf("HomeHandler returned wrong status code: got %v want %v", status, http.StatusOK)
    }

    // 检查响应内容
    if rr.Body.String() != "Hello, world!" {
        t.Errorf("HomeHandler returned wrong body: got %v want %v", rr.Body.String(), "Hello, world!")
    }
}

func main() {
    ctx := context.Background()
    if err := startServer(ctx); err != nil {
        log.Fatalf("Could not start server: %v", err)
    }
}