在写测试的时候, 有一种 mock 方法是这样的
var foo = func(){}
这个时候可以替换 foo, 然后跑测试, 最后在还原(隔离)
如果这么写要很长很长....
有了 patch 方法, 只需要两行
func TestH(t *testing.T) {
say := func() {
fmt.Println("say")
}
rec := patchFunction(&say, func() { fmt.Println("replace") })
say()
defer rec() # 还原
}
func patchFunction(item interface{}, new interface{}) func() {
old := reflect.ValueOf(item).Elem().Interface()
rec := func() {
reflect.ValueOf(item).Elem().Set(reflect.ValueOf(old))
}
reflect.ValueOf(item).Elem().Set(reflect.ValueOf(new))
return rec
}
另外求教一下, reflect.valueOf 一个 interface 拿到的直接就是里面的包的元素?? interface 内存里不是两部分吗, 我总觉得应该是 .Elem()拿到包裹的元素, 然后在.Elem().Set()改指针值.