您的位置:首页 > 编程语言 > Go语言

go 为方便单元测试的函数

2016-01-30 23:05 417 查看
写go语言程序,在单元测试时,经常需要比较两个值是否相等,比较一个值是否为nil,或者是否抛出了panic。分别写了如下三个函数来进行判断。具体应用时,可以根据实际情况进行调整。

func verify(t *testing.T, fun string , output, expected interface{}){
if output != expected {
t.Errorf( "%s: output %v != %v", fun, output, expected)
}
}

func verifyNil(t *testing.T, fun string , output interface{}){
v := reflect.ValueOf(output)
if v.IsValid() && !v.IsNil() {
t.Errorf( "%s: output %v is not nil", fun, output)
}
}

func verifyPanic(t *testing.T, fun string , errType interface{}, expectedMsg string){
msgFeild := "Message"

e := recover()
if e == nil {
t.Errorf( "%s: expected error [%T], but a nil!", fun, errType)
return
}

if reflect.TypeOf(errType) != reflect.TypeOf(e) {
t.Errorf( "%s: expected error [%T], but [%T]!", fun, errType, e)
return
}

errMsg := reflect.ValueOf(e).Elem().FieldByName(msgFeild)
if errMsg.String() != expectedMsg {
t.Errorf("%s: expected error message [%s], but [%s]!", fun, expectedMsg, errMsg)
return
}
}


1. 判断两个值是否相等,如下代码:

verify(t, "TestStateMachine 1", sm.GetCurrentState().ID(), "s2")
sm.SendEvent(e2)
verify(t, "TestStateMachine 2", sm.GetCurrentState().ID(), "s3")
sm.SendEvent(e3)
verify(t, "TestStateMachine 3", sm.GetCurrentState().ID(), "s1")


为了区分是在哪儿没有测试通过,在传入参数名时,加了序号。

2. 判断一个值是否为nil,如下代码:

verifyNil(t, "TestStart 1", sm.GetCurrentState())
verifyNil(t, "TestStart 2", sm.GetEvent())


3. 判断是否抛出了panic,如下代码:

expected := "Has no action executor for [ao2]."
defer verifyPanic(t, "TestHasNoActionExecutor", (*IllegalActionError)(nil), expected)


抛出panic处的代码如下:

panic(&IllegalActionError{"Has no action executor for [" + execName + "]."})
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: