专栏首页区块链入门【实践】Golang的单元测试入门go test

【实践】Golang的单元测试入门go test

go test命令,相信大家都不陌生,常见的情况会使用这个命令做单测试、基准测试和http测试。go test还是有很多flag 可以帮助我们做更多的分析,比如测试覆盖率,cpu分析,内存分析,也有很多第三方的库支持test,cpu和内存分析输出结果要配合pprof和go-torch来进行可视化显示。

go test文件命名是以_test.go为缀。例如userInfo_test.go。在github上写了一个小的项目,包含常见的测试方法: https://github.com/lpxxn/gotest 。app1里是基本的测试方法。app2里包含了一些第三方的库辅助我们更方便的测试。 大家可以下载来来 go test一下试试

测试函数以Test或者Bench为前缀开始,如:

func TestXXXXXXX(t *testing.T)
func BenchXXXXXX(b *testing.B)
func TestMain(m *testing.M)

看一下testing.T和testing.B者有组合 common

>type T struct {
    common
    isParallel bool context *testContext // For running tests and subtests.
}

type B struct {
    common
    importPath string // import path of the package containing the benchmark
    context          *benchContext
    N int previousN int           // number of iterations in the previous run
    previousDuration time.Duration // total duration of the previous run
    benchFunc        func(b *B)
    benchTime        time.Duration
    bytes            int64
    missingBytes bool // one of the subbenchmarks does not have bytes set.
    timerOn          bool showAllocResult bool result           BenchmarkResult
    parallelism int // RunParallel creates parallelism*GOMAXPROCS goroutines // The initial states of memStats.Mallocs and memStats.TotalAlloc.
 startAllocs uint64
    startBytes  uint64 // The net total of this test after being run.
 netAllocs uint64
    netBytes  uint64
}

common包含了T和B的所有公共方法,常见的比如Log()日志信息,Error() 错误信息,Fail()致命错误等方法,

TestMain(*testing.M)方法有些特殊,在一个包内只能有一个TestMain方法。这个方法会在测试方法运行前调用,相当于main()方法。我们可以在这个方法内做一些初始化数据操作等。看一下testing.M结构体

// M is a type passed to a TestMain function to run the actual tests.
type M struct {
    deps       testDeps
    tests      []InternalTest
    benchmarks []InternalBenchmark
    examples   []InternalExample

    timer *time.Timer
    afterOnce sync.Once

    numRun int }

专为TestMain准备

先以app1来对基本的test进行解说,app1的项目结构为。
 具体的代码大家看一下就好,都是一些特别简单的方法。

测试指定函数

简单的测试方法

func TestNewUserInfo(t *testing.T) {
    u := NewUserInfo() if len(u.Name) == 0 {
        t.Error("name is empty")
    }
}

得到新创建的user信息,检查name是否为空,如果为空则错误。

-run 后面的参数是正则,所有匹配这正则的方法都会被运行,比如测试所有包含user(不区分大小写)的测试方法:

go test -v -run="(?i)user"

image

-v 是用于输出所有Log的信息

也可以指写具体的方法名,只要包含这些名称的测试方法就会运行,如果要测试多个方法,名称用"|"分开

go test -v -run=TestGetOrderList go test -v -run="TestGetOrderList|TestNewUserInfo"

执行的结果不用我多说,运行是否通过,最后还有运行的时长,方法实在在简单了,执行的太快只精确到2位,所以0.00。

测试指定文件

测试指定的_test.go文件,需要注意的是在同一个包下,需要把测试文件和源文件都写出来:

go test -v user_test.go user.go

测试文件夹内所有的test文件

直接在某个目录运行go test命令就会运行这个文件夹下所有的_test.go文件内的测试方法。

go test -v

如果文件夹里还包含文件夹,可以添加 "./..."来递归测试。

go test -v

BenchMark 测试

benchMark通过一系列的性能指标来对我们的方法进行测试,比如cpu,内存。循环测试次数据等。

基本的命令是

go test -bench="."

-bench 类似于-run 可以跟正则表达式来控制执行的方法

测试方法

func BenchmarkUserInfoList(b *testing.B) {
    b.StopTimer() // do something
 b.StartTimer() for i := 0; i < b.N; i++ { // pretend delay //time.Sleep(time.Millisecond * 500)
        userInfoList := UserInfoList() if len(userInfoList) != 10 {
            b.Error("userInfoList is empty")
        }
    }
}
返回的结果

benchmark方法名加当前测试cpu内核的数量,这个可以通过-cpu 来设置数量,10000是执行的次数,就是代码中的b.N 171679 ns/op 每次操作的耗时。

可以通过flag benchtime来控制执行时长

-benchmem 用于显示内存的分配情况

808 B/op 表示每一调用需要808个字节, 35 allocs/op表示每一次调用有35次内存分配

当然还有很多flag 大家可以通过下面的命令查看官方文档

go help testflag

TestMain

一个包下面只能有一个TestMain方法。这个方法就和main()方法差不太多。他会在其他的Test方法之前执行,我们可以利用他做一些初始化数据操作,执行完后释放资源等操作。

例如我在TestMain里把当前的环境变量设置为dev。 然后加载配置,当然都是模拟的。

func TestMain(m *testing.M) {
    os.Setenv(config.ENV_STR, config.ENV_DEV)
    config.LoadConfig()
    fmt.Printf("Env is %s\n", os.Getenv(config.ENV_STR))
    fmt.Printf("config info %#v\n", config.ConfigInfo) // do something... eg: init data // ...
 os.Exit(m.Run())
}

在执行所有的go test时会先执行TestMain

测试代码覆盖率

测试覆盖率就是运行我们的测试方法所有跑过的代码占全部代码的比例,比如我们跑一下user_test.go的所有测试方法,然后看一下覆盖率:

两个命令:

go test -v -coverprofile cover.out user_test.go user.go go tool cover -html=cover.out -o cover.html

一个是执行测试,另一个是把输出的文件转换成html

 用浏览器打开生成的html,绿颜色表示运行到的代码,红颜色表示没有运行到的代码,我的代码是全部都运行到了。

测试http

先来一个原生的http handler方法

func HandleNewUser(w http.ResponseWriter, r *http.Request) {
    name := r.URL.Query().Get("name")
    fmt.Printf("url parameter user name is %s\n", name)

    say := r.FormValue("say")
    fmt.Printf("req say:' %s '\n", say)
    newUser := NewUserInfo()
    jData, _ := json.Marshal(newUser)
    w.WriteHeader(http.StatusOK)
    w.Header().Set("Content-Type", "application/json")
    w.Write(jData)
}

这个方法没有什么逻辑,在url中获取name参数,然后在post的form中获取say参数,再返回一个user的json。

再看一下测试方法

func TestHandleNewUser(t *testing.T) {
    postBody := url.Values{}
    postBody.Add("say", "hello world")
    req := httptest.NewRequest(http.MethodPost, "http://localhost/createuser?name=linus", strings.NewReader(postBody.Encode()))
    req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
    w := httptest.NewRecorder()
    HandleNewUser(w, req) if w.Code != http.StatusOK {
        t.Error("new user api error")
    } if w.Body.Len() == 0 {
        t.Error(" response is empty")
    }
    user := &model.UserInfo{}
    err := json.Unmarshal(w.Body.Bytes(), user) if err != nil {
        t.Error("response data error")
    }
    t.Logf("create user api response : %#v", user)
}

使用的是httptest包,先是创建了一个请求,url包含了name参数,body里有say参数。然后再通过NewRecorder创建一个responseWriter,把这两个参数传递给我们的handler,再测试有没有返回我们想要的执行结果。

如果你使用的web服务框架是gin。测试gin handler的代码我写在了app2里。有时间你可以看一下,大致的代码:

func TestNewUserInfo(t *testing.T) {
    a := assert.New(t)

    router := gin.New() const path = "/newUserInfo" router.POST(path, NewUserInfo)

    body := url.Values{}
    body.Set("say", "hello world")
    rr, err := testutils.PostFormRequst(path + "?name=lp", router, body)
    a.Nil(err)

    user := &model.UserInfo{}
    err = json.Unmarshal(rr.Body.Bytes(), user)
    a.Nil(err)
    a.NotEqual(user.Name, "")
    a.NotEqual(user.Age, 0)
    t.Logf("%#v\n", user)
}

推荐几个第三方的库

有几个我常用的第三方库给大家推荐一下,相关的测试代码都写到 app2_thirdlib里了

github.com/stretchr/testify github.com/jarcoal/httpmock

testify里有assert相信有其他语言基础的同学一定知道他是做什么的,断言处理比如

a.Nil(err) a.NotEqual(user.Name, "") a.NotEqual(user.Age, 0)

如果判断的结果为false则测试失败。

httpmock这个好玩,假如我们的项目有请求其他项目的api调用,但是我们没有源码,只知道返回结果。但是我们进行test测试时,要请求这个api。httpmock就是做这个用的,他们拦住我们的http请求,然后返回我们预置的response。

func TestUserRoleList(t *testing.T) {
    a := assert.New(t) // mock http
 testutils.HttpMockActivateNonDefault()
    httpmock.RegisterNoResponder(
        httpmock.NewStringResponder(http.StatusOK, fmt.Sprintf(`
        [
          { "id": 1, "name": "a" },
          { "id": 2, "name": "b" },
          { "id": 3, "name": "c" }
        ]
    `)))
    defer httpmock.DeactivateAndReset()

    router := gin.New() const path = "/userRoleList" router.GET(path, UserRoleList)
    rr, err := testutils.GetRequst(path, router)
    a.Nil(err)
    a.Equal(rr.Result().StatusCode, http.StatusOK)

    roles := make([]model.UserRole, 0)
    err = json.Unmarshal(rr.Body.Bytes(), &roles)
    a.Nil(err)
    a.NotEqual(len(roles), 0)
    t.Logf("len of roles: %d\n", len(roles))
}

我的UserRoleList方法调用了其他项目的api。httpmock会把这个http请求拦住,上面我们提前写好的数组。

大家可以下载我的源码,然后 go test ./... 一下试试。

作者:李鹏 出处:http://www.cnblogs.com/li-peng/ 本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

其他参考文档: (6)go test 测试用例那些事 https://cloud.tencent.com/developer/article/1376794 (7)GOLANG语言中文网 - Go语言标准库 - 第九章 测试 https://books.studygolang.com/The-Golang-Standard-Library-by-Example/chapter09/09.0.html (8)golang 单元测试(gotests、mockery自动生成) https://www.jishuwen.com/d/2vk4#tuit (9)golang分层测试之http接口测试入门 https://studygolang.com/articles/16742

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 铂链第1课 如何在WINDOWS操作系统下搭建BOTTOS开发环境

    铂链(Bottos)是国内首个基于区块链技术的数据共享、AI模型共享生态一站式应用平台,是实现将数据和模型通过点对点网络进行登记发行、转让交易的去中心化网络共享...

    辉哥
  • Python入门摘要

    在 Python 中,所有标识符可以包括英文、数字以及下划线(_),但不能以数字开头。

    辉哥
  • 【实践】golang pprof 实战-CPU,heap,alloc,goroutine,mutex,block

    如果要说在 golang 开发过程进行性能调优,pprof 一定是一个大杀器般的工具。但在网上找到的教程都偏向简略,难寻真的能应用于实战的教程。这也无可厚非,毕...

    辉哥
  • go test 测试用例那些事

        go test命令,相信大家都不陌生,常见的情况会使用这个命令做单测试、基准测试和http测试。go test还是有很多flag 可以帮助我们做更多的分...

    lpxxn
  • GlobeImposter2.0再出新变种,疑似利用PsExec内网传播

    近日,深信服安全团队发现了GlobeImposter2.0勒索病毒新变种,该变种疑似利用微软官网工具PsExec进行内网传播并使用了新的勒索界面。截止目前,国内...

    FB客服
  • 如何在Power Query中提取数据——列表篇(1)

    List.First(list as list, optionaldefaultValue as any)as any

    逍遥之
  • 使用Pytest创建一个Python测试自动化项目

    Python是当前最流行的编程语言之一。它为Web后端,数据科学笔记本,sysadmin脚本等提供支持。它的语法简洁,易读且优雅–非常适合初学者和专家。您可以想...

    用户7466307
  • Python多个装饰器的调用顺序实例解析

    一般情况下,在函数中可以使用一个装饰器,但是有时也会有两个或两个以上的装饰器。多个装饰器装饰的顺序是从里到外(就近原则),而调用的顺序是从外到里(就远原则)

    砸漏
  • redis超时原因排查

    1.低效操作产生的延迟。单命令操作一半很快不会造成这样,SORT,LREM, SUNION,keys ,* 等操作都会影响响应时间。 使用进程监控程序(top,...

    BGBiao
  • 【C语言系列】C语言编译流程分析

    前几天看了《程序员的自我修养——链接、装载与库》中的第二章“编译和链接”,主要根据其中的内容简单总结一下C程序编译的过程吧。 我现在一般都是用gcc,所以自然以...

    程序员互动联盟

扫码关注云+社区

领取腾讯云代金券