go 为方便单元测试的函数

来源：互联网发布：2016网络歌曲打包下载编辑：程序博客网时间：2024/05/18 03:46

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

0 0