写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 + "]."})
有疑问加站长微信联系(非本文作者)