一文搞懂Go语言单元测试

简介: 一文搞懂Go语言单元测试

文章目录



前言


平时根据需求写代码.人工进行测试往往不会面面俱到,还会因为需求的改变繁琐的进行测试


通过完成一个测试函数,可以大大简化测试的步骤,并且在需求该变的时候只需要改变一下测试的输入与期望


一、单元测试分类及其概念


1.基本分类


  • 测试函数 函数前缀为Test 主要用于测试程序的一些逻辑行为是否正确
  • 基准函数 函数名前缀为Benchmark 主要测试函数的性能
  • 示例函数 函数的前缀名为 Example 为文档提示示例文档


2.细说单元测试分类


①测试函数


  • 函数的基本测试
  • 函数的组测试
  • 函数的子测试
  • 测试函数覆盖率(也就是被测试函数有多少代码用于了执行)
  • 在进行测试的时候,要保证被测函数覆盖率为100%,测试函数覆盖率为60%以上,否则写的代码大部分用不到,需要优化
  • ①测试覆盖率可以使用go test -cover
  • ②go test -cover -coverprofile=c.out(将测试的结果存入文件c.out)
  • 然后使用go tool cover -html=c.out可以打开文件,显示哪些代码没有执行


②基准测试


  • 基准函数测试是做什么的:基准函数会自定义一个时间段用于执行代码,如果代码简洁,被测函数的执行次数需要成倍增加(直到达到基准测试函数给的预期,然后统计一共执行了多少轮,,每轮平均用时多少)
  • 执行基准函数时,首先要写基准函数
  • 基准测试函数的参数是:*testing.B对应的指针变量
  • 在测试函数内部要进行循环,循环的终止条件是b.N


③示例函数


二、结合代码细说每一种测试


1.基准测试


(1)基准测试函数需要注意的点


  进行基准测试时往往是对函数的算法进行测验,有时后一个算法在测试数据的基量不同时测试出的效果会不同
  我们需要对不同数量级的样本进行测试的时候。可以自己写一个盒子作为跳板,测试什么数量级的数据就只
  进行当前数量级的测试
  测试的时候命令为:go test -bench=.(将所有的跳板函数均执行一遍)
          或者=特定函数进行特定函数测试
          -benchtime=time 
          (在被测试函数在基准函数默认的时间内执行不完时可以拿这个参数进行时间的扩充)
  在进行基准函数测试的时候,可能需要进行一些前置工作,如果感觉前置工作浪费时间可以使用
  b.ResetTimer()对计时器进行重置


(2)基准测试代码


测试函数代码如下:


//文件名 subStr_test.go
package main
import (
  "reflect"
  "testing"
)
func BenchmarkSubStr(b *testing.B) {
  for i := 0; i < b.N; i++ {
    res := subStr("qwe:qw:es:wqe", ":")
    if !reflect.DeepEqual(res, []string{"qwe", "qw", "es", "wqe"}) {
      b.Errorf("不匹配")
    }
  }
}
func benchmarkFib(b *testing.B, n int) {
  for i := 0; i < b.N; i++ {
    Fib(n)
  }
}
func BenchmarkFib5(b *testing.B)  { benchmarkFib(b, 5) }
func BenchmarkFib10(b *testing.B) { benchmarkFib(b, 10) }
func BenchmarkFib15(b *testing.B) { benchmarkFib(b, 15) }
func BenchmarkFib20(b *testing.B) { benchmarkFib(b, 20) }


被测函数代码如下:


//文件名 subStr.go
package main
import (
  "fmt"
  "strings"
)
func subStr(str, stre string) []string {
  index := strings.Index(str, stre)
  var theSub []string
  for index >= 0 {
    // 先将分隔符前面的数据进行保存
    temp := str[:index]
    // 将字符串往后移
    str = str[index+1:]
    // 重新获取下标
    index = strings.Index(str, stre)
    if temp != "" {
      theSub = append(theSub, temp)
    } else {
      continue
    }
  }
  theSub = append(theSub, str)
  return theSub[:]
}
// 菲薄那契数列
func Fib(n int) int {
  if n < 2 {
    return n
  }
  return Fib(n-1) + Fib(n-2)
}
func main() {
  fmt.Println(subStr("q:w:ec:wer:cd:scn:cj:c:is:icc:cin:si", ":"))
  fmt.Printf("%#v\n", subStr("q:w:ec:wer:cd:scn:cj:c:is:icc:cin:si", ":"))
  fmt.Println(123)
}


  将以上两个文件放在同一目录下
  执行测试命令 go test
  得到测试结果
  goos: windows
  goarch: amd64
  cpu: Intel(R) Core(TM) i5-8265U CPU @ 1.60GHz
  BenchmarkSubStr-8   1213681    1012 ns/op   352 B/op  14 allocs/op
  PASS
  ok    _/d_/Go语言学习笔记/go语言语法部分/go语言进阶语法/8.单元测试/基准测试 2.410s


2.组测试与子测试


  对于一个函数的测试,不仅仅是一组测试用例,所以我们需要写测试组,
  对一个函数进行测试,测试的时候一般用集合键值对构成,每一个键对应一组测试数据
  进行组测试的时候可能会有个别的组出错,所以使用子测试可以对某个案例进行单独测试
  这就是测试组与子测试的产生背景


测试函数如下:


package main
import (
  "reflect"
  "testing"
)
type testS struct {
  str  string
  ste  string
  want []string
}
//进行组测试,批量进行测试,如果有哪个地方不满足则进行异常抛出
// func TestSubStr(t *testing.T) {
//  testMap := map[string]testS{
//    "case_1": {"123:eqwe:123", ":", []string{"123", "eqwe", "123"}},
//    "case_2": {"13:eqwe3:1@23", "3:", []string{"1", "eqwe", "1@23"}},
//    "case_3": {"12@3@:@eq@we@:1@23", "@:", []string{"12@3", "@eq@we", "1@23"}},
//    "case_4": {"123:eq@we:1@2@3", "q@", []string{"123:e", "we:1@2@3"}},
//    "case_5": {"123:eqwe:123", "2", []string{"1", "3:eqwe:1", "3"}},
//    "case_6": {"123:eqwe:123", "eqwe", []string{"123:", ":123"}},
//  }
//  for k, v := range testMap {
//    res := subStr(v.str, v.ste)
//    if !reflect.DeepEqual(res, v.want) {
//      t.Errorf("%v want:%#v got:%#v", k, v.want, res)
//    }
//  }
// }
// 进行子测试,可以针对某个子样本进行测试
func TestSubStr(t *testing.T) {
  testMap := map[string]testS{
    "case_1": {"13:eqwe:123", ":", []string{"123", "eqwe", "123"}},
    "case_2": {"3:eqwe3:1@23", "3:", []string{"1", "eqwe", "1@23"}},
    "case_3": {"2@3@:@eq@we@:1@23", "@:", []string{"12@3", "@eq@we", "1@23"}},
    "case_4": {"123:eq@we:1@2@3", "q@", []string{"123:e", "we:1@2@3"}},
    "case_5": {"23:eqwe:123", "2", []string{"1", "3:eqwe:1", "3"}},
    "case_6": {"123:eqwe:123", "eqwe", []string{"123:", ":123"}},
  }
  for k, v := range testMap {
    t.Run(k, func(t *testing.T) {
      res := subStr(v.str, v.ste)
      if !reflect.DeepEqual(res, v.want) {
        t.Errorf("want:%#v got:%#v", v.want, res)
      }
    })
  }
}


待测函数如下:


package main
import (
  "fmt"
  "strings"
)
func subStr(str, stre string) []string {
  index := strings.Index(str, stre)
  var theSub []string
  for index >= 0 {
    // 先将分隔符前面的数据进行保存
    temp := str[:index]
    // 将字符串往后移
    str = str[index+len(stre):]
    // 重新获取下标
    index = strings.Index(str, stre)
    if temp != "" {
      theSub = append(theSub, temp)
    } else {
      continue
    }
  }
  theSub = append(theSub, str)
  return theSub[:]
}
func main() {
  fmt.Println(subStr("q:w:ec:wer:cd:scn:cj:c:is:icc:cin:si", ":c"))
  fmt.Printf("%#v\n", subStr("q:w:ec:wer:cd:scn:cj:c:is:icc:cin:si", ":"))
  fmt.Println(123)
}


(1).组测试结果分析及命令:


组测试命令仍使用 go test


--- FAIL: TestSubStr (0.00s)
  --- FAIL: TestSubStr/case_1 (0.00s)
     subStr_test.go:46: want:[]string{"123", "eqwe", "123"} got:[]string{"13", "eqwe", "123"}        
  --- FAIL: TestSubStr/case_2 (0.00s)
       subStr_test.go:46: want:[]string{"1", "eqwe", "1@23"} got:[]string{"eqwe", "1@23"}
  --- FAIL: TestSubStr/case_3 (0.00s)
       subStr_test.go:46: want:[]string{"12@3", "@eq@we", "1@23"} got:[]string{"2@3", "@eq@we", "1@23"}
  --- FAIL: TestSubStr/case_5 (0.00s)
       subStr_test.go:46: want:[]string{"1", "3:eqwe:1", "3"} got:[]string{"3:eqwe:1", "3"}
  FAIL
  exit status 1
  FAIL    _/D_/Go语言学习笔记/go语言语法部分/go语言进阶语法/8.单元测试/组测试与子测试     0.155s


(2).子测试结果分析及命令:


  例如单独测试case_1,使用的命令为
  go test -v -run=TestSubStr/case_1(等号后面跟的是上面组测试失败案例FAIL后面的目录)


=== RUN   TestSubStr
=== RUN   TestSubStr/case_1
    subStr_test.go:46: want:[]string{"123", "eqwe", "123"} got:[]string{"13", "eqwe", "123"}
--- FAIL: TestSubStr (0.00s)
    --- FAIL: TestSubStr/case_1 (0.00s)
FAIL
exit status 1
FAIL    _/D_/Go语言学习笔记/go语言语法部分/go语言进阶语法/8.单元测试/组测试与子测试     0.186s


三、pprof调试工具


1.对主函数进行传参


(1)os.Args


package main1
import (
  "fmt"
  "os"
)
func main() {
  // os.Args可以在执行函数的时候传递参数,但是对于-name=xxx没有办法将其解析
  if os.Args != nil {
    for index, temp := range os.Args {
      fmt.Println("第", index+1, "个参数是:", temp)
    }
  }
  fmt.Println("hello")
}


(2)flag.Args


相对于os.Args来说,flag.Args使用起来更加方便


package main
import (
  "flag"
  "fmt"
  "time"
)
func main() {
  // 第一个参数为对应的参数,第二个参数为默认值,第三个参数为提示
  // 返回的是对应类型的指针
  // name := flag.String("name", "Tom", "输入name")
  // sex := flag.Bool("sex", true, "是不是男性")
  // age := flag.Int("age", 10, "年龄")
  // flag.Parse()
  // fmt.Println(*name, *sex, *age)
  var name string
  var age int
  var sex bool
  flag.StringVar(&name, "name", "Tom", "输入name")
  flag.BoolVar(&sex, "sex", true, "是不是男性")
  flag.IntVar(&age, "age", 10, "年龄")
  tim := flag.Duration("time", time.Hour, "时间")
  // 将输入的数据进行解析,不使用这句话的话不能获取到name sex age等属性对应的值
  flag.Parse()
  fmt.Println(name, sex, age, *tim)
  fmt.Println(flag.Args())  //以切片的方式返回命令行之外的其他参数
  fmt.Println(flag.NArg())  //返回命令行之外的其他参数的个数
  fmt.Println(flag.NFlag()) //返回使用命令行的参数的个数
}


2.pprof性能调优


pprof调试工具进行调试,主要看待测模块在内存于时间上的效益
调试的时候只会显示耗费时间空间较多的代码段.
生成调试代码块的文件:go run xx.exe -cpu....
使用go语言工具查看代码块所存在的问题:go tool pprof cpu.pprof


package main
import (
  "flag"
  "fmt"
  "os"
  "runtime/pprof"
  "time"
)
// 一段有问题的代码
func logicCode() {
  var c chan int
  for {
    select {
    case v := <-c:
      fmt.Printf("recv from chan, value:%v\n", v)
    default:
    }
  }
}
func main() {
  var isCPUPprof bool
  var isMemPprof bool
  //获取接收到的参数
  flag.BoolVar(&isCPUPprof, "cpu", false, "turn cpu pprof on")
  flag.BoolVar(&isMemPprof, "mem", false, "turn mem pprof on")
  flag.Parse()
  //判断进行什么测试,然后执行分支,并将信息保存到相应的文件内。
  if isCPUPprof {
    file, err := os.Create("./cpu.pprof")
    if err != nil {
      fmt.Printf("create cpu pprof failed, err:%v\n", err)
      return
    }
    pprof.StartCPUProfile(file)
    defer pprof.StopCPUProfile()
  }
  for i := 0; i < 8; i++ {
    go logicCode()
  }
  time.Sleep(20 * time.Second)
  if isMemPprof {
    file, err := os.Create("./mem.pprof")
    if err != nil {
      fmt.Printf("create mem pprof failed, err:%v\n", err)
      return
    }
    pprof.WriteHeapProfile(file)
    file.Close()
  }
}


总结


这里对常用的测试函数,子测试进行了详细的讲解,对于pprof参数调优仅仅做了个介绍,可以根据自己的兴趣在网上找一部分pprof的例子深入学习,进行自动化测试非常方便。

目录
打赏
0
0
0
0
3
分享
相关文章
监控局域网其他电脑:Go 语言迪杰斯特拉算法的高效应用
在信息化时代,监控局域网成为网络管理与安全防护的关键需求。本文探讨了迪杰斯特拉(Dijkstra)算法在监控局域网中的应用,通过计算最短路径优化数据传输和故障检测。文中提供了使用Go语言实现的代码例程,展示了如何高效地进行网络监控,确保局域网的稳定运行和数据安全。迪杰斯特拉算法能减少传输延迟和带宽消耗,及时发现并处理网络故障,适用于复杂网络环境下的管理和维护。
揭秘 Go 语言中空结构体的强大用法
Go 语言中的空结构体 `struct{}` 不包含任何字段,不占用内存空间。它在实际编程中有多种典型用法:1) 结合 map 实现集合(set)类型;2) 与 channel 搭配用于信号通知;3) 申请超大容量的 Slice 和 Array 以节省内存;4) 作为接口实现时明确表示不关注值。此外,需要注意的是,空结构体作为字段时可能会因内存对齐原因占用额外空间。建议将空结构体放在外层结构体的第一个字段以优化内存使用。
|
3月前
|
Go 语言入门指南:切片
Golang中的切片(Slice)是基于数组的动态序列,支持变长操作。它由指针、长度和容量三部分组成,底层引用一个连续的数组片段。切片提供灵活的增减元素功能,语法形式为`[]T`,其中T为元素类型。相比固定长度的数组,切片更常用,允许动态调整大小,并且多个切片可以共享同一底层数组。通过内置的`make`函数可创建指定长度和容量的切片。需要注意的是,切片不能直接比较,只能与`nil`比较,且空切片的长度为0。
Go 语言入门指南:切片
|
3月前
|
公司局域网管理系统里的 Go 语言 Bloom Filter 算法,太值得深挖了
本文探讨了如何利用 Go 语言中的 Bloom Filter 算法提升公司局域网管理系统的性能。Bloom Filter 是一种高效的空间节省型数据结构,适用于快速判断元素是否存在于集合中。文中通过具体代码示例展示了如何在 Go 中实现 Bloom Filter,并应用于局域网的 IP 访问控制,显著提高系统响应速度和安全性。随着网络规模扩大和技术进步,持续优化算法和结合其他安全技术将是企业维持网络竞争力的关键。
70 2
公司局域网管理系统里的 Go 语言 Bloom Filter 算法,太值得深挖了
|
3月前
|
【02】客户端服务端C语言-go语言-web端PHP语言整合内容发布-优雅草网络设备监控系统-2月12日优雅草简化Centos stream8安装zabbix7教程-本搭建教程非docker搭建教程-优雅草solution
【02】客户端服务端C语言-go语言-web端PHP语言整合内容发布-优雅草网络设备监控系统-2月12日优雅草简化Centos stream8安装zabbix7教程-本搭建教程非docker搭建教程-优雅草solution
109 20
eino — 基于go语言的大模型应用开发框架(二)
本文介绍了如何使用Eino框架实现一个基本的LLM(大语言模型)应用。Eino中的`ChatModel`接口提供了与不同大模型服务(如OpenAI、Ollama等)交互的统一方式,支持生成完整响应、流式响应和绑定工具等功能。`Generate`方法用于生成完整的模型响应,`Stream`方法以流式方式返回结果,`BindTools`方法为模型绑定工具。此外,还介绍了通过`Option`模式配置模型参数及模板功能,支持基于前端和用户自定义的角色及Prompt。目前主要聚焦于`ChatModel`的`Generate`方法,后续将继续深入学习。
472 7
企业监控软件中 Go 语言哈希表算法的应用研究与分析
在数字化时代,企业监控软件对企业的稳定运营至关重要。哈希表(散列表)作为高效的数据结构,广泛应用于企业监控中,如设备状态管理、数据分类和缓存机制。Go 语言中的 map 实现了哈希表,能快速处理海量监控数据,确保实时准确反映设备状态,提升系统性能,助力企业实现智能化管理。
48 3
eino — 基于go语言的大模型应用开发框架(一)
Eino 是一个受开源社区优秀LLM应用开发框架(如LangChain和LlamaIndex)启发的Go语言框架,强调简洁性、可扩展性和可靠性。它提供了易于复用的组件、强大的编排框架、简洁明了的API、最佳实践集合及实用的DevOps工具,支持快速构建和部署LLM应用。Eino不仅兼容多种模型库(如OpenAI、Ollama、Ark),还提供详细的官方文档和活跃的社区支持,便于开发者上手使用。
556 8
Go 语言中的 Sync.Map 详解:并发安全的 Map 实现
`sync.Map` 是 Go 语言中用于并发安全操作的 Map 实现,适用于读多写少的场景。它通过两个底层 Map(`read` 和 `dirty`)实现读写分离,提供高效的读性能。主要方法包括 `Store`、`Load`、`Delete` 等。在大量写入时性能可能下降,需谨慎选择使用场景。
阿里双十一背后的Go语言实践:百万QPS网关的设计与实现
解析阿里核心网关如何利用Go协程池、RingBuffer、零拷贝技术支撑亿级流量。 重点分享: ① 如何用gRPC拦截器实现熔断限流; ② Sync.Map在高并发读写中的取舍。
108 0
AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等