Go语言中怎么使用stub和mock实现单元测试

1. mock例子：

mock 的做法是创建一个struct，然后实现所需要接口中的函数，在测试的时候，将调用的接口替换成测试mock出来的struct就可以了。

mock采用的替换，是替换整个接口的方式(备注：接口可以是入参，也可以是struct中的变量，不过都需要在测试之前，替换完成)。

例子如下：

GetType函数原本是根据不同的入参yu，多不同的操作，加、减和赋值为零。

在mock之后，指定GetType类型只做一种操作 “加法”。

Output:

通过输出可以看出，mock之前的是三种操作，mock替换之后变成了一种操作加法。

2. stub例子：

使用stub做单元测试的话，代码是侵入式的，需要将对应的函数做成回调函数的方式，以方便后续的测试代码可以将这个函数替换成，测试希望指定的功能实现。

例子如下：（备注：这个例子比较简单，只是为了演示stub的使用。）

GetType函数原本是根据不同的入参，多不同的操作，加、减和赋值为零。

在Stub之后，指定GetType类型只做一种操作加法。

Output:

打桩之前：功能函数会根据入参的数值，来做不同的操作，输出不同的结果，分别是：取0；做加法：2+5 = 7；做减法：5-1 = 4。
打桩之后：桩函数StubFunc只返回一种操作“加法”，所以测试用例不管入参的类型是什么，都是按照加法来操作，分别是：1+3 = 4；2+5 = 7; 5+1 = 6;

三、mock与stub的比较

1）mock和stub都是采用替换的方式来实现，被测试的函数中的依赖关系，不过mock采用的是接口替换的方式，stub采用的是函数替代的方式。
2）mock的实现对功能代码没有侵入性，stub的侵入性比较强，在实现功能函数的时候，就需要为了测试设置一些回调函数，也就是这里所谓的桩。
3）对于控制被替代的方法来讲，mock如果想支持不同的输出，就需要提前实现不同的分支代码，甚至需要定义不同的mock结构体来实现，这样的mock代码会变成一个支持所有逻辑分支的一个最大集合，mock代码复杂性会变高；stub却能很好的控制桩函数的不同分支，因为stub替换的是函数，那么只要需要再用到这种输出的时候，定义一个函数即可，而这个函数甚至都可以是匿名函数。

基于上面的分析，我们可以设计一种方式，让stub和mock结合起来，例子如下：

被测试代码与mock例子中的一致，mock的代码中新增了回调函数和是否使用回调函数2个变量，在测试用例中，通过实现不同的stub函数来控制桩函数的输出结果。