golang使用闭包降低测试代码行数

转载请注明出处,原文链接http://tailnode.tk/2017/01/go...

有如下函数,简单来说就是有错误则直接返回,没错误则执行f函数

func (t *transaction) Do(f func()) *transaction {
    if t.fail || t.rollback || t.finish {
        return t
    }
    f()
    return t
}

函数很简单,但如何测试呢,简单但丑陋的方法

func Test_func(t *testing.T) {
    isCalled := false
    f := func() {
        isCalled = true
    }
    trans := New()
    // do something
    trans.Do(f)
    // check
    if isCalled {
        // do something
    }
}

f修改外部变量,然后判断变量是否变化就可以知道f是否被执行。但我们一般需要测试多种情况,比如对于Do函数,我们需要将 t.fail t.rollback t.finish设置不同值进行测试,将上面测试代码扩充(如果需要测试这三个变量组合的情况,代码就更长了):

func Test_func(t *testing.T) {
    isCalled1 := false
    isCalled2 := false
    isCalled3 := false
    f1 := func() {
        isCalled1 = true
    }
    f2 := func() {
        isCalled2 = true
    }
    f3 := func() {
        isCalled3 = true
    }
    trans := New()
    // do something
    trans.Do(f1)
    trans.Do(f2)
    trans.Do(f3)
    // check
    if isCalled1 {
        // do something
    }
    if isCalled2 {
        // do something
    }
    if isCalled3 {
        // do something
    }
}

在上面代码f1 f2 f3函数的逻辑都一样,这时可以通过使用闭包来消除冗余代码

func Test_func(t *testing.T) {
    genTestFunc := func() (func(),func() bool) {
        isCalled := false
        return func() {
                isCalled = true
            },func() bool {
                return isCalled
            }
    }
    f1,f1Called := genTestFunc()
    f2,f2Called := genTestFunc()
    f3,f3Called := genTestFunc()
    trans := New()
    // do something
    trans.Do(f1)
    trans.Do(f2)
    trans.Do(f3)
    // check
    if f1Called() {
        // do something
    }
    if f2Called() {
        // do something
    }
    if f3Called() {
        // do something
    }
}

解释一下,genTestFunc 返回值是两个函数,第一个函数可传入Do中,第二个函数用来判断是否被Do调用
粗略看改动前后代码行数基本相同,但如果f变复杂或者需要更多的测试case时,改动后的代码更加简洁,易于维护。

相关文章

程序目录结构 简单实现,用户登录后返回一个jwt的token,下次请求带上token请求用户信息接口并返回信息...
本篇博客的主要内容是用go写一个简单的Proof-of-Work共识机制,不涉及到网络通信环节,只是一个本地的简...
简介 默克尔树(MerkleTree)是一种典型的二叉树结构,其主要特点为: 最下面的叶节点包含存储数据或其...
接下来学习并发编程, 并发编程是go语言最有特色的地方, go对并发编程是原生支持. goroutine是go中最近本...
先普及一下, 什么是广度优先搜索 广度优先搜索类似于树的层次遍历。从图中的某一顶点出发,遍历每一个顶...
第一天: 接口的定义和实现 第二天: 一. go语言是面向接口编程. 在学习继承的时候说过, go语言只有封装,...