最全为什么在Go中我们需要使用Mock测试（附源码）_# bou，如何成为杰出的程序员

本文链接：https://blog.csdn.net/2401_84920052/article/details/138784426

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

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`进行测试。


测试结果：


![](https://img-blog.csdnimg.cn/img_convert/8a0e823169c8f63f07d401867b06fac1.webp?x-oss-process=image/format,png)


可以看到，我们明明在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
}
复制代码


* 这里的jumpData保存了to的jump信息
* *rawMemoryAccess()* 返回的是一个结构体f，包含了from(上文的target)的元信息
* original拷贝了f（即from）的元信息，也是返回值
* 之后将jumpData的信息复制到from（即将from的地址改为to的）


至于Unpatch就是从patches这个map中取到原来被替换的函数的信息，将其恢复，就不多赘述


## 4、tips


* 在测试时，有时候会有patch了但替换失败的情况出现，这是由于**函数内联**导致的。我们需要在go test中加上`-gcflags=-l`参数来禁止函数内联。
* 在部分面向安全的操作系统上，系统不允许一个内存页同时写和执行，这会使得Monkey不能运行
* Monkey不是绝对线程安全的



![img](https://img-blog.csdnimg.cn/img_convert/aaf2ef9390294797b9be1b59ae659588.png)
![img](https://img-blog.csdnimg.cn/img_convert/405c7ed467d1163c1d0c3b2132b22b60.png)

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

**[需要这份系统化的资料的朋友，可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**


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

s.csdn.net/topics/618658159)**


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