一。介绍

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是一款非常常用的打桩工具,务必学会使用。