脚本之家，脚本语言编程技术及教程分享平台！
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - 使用Gomock进行单元测试的方法示例

使用Gomock进行单元测试的方法示例

2020-05-21 10:34煎鱼 Golang

这篇文章主要介绍了使用Gomock进行单元测试的方法示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

在开发过程中往往需要配合单元测试，但是很多时候，单元测试需要依赖一些比较复杂的准备工作，比如需要依赖数据库环境，需要依赖网络环境，单元测试就变成了一件非常麻烦的事情。举例来说，比如我们需要请求一个网页，并将请求回来的数据进行处理。在刚开始的时候，我通常都会先启动一个简单的http服务，然后再运行我的单元测试。可是这个单元测试测起来似乎非常笨重。甚至在持续集成过程中，我还为了能够自动化测试，特意写了一个脚本自动启动相应的服务。事情似乎需要进行一些改变。

mock对象就是为了解决上面的问题而诞生的，mock(模拟)对象能够模拟实际依赖对象的功能，同时又不需要非常复杂的准备工作，你需要做的，仅仅就是定义对象接口，然后实现它，再交给测试对象去使用。

go-mock是专门为go语言开发的mock库，该库使用方式简单，支持自动生成代码，可以说是不可多得的好工具。

前言

在实际项目中，需要进行单元测试的时候。却往往发现有一大堆依赖项。这时候就是 Gomock 大显身手的时候了

Gomock 是 Go 语言的一个 mock 框架，官方的那种 

安装

?
1
2
$ go get -u github.com/golang/mock/gomock
$ go install github.com/golang/mock/mockgen

1、 第一步：我们将安装 gomock 第三方库和 mock 代码的生成工具 mockgen。而后者可以大大的节省我们的工作量。只需要了解其使用方式就可以

2、 第二步：输入 mockgen 验证代码生成工具是否安装正确。若无法正常响应，请检查 bin 目录下是否包含该二进制文件

用法

在 mockgen 命令中，支持两种生成模式：

1、 source：从源文件生成 mock 接口（通过 -source 启用）

?
1
mockgen -source=foo.go [other options]

2、 reflect：通过使用反射程序来生成 mock 接口。它通过传递两个非标志参数来启用：导入路径和逗号分隔的接口列表

?
1
mockgen database/sql/driver Conn,Driver

从本质上来讲，两种方式生成的 mock 代码并没有什么区别。因此选择合适的就可以了

写测试用例

在本文将模拟一个简单 Demo 来编写测试用例，熟悉整体的测试流程

步骤

  1. 想清楚整体逻辑
  2. 定义想要（模拟）依赖项的 interface（接口）
  3. 使用 mockgen 命令对所需 mock 的 interface 生成 mock 文件
  4. 编写单元测试的逻辑，在测试中使用 mock
  5. 进行单元测试的验证

目录

?
1
2
3
4
5
6
├── mock
├── person
│  └── male.go
└── user
  ├── user.go
  └── user_test.go

编写

interface 方法

打开 person/male.go 文件，写入以下内容：

?
1
2
3
4
5
package person
 
type Male interface {
  Get(id int64) error
}

调用方法

打开 user/user.go 文件，写入以下内容：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
package user
 
import "github.com/EDDYCJY/mockd/person"
 
type User struct {
  Person person.Male
}
 
func NewUser(p person.Male) *User {
  return &User{Person: p}
}
 
func (u *User) GetUserInfo(id int64) error {
  return u.Person.Get(id)
}

生成 mock 文件

回到 mockd/ 的根目录下，执行以下命令

 

复制代码 代码如下:
$ mockgen -source=./person/male.go -destination=./mock/male_mock.go -package=mock

 

在执行完毕后，可以发现 mock/ 目录下多出了 male_mock.go 文件，这就是 mock 文件。那么命令中的指令又分别有什么用呢？如下：

  • -source：设置需要模拟（mock）的接口文件
  • -destination：设置 mock 文件输出的地方，若不设置则打印到标准输出中
  • -package：设置 mock 文件的包名，若不设置则为 mock_ 前缀加上文件名（如本文的包名会为 mock_person）

想了解更多的指令符，可参见 官方文档

输出的 mock 文件

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
// Code generated by MockGen. DO NOT EDIT.
// Source: ./person/male.go
 
// Package mock is a generated GoMock package.
package mock
 
import (
  gomock "github.com/golang/mock/gomock"
  reflect "reflect"
)
 
// MockMale is a mock of Male interface
type MockMale struct {
  ctrl   *gomock.Controller
  recorder *MockMaleMockRecorder
}
 
// MockMaleMockRecorder is the mock recorder for MockMale
type MockMaleMockRecorder struct {
  mock *MockMale
}
 
// NewMockMale creates a new mock instance
func NewMockMale(ctrl *gomock.Controller) *MockMale {
  mock := &MockMale{ctrl: ctrl}
  mock.recorder = &MockMaleMockRecorder{mock}
  return mock
}
 
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockMale) EXPECT() *MockMaleMockRecorder {
  return m.recorder
}
 
// Get mocks base method
func (m *MockMale) Get(id int64) error {
  ret := m.ctrl.Call(m, "Get", id)
  ret0, _ := ret[0].(error)
  return ret0
}
 
// Get indicates an expected call of Get
func (mr *MockMaleMockRecorder) Get(id interface{}) *gomock.Call {
  return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockMale)(nil).Get), id)
}

测试用例

打开 user/user_test.go 文件，写入以下内容：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
package user
 
import (
  "testing"
 
  "github.com/EDDYCJY/mockd/mock"
 
  "github.com/golang/mock/gomock"
)
 
func TestUser_GetUserInfo(t *testing.T) {
  ctl := gomock.NewController(t)
  defer ctl.Finish()
 
  var id int64 = 1
  mockMale := mock.NewMockMale(ctl)
  gomock.InOrder(
    mockMale.EXPECT().Get(id).Return(nil),
  )
 
  user := NewUser(mockMale)
  err := user.GetUserInfo(id)
  if err != nil {
    t.Errorf("user.GetUserInfo err: %v", err)
  }
}
  1. gomock.NewController：返回 gomock.Controller，它代表 mock 生态系统中的顶级控件。定义了 mock 对象的范围、生命周期和期待值。另外它在多个 goroutine 中是安全的
  2. mock.NewMockMale：创建一个新的 mock 实例
  3. gomock.InOrder：声明给定的调用应按顺序进行（是对 gomock.After 的二次封装）
  4. mockMale.EXPECT().Get(id).Return(nil)：这里有三个步骤，EXPECT()返回一个允许调用者设置期望和返回值的对象。Get(id) 是设置入参并调用 mock 实例中的方法。Return(nil) 是设置先前调用的方法出参。简单来说，就是设置入参并调用，最后设置返回值
  5. NewUser(mockMale)：创建 User 实例，值得注意的是，在这里注入了 mock 对象，因此实际在随后的 user.GetUserInfo(id) 调用（入参：id 为 1）中。它调用的是我们事先模拟好的 mock 方法
  6. ctl.Finish()：进行 mock 用例的期望值断言，一般会使用 defer 延迟执行，以防止我们忘记这一操作

测试

回到 mockd/ 的根目录下，执行以下命令

?
1
2
$ go test ./user
ok   github.com/EDDYCJY/mockd/user

看到这样的结果，就大功告成啦！你可以自己调整一下 Return() 的返回值，以此得到不一样的测试结果哦 

查看测试情况

测试覆盖率

?
1
2
$ go test -cover ./user
ok   github.com/EDDYCJY/mockd/user  (cached)  coverage: 100.0% of statements

可通过设置 -cover 标志符来开启覆盖率的统计，展示内容为 coverage: 100.0%。

可视化界面

1、 生成测试覆盖率的 profile 文件

?
1
$ go test ./... -coverprofile=cover.out

2、 利用 profile 文件生成可视化界面

?
1
$ go tool cover -html=cover.out

3、 查看可视化界面，分析覆盖情况

使用Gomock进行单元测试的方法示例

更多

一、常用 mock 方法

调用方法

  • Call.Do()：声明在匹配时要运行的操作
  • Call.DoAndReturn()：声明在匹配调用时要运行的操作，并且模拟返回该函数的返回值
  • Call.MaxTimes()：设置最大的调用次数为 n 次
  • Call.MinTimes()：设置最小的调用次数为 n 次
  • Call.AnyTimes()：允许调用次数为 0 次或更多次
  • Call.Times()：设置调用次数为 n 次

参数匹配

  • gomock.Any()：匹配任意值
  • gomock.Eq()：通过反射匹配到指定的类型值，而不需要手动设置
  • gomock.Nil()：返回 nil

建议更多的方法可参见 官方文档

二、生成多个 mock 文件

你可能会想一条条命令生成 mock 文件，岂不得崩溃？

当然，官方提供了更方便的方式，我们可以利用 go:generate 来完成批量处理的功能

?
1
go generate [-run regexp] [-n] [-v] [-x] [build flags] [file.go... | packages]

修改 interface 方法

打开 person/male.go 文件，修改为以下内容：

?
1
2
3
4
5
6
7
package person
 
//go:generate mockgen -destination=../mock/male_mock.go -package=mock github.com/EDDYCJY/mockd/person Male
 
type Male interface {
  Get(id int64) error
}

我们关注到 go:generate 这条语句，可分为以下部分：

  1. 声明 //go:generate （注意不要留空格）
  2. 使用 mockgen 命令
  3. 定义 -destination
  4. 定义 -package
  5. 定义 source，此处为 person 的包路径
  6. 定义 interfaces，此处为 Male

重新生成 mock 文件

回到 mockd/ 的根目录下，执行以下命令

?
1
$ go generate ./...

再检查 mock/ 发现也已经正确生成了，在多个文件时是不是很方便呢

  • 单元测试
  • Gomock

    • 延伸 · 阅读

    精彩推荐
    • GolangGolang中Bit数组的实现方式

      Golang中Bit数组的实现方式

      这篇文章主要介绍了Golang中Bit数组的实现方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      天易独尊11682021-06-09
    • Golanggolang json.Marshal 特殊html字符被转义的解决方法

      golang json.Marshal 特殊html字符被转义的解决方法

      今天小编就为大家分享一篇golang json.Marshal 特殊html字符被转义的解决方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧 ...

      李浩的life12792020-05-27
    • Golanggo日志系统logrus显示文件和行号的操作

      go日志系统logrus显示文件和行号的操作

      这篇文章主要介绍了go日志系统logrus显示文件和行号的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      SmallQinYan12302021-02-02
    • Golanggolang的httpserver优雅重启方法详解

      golang的httpserver优雅重启方法详解

      这篇文章主要给大家介绍了关于golang的httpserver优雅重启的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，...

      helight2992020-05-14
    • GolangGolang通脉之数据类型详情

      Golang通脉之数据类型详情

      这篇文章主要介绍了Golang通脉之数据类型，在编程语言中标识符就是定义的具有某种意义的词，比如变量名、常量名、函数名等等，Go语言中标识符允许由...

      羌4272021-11-24
    • Golanggo语言制作端口扫描器

      go语言制作端口扫描器

      本文给大家分享的是使用go语言编写的TCP端口扫描器，可以选择IP范围，扫描的端口，以及多线程，有需要的小伙伴可以参考下。 ...

      脚本之家3642020-04-25
    • Golanggolang 通过ssh代理连接mysql的操作

      golang 通过ssh代理连接mysql的操作

      这篇文章主要介绍了golang 通过ssh代理连接mysql的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      a165861639710342021-03-08
    • Golanggolang如何使用struct的tag属性的详细介绍

      golang如何使用struct的tag属性的详细介绍

      这篇文章主要介绍了golang如何使用struct的tag属性的详细介绍，从例子说起，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看...

      Go语言中文网11352020-05-21
    最近更新
    编辑推荐
    阅读排行
    热门标签
    gosublime 　 指针类型 　 new() 　 make() 　 并发模型 　 切片 　 静态WEB服务器 　 端口扫描器 　 Go语言入门教程 　 基础语 　 Slices 　 Maps 　 Golang编程 　 panic 　 接受者类型 　 嵌入类型 　 Slice 　 Golang教程 　 Golang项目 　 内存模型 　 开发环境搭建 　 大小端判断 　 代码组织结构 　 复合类型 　 GOPATH 　 工作目录 　 实例 　 struct类型 　 并发技术 　 buffered 　
    © 2019-2023 脚本之家 | 服务器之家(www.zzvips.com)旗下站点 版权所有关于我们联系我们版权申明网站地图