用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是如何工作的: 我在 package hellomock type Talker interface { SayHello(word string)(response string) } 然后我们需要一个实现了
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) } 现在我们的 现在假设,我们有一个公司,公司有一个迎宾员,也就是我们的前台妹子,这个妹子实现了
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) } 我们的场景已经设计好了,那么我们传统的话,会如何实现单元测试呢?
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 王尼玛,my name is 王尼美 ok hellomock 0.013s 现在我们构造一个 ? hellomock> mkdir mock ? hellomock> mockgen -source=hellomock.go > mock/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,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对象的 /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还有很多高级用法,希望大家能够自行探索。
转载请注明出处 http://www.chenquan.me (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |