用gomock进行mock测试

在开发过程当中每每须要配合单元测试，可是不少时候，单元测试须要依赖一些比较复杂的准备工做，好比须要依赖数据库环境，须要依赖网络环境，单元测试就变成了一件很是麻烦的事情。举例来讲，好比咱们须要请求一个网页，并将请求回来的数据进行处理。在刚开始的时候，我一般都会先启动一个简单的http服务，而后再运行个人单元测试。但是这个单元测试测起来彷佛很是笨重。甚至在持续集成过程当中，我还为了可以自动化测试，特地写了一个脚本自动启动相应的服务。事情彷佛须要进行一些改变。

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

go-mock是专门为go语言开发的mock库，该库使用方式简单，支持自动生成代码，能够说是不可多得的好工具。下面我就简单地展现一下go-mock是如何工做的:

首先你须要作的是将依赖下载到本地：

go get github.com/golang/mock/gomock
go get github.com/golang/mock/mockgen

第一个是代码依赖，第二个是命令行工具（特别好用）。

下面用一个很是简单的例子来讲明gomock是如何工做的：

我在$GOPATH/src目录下新建一个项目：hellomock，在$GOPATH/src/hellomock目录下新建hellomock.go，并定义一个接口Talker:

package hellomock

type Talker interface {
    SayHello(word string)(response string)
}

而后咱们须要一个实现了Talker功能的结构体，假设咱们有这样的场景，咱们如今有一个迎宾的岗位，须要一个可以迎宾的迎宾员，固然这个迎宾员能够是一我的，或者是一只鹦鹉。那么咱们须要作的是，定义一个Persion结构体（或者是鹦鹉或者是别的什么东西），并实现Talker接口：

person.go

package hellomock

import "fmt"

type Person struct{
  name string
}

func NewPerson(name string)*Person{
  return &Person{
      name:name,
  }
}


func (p *Person)SayHello(name string)(word string) {
  return fmt.Sprintf("Hello %s, welcome come to our company, my name is %s",name,p.name)    
}

如今咱们的Person已经实现了Talker接口，如今咱们让他发挥做用吧！

如今假设，咱们有一个公司，公司有一个迎宾员，也就是咱们的前台妹子，这个妹子实现了Talker接口.她可以自动向来的客人SayHello:

company.go

package hellomock

type Company struct {
  Usher Talker
}

func NewCompany(t Talker) *Company{
  return &Company{
    Usher:t,
  }
}

func ( c *Company) Meeting(gusetName string)string{
  return c.Usher.SayHello(gusetName)
}

咱们的场景已经设计好了，那么咱们传统的话，会如何实现单元测试呢？

company_test.go

package hellomock

import "testing"

func TestCompany_Meeting(t *testing.T) {
    person := NewPerson("王尼美")
    company := NewCompany(person)
    t.Log(company.Meeting("王尼玛"))
}

测试之：

/usr/local/go/bin/go test -v hellomock -run ^TestCompany_Meeting$

    company_test.go:8: Hello 王尼玛, welcome come to our company, my name is 王尼美

ok      hellomock    0.013s

如今咱们构造一个王尼美仍是很简单的，可是咱们如今要用mock对象进行模拟,这时mockgen就登场了：

➜  hellomock> mkdir mock                                        
➜  hellomock> mockgen -source=hellomock.go > mock/mock_Talker.go

这个时候，将会生成mock/mock_Talker.go文件：

须要注意的是，自动生成的文件同源文件在不一样的包下，须要新建一个目录存放

咱们并不须要在乎生成文件的内容，咱们只须要知道如何去使用便可

mock_Talker.go

// Automatically generated by MockGen. DO NOT EDIT!
// Source: hellomock.go

package mock_hellomock

import (
    gomock "github.com/golang/mock/gomock"
)

// MockTalker is a mock of Talker interface
type MockTalker struct {
    ctrl     *gomock.Controller
    recorder *MockTalkerMockRecorder
}

// MockTalkerMockRecorder is the mock recorder for MockTalker
type MockTalkerMockRecorder struct {
    mock *MockTalker
}

// NewMockTalker creates a new mock instance
func NewMockTalker(ctrl *gomock.Controller) *MockTalker {
    mock := &MockTalker{ctrl: ctrl}
    mock.recorder = &MockTalkerMockRecorder{mock}
    return mock
}

// EXPECT returns an object that allows the caller to indicate expected use
func (_m *MockTalker) EXPECT() *MockTalkerMockRecorder {
    return _m.recorder
}

// SayHello mocks base method
func (_m *MockTalker) SayHello(name string) string {
    ret := _m.ctrl.Call(_m, "SayHello", name)
    ret0, _ := ret[0].(string)
    return ret0
}

// SayHello indicates an expected call of SayHello
func (_mr *MockTalkerMockRecorder) SayHello(arg0 interface{}) *gomock.Call {
    return _mr.mock.ctrl.RecordCall(_mr.mock, "SayHello", arg0)
}

接下来看看如何去使用这个mock对象，新建一个单元测试：

func TestCompany_Meeting2(t *testing.T) {
    ctl := gomock.NewController(t)
    mock_talker := mock_hellomock.NewMockTalker(ctl)
    mock_talker.EXPECT().SayHello(gomock.Eq("王尼玛")).Return("这是自定义的返回值，能够是任意类型。")

    company := NewCompany(mock_talker)
    t.Log(company.Meeting("王尼玛"))
    //t.Log(company.Meeting("张全蛋"))
}

测试之：

/usr/local/go/bin/go test -v hellomock -run ^TestCompany_Meeting2$
    company_test.go:21: 这是自定义的返回值，能够是任意类型。
ok      hellomock    0.015s

能够看到，返回的是咱们在mock对象上定义的返回值。

须要说明的一点是，mock对象的SayHello能够接受的参数有gomock.Eq(x interface{})和gomock.Any()，前一个要求测试用例入餐严格符合x，第二个容许传入任意参数。好比咱们在注释掉的测试中传入了"张全蛋"，结果报错，测试失败：

/usr/local/go/bin/go test -v hellomock -run ^TestCompany_Meeting2$
    controller.go:113: no matching expected call: *mock_hellomock.MockTalker.SayHello([张全蛋])
exit status 1
FAIL    hellomock    0.007s

本文做为抛砖引玉，gomock还有不少高级用法，但愿你们可以自行探索。

参考文章：

https://github.com/golang/moc...

https://github.com/grpc/grpc-...

转载请注明出处 http://www.chenquan.me