go測試框架gomonkey的使用

一。介紹

gomonkey 是 golang 的一款打樁框架,目標是讓用戶在單元測試中低成本的完成打樁,從而將精力聚焦于業務功能的開發。

實現原理《golang實現運行時替換函數體及其原理》

gomonkey支持的特性以及使用方法

• 支持為函數/ 接口打一個樁
• 支持為函數/ 接口打一個特定的樁序列
• 支持為成員方法打一個樁
• 支持為成員方法打一個特定的樁序列
• 支持為函數變量打一個樁
• 支持為函數變量打一個特定的樁序列
• 支持為全局變量打一個樁

注意:Reset方法用于刪除樁, 注意使用defer將刪除樁壓棧。

二。 使用

func Test_UserDao2(t *testing.T) {Convey("Give test start", t, func() {Convey("When ApplyFunc For Func", func() {// 支持為函數打一個樁want := "replace SelectOne Return"selectOnePatch := gomonkey.ApplyFunc(SelectOne, func(id string) string {return want})defer selectOnePatch.Reset()ret := SelectOne("1")So(ret, ShouldEqual, want)})// 支持為函數打一個特定的樁序列Convey("When ApplyFuncSeq For Func", func() {want := []gomonkey.OutputCell{{Values: gomonkey.Params{"1"}},{Values: gomonkey.Params{"2"}},{Values: gomonkey.Params{"3"}},}selectList := gomonkey.ApplyFuncSeq(SelectList, want)defer selectList.Reset()Convey("Then Test ApplyFuncSeq Patch", func() {So(SelectList(), ShouldEqual, "1")So(SelectList(), ShouldEqual, "2")So(SelectList(), ShouldEqual, "3")})})// 支持為成員方法打一個樁Convey("When ApplyMethod For userDao Func", func() {u := &userDao{}updateF := gomonkey.ApplyMethod(reflect.TypeOf(u), "update", func(_ *userDao, id, name, phoneNumber string) int64 {fmt.Println(id, name, phoneNumber)return 1})defer updateF.Reset()Convey("Then Test ApplyMethod userDao Func Patch", func() {So(u.update("", "", ""), ShouldEqual, int64(1))})})// 支持為全局變量打一個樁num := 1Convey("When ApplyGlobalVar", func() {globalVar := gomonkey.ApplyGlobalVar(&num, 10)defer globalVar.Reset()Convey("Then Test ApplyGlobalVar Patch", func() {So(num, ShouldEqual, int64(10))})})// 定義函數變量f1, f2f1 := func() string {return "0"}f2 := func() string {return "0"}// 支持為函數變量打樁Convey("When ApplyFuncVarSeq", func() {want := []gomonkey.OutputCell{{Values: gomonkey.Params{"1"}},{Values: gomonkey.Params{"2"}},{Values: gomonkey.Params{"3"}},}// 支持為函數變量打一個特定的樁序列funcVarSeq := gomonkey.ApplyFuncVarSeq(&f1, want)defer funcVarSeq.Reset()// 支持為函數變量打一個樁objVar := gomonkey.ApplyFuncVar(&f2, func() string {return "ggr"})defer objVar.Reset()Convey("Then Test ApplyFuncVarSeq&ApplyFuncVar Patch", func() {So(f1(), ShouldEqual, "1")So(f1(), ShouldEqual, "2")So(f1(), ShouldEqual, "3")So(f2(), ShouldEqual, "ggr")})})})}

gomonkey是一款非常常用的打樁工具,務必學會使用。