在写每个功能的时候,都会涉及到编写测试代码。那是因为 TDD(Test-Driven Development,测试驱动开发)中提倡先编写测试代码,然后再编写功能代码,每做一个修改后,都要执行一次单元测试和基准测试,以此来验证功能和性能是否有问题。
特别是业务系统代码经常变更,单元测试和基准测试也就显得非常重要。而每种语言都有自己的测试框架,比如 Go 语言,它是一门注重工程效率的语言,有着非常强大的工具链,它自带的测试框架,能满足我们大部分测试要求。
所以,这里介绍如何使用 Go 测试框架做性能测试中的单元测试和基准测试。
单元测试
Go 测试框架中支持白盒测试和黑盒测试。现在我就以 xx.go 这个文件为例,给你详细介绍下如何做单元测试。
总体步骤
总的来说,用 Go 测试框架做单元测试主要有这几个步骤。
第一,Go 测试框架要求测试代码文件名以 _test.go 结尾。为了测试 cache.go,我们需要在 infrastructure/stores 目录下创建一个 cache_test.go 文件。
第二,cache_test.go 中第一行如果是 package stores,则表示该测试是白盒测试,这意味着除了这个包的全局函数外,你还可以测试它的私有函数;如果是 package stores_test,则表示黑盒测试,你只可以测试全局函数,里面的具体实现对于你来说是个黑盒子。
第三,Go 测试框架要求单元测试函数需要以 Test 开头。为了测试 IntCache 和 ObjCache,我们需要实现 TestIntCache 和 TestObjCache 这两个函数,它们的参数类型都是 testing.T 指针。
第四,在测试过程中,如果发现错误,可以通过测试框架的 Error 方法或者 Fatal 方法输出错误。不同的是,Error 方法仅仅输出错误,而 Fatal 方法却会结束当前测试。
第五,在终端进入项目根目录下,执行 go test ./infrastructure/stores 命令,将会执行 infrastructure/stores 目录下的所有单元测试。