GoMock進行單元測試
阿新 • • 發佈:2020-12-14
本文引用自
安裝
$ 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 程式碼並沒有什麼區別。因此選擇合適的就可以了
寫測試用例
步驟
- 想清楚整體邏輯
- 定義想要(模擬)依賴項的 interface(介面)
- 使用
mockgen
命令對所需 mock 的 interface 生成 mock 檔案 - 編寫單元測試的邏輯,在測試中使用 mock
- 進行單元測試的驗證
目錄
├── mock ├── person │ └── male.go └── user ├── user.go └── user_test.go
編寫
interface 方法
開啟 person/male.go 檔案,寫入以下內容:
package person
type Male interface {
Get(id int64) error
}
呼叫方法
開啟 user/user.go 檔案,寫入以下內容:
package user import "../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 (
reflect "reflect"
gomock "github.com/golang/mock/gomock"
)
// 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 {
m.ctrl.T.Helper()
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 {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockMale)(nil).Get), id)
}
測試用例
開啟 user/user_test.go 檔案,寫入以下內容:
package user
import (
"testing"
"../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()
的返回值,以此得到不一樣的測試結果哦