GOLANG宽泛接口在测试中的大用处

前端之家收集整理的这篇文章主要介绍了GOLANG宽泛接口在测试中的大用处前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

原文: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这种函数连接数据库

备注:上面只是拿数据库连接打个比方,实际上从MOCK角度来说,构造函数只能初始化内存对象,其他的应该啥也不干。

在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这个方法就可以了。

一行代码处,深藏功与名~

猜你在找的Go相关文章