技术改变生活

Technology changes life,Life-long learning

Go 语言 Unit Testing 单元测试

关于 Go 的基本语法,参见:半天时间 Go 语言的基本实践

单元测试

Go 中提供了 testing 这个 package 和 go test 这个命令来进行单元测试。

假设我们要创建一个数学运算的 package,名字叫 algo,提供了加法运算 add() 和减法运算 sub()
首先我们创建一个 algo 的文件夹,在里面创建一个 add.go 文件和 sub.go 文件,分别如下:

package algo  func add(x int, y int) int {    return x + y } 
package algo  func sub(x int, y int) int {     return x - y } 

现在我们要对 add()sub() 方法进行单元测试,需要有如下几步:

  • 创建一个测试文件,文件名以 XXX_test.go 结尾,例如 algo_test.go
    • 该测试文件放在被测试文件相同的 package 中:package algo
    • Go doesn’t ship your tests。该测试文件不会会编译器放到编译后的 Binary 中
  • 导入 testing 这个 package:import "testing"
  • 创建方法 func TestXxx(t *testing.T),在这里面定义测试:
    • func TestAdd(t *testing.T)
    • func TestSub(t *testing.T)
  • 使用 t.Errort.Fail 等方法来标记测试错误
    • 可以使用 t. Errorf 来打印详情
    • 可以使用 t.Log 来打印日志文件
  • 最后使用 go test 命令来执行单元测试

我们创建 algo_test.go 测试文件如下:

package algo  import "testing"  func TestAdd(t *testing.T) {         want := 3         got := add(1, 2)         if got != want {                 t.Errorf("add was incorrect, got: %d, want: %d.", got, want)         } }  func TestSub(t *testing.T) {         want := -1         got := sub(1, 2)         if got != want {                 t.Errorf("sub was incorrect, got: %d, want: %d.", got, want)         } } 

最后使用 go test 命令来执行单元测试:

《Go 语言 Unit Testing 单元测试》

go test

如果我们故意把 sub 方法写错,go test 命令会标记测试错误:

《Go 语言 Unit Testing 单元测试》

go test

如何创建多个测试用例?

在上述的代码中,我们为 add() 方法只创建了一个测试用例 add(1, 2) = 3
我们可以通过 Go 中的 slice 来创建多个测试用例:

func TestAdd(t *testing.T) {         cases := []struct {                 p1 int                 p2 int                 want int         } {                 {p1: 1, p2: 2, want: 3},                 {p1: 3, p2: 4, want: 7},                 {p1: 5, p2: 6, want: 11},         }          for _, tc := range cases {                 got := add(tc.p1, tc.p2)                 if got != tc.want {                         t.Errorf("add was incorrect, got: %d, want: %d.", got, tc.want)                 }         } } 

如何得到语句覆盖率

使用 go test -cover 命令:

《Go 语言 Unit Testing 单元测试》

go test -cover

Mockup 依赖

假设我们的 add() 方法在返回 x + y 的时候,需要再乘以一个系数,这个系数通过一个服务得到。

  • 这个服务对应的接口叫做 ParamServiceInterface
  • 这个服务的真正实现叫做 ParamServiceImpl
package algo  type ParamServiceInterface interface {     getParam() int }  type ParamServiceImpl struct {     // 依赖的一些其他 service }  func (impl ParamServiceImpl) getParam() int {     // 复杂的计算逻辑     // return ....     return 1 }  func add(x int, y int, paramService ParamServiceInterface) int {    return paramService.getParam() * (x + y) } 

我们在做单元测试的时候,无法得到实现类 ParamServiceImpl 来传入 add() 方法。
因此可以 Mock 出一个服务,代码如下:

type ParamServiceFake struct {     param int }  func (fake ParamServiceFake) getParam() int {     return fake.param }  func TestAdd(t *testing.T) {         cases := []struct {         p1 int         p2 int         fake ParamServiceFake         want int     } {         {p1: 1, p2: 2, fake: ParamServiceFake{1}, want: 3},         {p1: 3, p2: 4, fake: ParamServiceFake{2}, want: 14},         {p1: 5, p2: 6, fake: ParamServiceFake{3}, want: 33},     }      for _, tc := range cases {         got := add(tc.p1, tc.p2, tc.fake)         if got != tc.want {                     t.Errorf("add was incorrect, got: %d, want: %d.", got, tc.want)             }     } } 

引用:
testing – The Go Programming Language

原文始发于:Go 语言 Unit Testing 单元测试

点赞

发表评论

电子邮件地址不会被公开。 必填项已用*标注