使用 Gomock 進行單元測試

在實際項目中，需要進行單元測試的時候。卻往往發現有一大堆依賴項。這時候就是 Gomock 大顯身手的時候了
Gomock 是 Go 語言的一個 mock 框架，官方的那種。

安裝

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

第一步：我們將安裝 gomock 第三方庫和 mock 代碼的生成工具 mockgen。而後者可以大大的節省我們的工作量。只需要瞭解其使用方式就可以
第二步：輸入 mockgen 驗證代碼生成工具是否安裝正確。若無法正常響應，請檢查 bin 目錄下是否包含該二進制文件

用法

在 mockgen 命令中，支持兩種生成模式：

source：從源文件生成 mock 接口（通過 -source 啓用）

mockgen -source=foo.go [other options]

reflect：通過使用反射程序來生成 mock 接口。它通過傳遞兩個非標誌參數來啓用：導入路徑和逗號分隔的接口列表

mockgen database/sql/driver Conn,Driver

從本質上來講，兩種方式生成的 mock 代碼並沒有什麼區別。因此選擇合適的就可以了

寫測試用例

在本文將模擬一個簡單 Demo 來編寫測試用例，熟悉整體的測試流程

步驟

想清楚整體邏輯
定義想要（模擬）依賴項的 interface（接口）
使用 mockgen 命令對所需 mock 的 interface 生成 mock 文件
編寫單元測試的邏輯，在測試中使用 mock
進行單元測試的驗證

編寫

interface 方法
打開 person/male.go 文件，寫入以下內容：

package person

type Male interface {
	Get(id int64) error
}

調用方法
打開 user/user.go 文件，寫入以下內容：

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 文件

// 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 文件，寫入以下內容：

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

gomock.NewController：返回 gomock.Controller，它代表 mock 生態系統中的頂級控件。定義了 mock 對象的範圍、生命週期和期待值。另外它在多個 goroutine 中是安全的
mock.NewMockMale：創建一個新的 mock 實例
gomock.InOrder：聲明給定的調用應按順序進行（是對 gomock.After 的二次封裝）
mockMale.EXPECT().Get(id).Return(nil)：這裏有三個步驟，EXPECT() 返回一個允許調用者設置期望和返回值的對象。Get(id) 是設置入參並調用 mock 實例中的方法。Return(nil) 是設置先前調用的方法出參。簡單來說，就是設置入參並調用，最後設置返回值
NewUser(mockMale)：創建 User 實例，值得注意的是，在這裏注入了 mock 對象，因此實際在隨後的 user.GetUserInfo(id) 調用（入參：id 爲 1）中。它調用的是我們事先模擬好的 mock 方法
ctl.Finish()：進行 mock 用例的期望值斷言，一般會使用 defer 延遲執行，以防止我們忘記這一操作

測試

回到 mockd/ 的根目錄下，執行以下命令

$ go test ./user
ok  	github.com/EDDYCJY/mockd/user

看到這樣的結果，就大功告成啦！你可以自己調整一下 Return() 的返回值，以此得到不一樣的測試結果哦

查看測試情況

測試覆蓋率

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

可通過設置 -cover 標誌符來開啓覆蓋率的統計，展示內容爲 coverage: 100.0%。

可視化界面

生成測試覆蓋率的 profile 文件

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

利用 profile 文件生成可視化界面

$ go tool cover -html=cover.out

查看可視化界面，分析覆蓋情況

一、常用 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 來完成批量處理的功能

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

修改 interface 方法
打開 person/male.go 文件，修改爲以下內容：

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 這條語句，可分爲以下部分：

聲明 //go:generate （注意不要留空格）
使用 mockgen 命令
定義 -destination
定義 -package
定義 source，此處爲 person 的包路徑
定義 interfaces，此處爲 Male

重新生成 mock 文件
回到 mockd/ 的根目錄下，執行以下命令

$ go generate ./...

再檢查 mock/ 發現也已經正確生成了，在多個文件時是不是很方便呢

總結

在單元測試這一環，gomock 給我們提供了極大的便利。能夠 mock 掉許許多多的依賴項
其中還有很多的使用方式和功能。你可以 mark 住後詳細閱讀下官方文檔，記憶會更深刻

本文由 Readfog 進行 AMP 轉碼，版權歸原作者所有。
來源：https://blog.csdn.net/u013474436/article/details/105477636

安裝

用法