2024年为什么在Go中我们需要使用Mock测试（附源码）_# bou，这份354页笔记的软件测试进阶知识+大厂高频面试题

本文链接：https://blog.csdn.net/2401_84563987/article/details/138554001

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

func TestReadFirstLineMock(t *testing.T) {
   monkey.Patch(ReadFirstLine, func() string {
      return "hi"
   })
   defer monkey.Unpatch(ReadFirstLine)
   line := ReadFirstLine()
   assert.Equal(t, line, "hi")
}
复制代码

其中，Patch方法是我们关注的对象，其作用是更改 第一个参数函数(例子中的ReadFirstLine)的地址，将其改为第二个参数函数的地址，这样无论在什么环境下，我们调用ReadFirstLine()返回的都是"hi"，而UnPatch就是将ReadFirstLine恢复为原来的地址。

运行go test -v -run TestReadFirstLineMock$ read_file.go read_file_test.go进行测试。

测试结果：

可以看到，我们明明在test.txt中第一行为 hello ，但是测试中判断第一行为 hi 却通过了

3、Patch源码

我们点进Patch()源码：

func Patch(target, replacement interface{}) *PatchGuard {
	t := reflect.ValueOf(target)
	r := reflect.ValueOf(replacement)
	patchValue(t, r)

	return &PatchGuard{t, r}
}
复制代码

首先通过反射获取了target和replacement的函数地址，之后调用patchValue()将target地址改为replacement。我们点进patchValue(t,r)

func patchValue(target, replacement reflect.Value) {
	lock.Lock()
	defer lock.Unlock()

	if target.Kind() != reflect.Func {
		panic("target has to be a Func")
	}

	if replacement.Kind() != reflect.Func {
		panic("replacement has to be a Func")
	}

	if target.Type() != replacement.Type() {
		panic(fmt.Sprintf("target and replacement have to have the same type %s != %s", target.Type(), replacement.Type()))
	}

	if patch, ok := patches[target.Pointer()]; ok {
		unpatch(target.Pointer(), patch)
	}

	bytes := replaceFunction(target.Pointer(), (uintptr)(getPtr(replacement)))
	patches[target.Pointer()] = patch{bytes, &replacement}
}

复制代码

在这一步，首先做了加锁操作，保证一定并发的安全性。
接着，对各个参数进行校验。
再下来，查看这个函数是不是先前有patch过（这里维护了一个map即patches，存储已pacth的信息）。如果有，那么先unpatch，保证只patch一次。
之后的 replaceFunction(target.Pointer(), (uintptr)(getPtr(replacement))) 便是真正替代函数地址的方法，其返回的bytes包含了target的元信息。
最后将我们patch的原函数放入patches中，保存它的信息，以便之后unpatch时修改回来

我们再点进 replaceFunction（）

func replaceFunction(from, to uintptr) (original []byte) {
   jumpData := jmpToFunctionValue(to)
   f := rawMemoryAccess(from, len(jumpData))
   original = make([]byte, len(f))
   copy(original, f)

   copyToLocation(from, jumpData)
   return
}
复制代码