为什么在Go中我们需要使用Mock测试(附源码)_# bou,2024年最新2024最新Golang面试题及答案

52 阅读2分钟

当我们进行测试时,有时测试代码依赖于环境,耦合度较高。这样的测试会导致在不同环境下产生不一致的结果。

举个栗子,我们有一个函数读取文件的第一行:

func ReadFirstLine() string {
   open, err := os.Open("test.txt")
   defer open.Close()
   if err != nil {
      log.Print(err)
      return ""
   }
   reader := bufio.NewReader(open)
   line, _, err := reader.ReadLine()
   if err != nil {
      log.Print(err)
      return ""
   }
   return string(line)
}
复制代码

当我们使用常规测试时,需要在本环境下拥有test.txt文件

测试代码:

func TestReadFirstLine(t *testing.T) {
   line := ReadFirstLine()
   assert.Equal(t, line, "hello")
}
复制代码

执行以下代码 go test -v -run TestReadFirstLine$ read_file.go read_file_test.go

测试结果:

2、使用Monkey进行mock测试

我们可以使用 monkeypatching进行mock测试,使用go get -v bou.ke/monkey获取

同样的,针对上面的案例,使用如下测试代码进行mock测试:

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)
	}


![img](https://p6-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/30b3b3197c3a47269537e0f09ac57870~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1770647921&x-signature=UFsc3pbuPbineFZDj1M%2FVul51Rk%3D)
![img](https://p6-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/0dd203410f2d42bfa029467b0ccde2be~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1770647921&x-signature=zXw1MRl3PkGEfNGwIXDHiJgtAPM%3D)

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

**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://gitee.com/vip204888)**


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