golang学习笔记之单元测试和压力测试

文件结构:

math.go:待测试的包

math_test.go:单元测试

bench_test.go:压力测试

math.go

package main
 
func Add(a int, b int) int {
    return a + b
}
 
func Sub(a int, b int) int {
    return a - b
}

单元测试:

math_test.go

package main
 
import "testing"
 
func TestAdd(t *testing.T) {
    a := 10
    b := 20
    c := Add(a, b)
    if c != 30 {
        t.Fatalf("add func error, c = %d", c)
    }
    t.Logf("a = %d b = %d sum = %d\n", a, b, c)
}
 
func TestSub(t *testing.T) {
    a := 50
    b := 20
    c := Sub(a, b)
    if c != 30 {
        t.Fatalf("sub func error, c = %d", c)
    }
    t.Logf("a = %d b = %d sub = %d\n", a, b, c)
}

在文件所在的路径下,命令行执行

//go test -v    显示详细过程,也可以配合-run使用
//go test -run TestAdd 只测试TestAdd方法
//go test -run TestSub 只测试TestSub方法
go test 

压力测试:

bench_test.go

package main
 
import "testing"
 
//注意命名规范 Benchmark+首字母大写的方法名 参数固定
func BenchmarkAdd(b *testing.B) {
    for i := 0; i < b.N; i++ {
        a := 10
        b := 20
        Add(a, b)
    }
}
 
//注意命名规范 Benchmark+首字母大写的方法名 参数固定
func BenchmarkSub(b *testing.B) {
    for i := 0; i < b.N; i++ {
        a := 10
        b := 20
        Sub(a, b)
    }
}

在文件所在的路径下,命令行执行

//go test -bench BenchmarkAdd 只压测BenchmarkAdd方法
//go test -bench BenchmarkSub 只压测BenchmarkSub方法
//go test -bench .    点表示测试该路径下所有压力测试
go test -bench .