写了一个 mock 任何 变量 的函数

[petelin]

在写测试的时候, 有一种 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()改指针值.