go 为方便单元测试的函数

写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 + "]."})

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传