原文:https://gocn.io/publish/article/368
考虑测试一个函数:
func request(ctx context.Context,hc *http.Client,api string) (err error) {
var hreq *http.Request
if hreq,err = http.NewRequest("GET",api,nil); err != nil {
return nil,errors.Wrap(err,"create request")
}
var hres *http.Response
if hres,err = hc.Do(hreq.WithContext(ctx)); err != nil {
return nil,"do request")
}
defer hres.Body.Close()
var body []byte
if body,err = IoUtil.ReadAll(hres.Body); err != nil {
return nil,"read body")
}
// ......
return nil
}
这个函数的参数是一个*http.Client
,而不是接口,这个该如何测试?内嵌一个http.Client
像这样吗?
type mockHttpClient struct { http.Client }
但是,问题是这样总是很恶心不是吗?就像如果是C++中,我们只能写一个mock类从要测试的类继承,但是我们只需要重写Do
这个方法啊。
注意:对于C++而言,这是为何要求构造函数只是初始化,而不能包含逻辑,想象一个类在构造函数就访问了数据库,请问如何MOCK它?是做不到的,因此只能在构造函数初始化数据库的IP和账号等信息,提供
connect
这种函数连接数据库。
在GOLANG中,有个非常牛逼的方法,就是创建一个私有的接口,使用时用接口:
type httpDoer interface { Do(req *http.Request) (*http.Response,error) }
func request(ctx context.Context,hc httpDoer,api string) (err error) {
// ......
可以发现,很神奇的是,调用者也可以给*http.Client
,对这个改动一无所知,这难道不是极其巧妙的设计吗?我们在mock中只需要mock这个方法就可以了。
一行代码处,深藏功与名~