单元测试

意义

  1. 确保每个函数是可运行，并且运行结果是正确的
  2. 确保写出来的代码性能是好的
  3. 单元测试能及时的发现程序设计或实现的逻辑错误，使问题暴露，便于问题的定位解决，而性能测试的重点在于发现程序设计上的一些问题，让程序能够在高并发的情况下还能保持稳定

Go 语言推荐测试文件和源代码文件放在一块，测试文件以 _test.go 结尾

需要注意的是

  1. 测试用例文件名必须以_test.go结尾
  2. 测试用例函数必须以Test开头，一般来说就是Test+被测试的函数名
  3. 文件夹不能用中文，否则使用终端 cd 到文件夹下，使用以下命令
bash
go test calc_test.go calc.go -v
=== RUN   TestAdd
    calc_test.go:8: 测试失败
--- FAIL: TestAdd (0.00s)
FAIL
FAIL    command-line-arguments  1.078s
FAIL

例子

源文件

go
package main


func Add(n1, n2 int) int {
  return n1 + n2 + 1
}

测试文件

go
package main


import "testing"


func TestAdd(t *testing.T) {
  res := Add(1, -1)
  if res != 0 {
    t.Errorf("测试失败")
    return
  }
  t.Logf("测试通过")
}

-v 参数会显示每个用例的测试结果

-run参数可以指定测试某个函数

单元测试框架提供的日志方法

方 法 备 注 测试结果
Log 打印日志，同时结束测试 PASS
Logf 格式化打印日志，同时结束测试 PASS
Error 打印错误日志，同时结束测试 FAIL
Errorf 格式化打印错误日志，同时结束测试 FAIL
Fatal 打印致命日志，同时结束测试 FAIL
Fatalf 格式化打印致命日志，同时结束测试 FAIL

如果不同的测试用例每次都这么写很麻烦

子测试

如果需要给一个函数，调用不同的测试用例，可以使用子测试

子测试里面的Fatal，是不会终止程序的

go
package main


import "testing"


func TestAdd(t *testing.T) {
  t.Run("add1", func(t *testing.T) {
    if Add(1, -1) != 1 {
      t.Errorf("测试失败")
      return
    }


  })
  t.Run("add1", func(t *testing.T) {
    if Add(1, -1) != 2 {
      t.Errorf("测试失败")
      return
    }
  })


  //res := Add(1, -1)
  //if res != 0 {
  //  t.Errorf("测试失败")
  //  return
  //}
  //t.Logf("测试通过")
}

如果测试用例很多，还可以用一个类似表格去表示

go
package main


import (
  "testing"
)


func TestAdd(t *testing.T) {
  cases := []struct {
    Name           string
    A, B, Expected int
  }{
    {"a1", 2, 3, 5},
    {"a2", 2, -3, -1},
    {"a3", 2, 0, 2},
  }
  for _, s := range cases {
    t.Run(s.Name, func(t *testing.T) {
      if Add(s.A, s.B) != s.Expected {
        t.Errorf("测试失败")
        return
      }


    })
  }
}

如果用例较多，还可以通过读取 csv 的文件进行测试

TestMain函数

它是测试的入口

我们可以在TestMain里面实现测试流程的生命周期

go
package main


import (
  "fmt"
  "os"
  "testing"
)


// 测试前执行
func setup() {
  fmt.Println("Before all tests")
}


// 测试后执行
func teardown() {
  fmt.Println("After all tests"
}


func Test1(t *testing.T) {
  fmt.Println("I'm test1")
}


func Test2(t *testing.T) {
  fmt.Println("I'm test2")
}


// 必须叫这个名字  测试主入口
func TestMain(m *testing.M) {
  // 测试前执行
  setup()
  code := m.Run()
  // 测试后执行
  teardown()
  os.Exit(code)
}