本篇文章给大家分享的是有关使用gomock怎么对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怎么对mock进行测试,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注创新互联成都网站设计公司行业资讯频道。
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
售后响应及时
7×24小时客服热线数据备份
更安全、更高效、更稳定价格公道精准
项目经理精准报价不弄虚作假合作无风险
重合同讲信誉,无效全额退款