1.说明

gomonkey 是 golang 的一款打桩框架，目标是让用户在单元测试中低成本的完成打桩，从而将精力聚焦于业务功能的开发。

参考文档：https://blog.csdn.net/weixin_36163504/article/details/112583704

2.github仓库

https://github.com/agiledragon/gomonkey

3.用法参考

参考：https://blog.csdn.net/u013276277/article/details/104993370

4.使用场景

4.1为函数打桩

注意：下面的文件进行测试时，需要加上参数-gcflags=-l。原因参考：https://www.jianshu.com/p/19089baf79ca

文件：isConfigFileContain.go

package funmonkey
 
import (
    "fmt"
    "io/ioutil"
    "os"
    "strings"
)
 
// 判断/opt/container/config.properties文件中是否包含str字符串
func isConfigFileContain(str string) (bool, error) {
    defer func() {
        if err := recover(); err != nil {
            fmt.Printf("错误:%v\n", err)
        }
    }()
 
    file, err := os.Open("/opt/container/config.properties")
    if err != nil {
        fmt.Printf("open错误:%v\n", err)
        return false, err
    }
    defer file.Close()
 
    content, err := ioutil.ReadAll(file)
    if err != nil {
        fmt.Printf("ReadAll错误:%v\n", err)
        return false, err
    }
    return strings.Contains(string(content), str), err
}

文件：isConfigFileContain_test.go

package funmonkey
 
import (
    "fmt"
    "io"
    "io/ioutil"
    "os"
    "testing"
    "github.com/agiledragon/gomonkey"
    // "github.com/stretchr/testify/assert"
    . "github.com/smartystreets/goconvey/convey"
)
 
// 用普通的方式测试会找不到文件
// func TestIsConfigFileContain(t *testing.T) {
//  ast := assert.New(t)
//  ast.True(isConfigFileContain("test"))
// }
 
// 以下用monkey patch进行测试
func TestIsConfigFileContain(t *testing.T) {
    Convey("TestIsConfigFileContain", t, func() {
 
        //下面正常逻辑
        Convey("return nil", func() {
            //open返回nil
            patches := gomonkey.ApplyFunc(os.Open, func(_ string) (*os.File, error) {
                return &os.File{}, nil
            })
            defer patches.Reset()
            //readAll返回nil
            patches.ApplyFunc(ioutil.ReadAll, func(_ io.Reader) ([]byte, error) {
                return nil, nil
            })
            flag, err := isConfigFileContain("str")
            So(err, ShouldBeNil)
            So(flag, ShouldEqual, false)
        })
        //返回错误
        Convey("return error when os.open error", func() {
            //open返回error
            patches := gomonkey.ApplyFunc(os.Open, func(str string) (*os.File, error) {
                return nil, fmt.Errorf("os.open panic")
            })
            defer patches.Reset()
            flag, err := isConfigFileContain("str")
            So(err, ShouldNotBeNil)
            So(flag, ShouldEqual, false)
        })
        //返回错误
        Convey("should be return error when ioutil.ReadAll error", func() {
            patches := gomonkey.ApplyFunc(os.Open, func(str string) (*os.File, error) {
                return nil, nil
            })
            defer patches.Reset()
            //readAll返回error
            patches.ApplyFunc(ioutil.ReadAll, func(io.Reader) ([]byte, error) {
                return nil, fmt.Errorf("ioutil.ReadAll panic")
            })
            flag, err := isConfigFileContain("str")
            So(err, ShouldNotBeNil)
            So(flag, ShouldEqual, false)
        })
    })
}

4.2为方法打桩

参考：https://blog.csdn.net/u013276277/article/details/104993370

4.3为变量打桩

文件：one.go

package main
 
import (
    "fmt"
 
    "xorm.io/xorm"
)
 
type Mode struct {
    Id   int64
    Type string
}
type Orm struct {
    *xorm.Engine
}
 
var GDB *Orm
 
func Init() {
    conn := "root" + ":" + "eisoo.com123" + "@tcp(" + "10.2.33.180" + ":" +
        "3306" + ")/" + "metadata" + "?charset=" + "utf8"
    //新建一个连接
    engine := new(xorm.Engine)
    engine, err := xorm.NewEngine("mysql", conn)
    if err != nil {
        fmt.Println(err)
        return
    }
    //开辟空间
    GDB = new(Orm)
    GDB.Engine = engine
}
 
func main() {
    var mode []Mode
    //使用全局变量进行查询
    GDB.Engine.Where("select").Find(&mode)
    fmt.Println(mode)
}

文件：one_test.go

package main
 
import (
    "fmt"
 
    "xorm.io/xorm"
)
 
type Mode struct {
    Id   int64
    Type string
}
type Orm struct {
    *xorm.Engine
}
 
var GDB *Orm
 
func Init() {
    conn := "root" + ":" + "123456" + "@tcp(" + "10.2.33.184" + ":" +
        "3306" + ")/" + "metadata" + "?charset=" + "utf8"
    //新建一个连接
    engine := new(xorm.Engine)
    engine, err := xorm.NewEngine("mysql", conn)
    if err != nil {
        fmt.Println(err)
        return
    }
    //开辟空间
    GDB = new(Orm)
    GDB.Engine = engine
}
 
func main() {
    var mode []Mode
    //使用全局变量进行查询
    GDB.Engine.Where("select").Find(&mode)
    fmt.Println(mode)
}