Go 语言猴子补丁(monkey patching)框架。
本项目对 Bouke 的项目做了优化,不同协程可以独立 patch 同一个函数而互不影响。从而可以并发运行单元测试。
工作原理请参考我的系列文章:
Bouke 已经不再维护原项目,所以只能开一个新项目了🤣。
首先,引入 monkey 包
go get github.com/go-kiss/monkey
然后,调用 monkey.Patch
方法 mock 指定函数。
package main
import (
"fmt"
"github.com/go-kiss/monkey"
)
func sum(a, b int) int { return a + b }
func sub1[T int|float64](a, b T) T { return a - b }
func sub2[T int|float64](a, b T) T { return a + b }
type S struct { i int }
func (s *S) Get() int { return i }
func main() {
// mock 普通函数
monkey.Patch(sum, func(a b int) int { return a - b })
fmt.Println(sum(1, 2)) // 输出 -1
// mock 泛型函数
monkey.Patch(sub1[int], sub2[int])
fmt.Println(sub1(1, 2)) // 输出 3
// mock 结构体方法
monkey.Patch((*s).Get, func(s *S) int { return -1 })
var s S
fmt.Println(s.Get()) // 输出 -1
}
go test -gcflags='all=-N -l'
。