Golang如何进行单元测试与集成测试_Golang 单元测试集成实践

// math.go
package calc

func Add(a, b int) int {
    return a + b
}

// math_test.go
package calc

import "testing"

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("Add(2, 3) = %d; want 5", result)
    }
}

func TestAdd(t *testing.T) {
    tests := []struct {
        name     string
        a, b     int
        expected int
    }{
        {"positive", 2, 3, 5},
        {"negative", -1, -1, -2},
        {"zero", 0, 0, 0},
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            if got := Add(tt.a, tt.b); got != tt.expected {
                t.Errorf("Add(%d, %d) = %d; want %d", tt.a, tt.b, got, tt.expected)
            }
        })
    }
}

// api_integration_test.go
//go:build integration
// +build integration

package main

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

func TestUserEndpoint(t *testing.T) {
    server := httptest.NewServer(SetUpRouter()) // 模拟HTTP服务
    defer server.Close()

    resp, err := http.Get(server.URL + "/user/1")
    if err != nil {
        t.Fatal(err)
    }
    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected status 200, got %d", resp.StatusCode)
    }
}

func BenchmarkAdd(b *testing.B) {
    for i := 0; i < b.N; i++ {
        Add(2, 3)
    }
}