go基础-17.单元测试-CSDN博客

本文链接：https://blog.csdn.net/csdngzy/article/details/142387326

Go语言中自带有一个轻量级的测试框架testing和自带的go test命令来实现单元测试和性能测试，testing框架和其他语言的测试框架相似，可以基于这个框架写针对相应函数的测试用例，也可以基于该框架写相应的压力测试用例。通过单元测试，可以解决：

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

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

需要注意的是

测试用例文件名必须以_test.go结尾
测试用例函数必须以Test开头，一般来说就是Test+被测试的函数名

单元测试

例如我现在有两个用于计算的文件，叫calc.go

package main

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

func Mul(a int, b int) int {
    return a * b
}

那么我的测试文件就是calc_test.go

package main

import "testing"

func TestAdd(t *testing.T) {
  if ans := Add(1, 2); ans != 3 {
    // 如果不符合预期，那就是测试不通过
    t.Errorf("1 + 2 expected be 3, but %d got", ans)
  }

  if ans := Add(-10, -20); ans != -30 {
    t.Errorf("-10 + -20 expected be -30, but %d got", ans)
  }
}

测试的话，就在界面上点绿色的按钮就可以了

如果是命令的话

go test // 可以运行某个包下的所有测试用例

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

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

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

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

子测试

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

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

package main

import (
  "testing"
)

func TestAdd(t1 *testing.T) {
  t1.Run("add1", func(t *testing.T) {
    if ans := Add(1, 2); ans != 3 {
      // 如果不符合预期，那就是测试不通过
      t.Fatalf("1 + 2 expected be 3, but %d got", ans)
    }
  })
  t1.Run("add2", func(t *testing.T) {
    if ans := Add(-10, -20); ans != -30 {
      t.Fatalf("-10 + -20 expected be -30, but %d got", ans)
    }
  })

}

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

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 _, c := range cases {
    t.Run(c.Name, func(t *testing.T) {
      if ans := Add(c.A, c.B); ans != c.Expected {
        t.Fatalf("%d * %d expected %d, but %d got",
          c.A, c.B, c.Expected, ans)
      }
    })
  }
}

TestMain函数

它是测试的入口

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

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)
}