Golang单元测试基础与函数编写方法

math.go

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

math_test.go

package main

import "testing"

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("期望 5，但得到 %d", result)
    }
}

func TestAdd(t *testing.T) {
    tests := []struct {
        name     string
        a, b     int
        expected int
    }{
        {"正数相加", 2, 3, 5},
        {"负数相加", -1, -2, -3},
        {"零值测试", 0, 0, 0},
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            result := Add(tt.a, tt.b)
            if result != tt.expected {
                t.Errorf("期望 %d，但得到 %d", tt.expected, result)
            }
        })
    }
}

go test -coverprofile=coverage.out
go tool cover -html=coverage.out

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

var now = time.Now

func IsToday(t time.Time) bool {
    return now().Date == t.Date
}