我可以为我的测试用例编写自己版本的 go 的 http client.Do() 函数吗？

Golang不知道大家是否熟悉？今天我将给大家介绍《我可以为我的测试用例编写自己版本的 go 的 http client.Do() 函数吗？》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

我有一个名为 user-service.go 的文件和相应的测试文件，名为 user-service_test.go。当我尝试获得完整的代码覆盖率时，我正在努力让一些错误条件真正发生。

这是函数：getorcreatebyaccesstoken()

//getorcreatebyaccesstoken gets a user from the database with the given access token
func (s *service) getorcreatebyaccesstoken(at string, client *client) (*user.user, fcerr.fcerr) {

var currentuser user.oauthuser

req, err := http.newrequest("get", "https://openidconnect.googleapis.com/v1/userinfo?access_token="+at, nil)
if err != nil {
    return nil, fcerr.newinternalservererror("error when setting up the network request")
}

response, err := client.httpclient.do(req)
if err != nil {
    fmt.println("error when getting the userinfo with the access token")
    return nil, fcerr.newinternalservererror("error when trying to verify user identity")
}

defer response.body.close()

contents, err := io.readall(response.body)
if err != nil {
    return nil, fcerr.newinternalservererror("error when trying to read response from google about user identity")
}

我对测试的主要控制是我可以传入 *client。

这是测试用例的一部分，我想让 io.readall 抛出错误：

h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    //manually return the message google would return on an actual request
    w.Write([]byte(googleAPIOKResponse))
})
//Call the testHTTPClient() function defined in the test file to substitute my own HandlerFunc
httpClient, teardown := testHTTPClient(h)
defer teardown()

//Call the real NewClient() from my user-service.go
client := NewClient()

//Substitute the default httpClient for the one I've just set up.
client.httpClient = httpClient

resultingUser, err := userService.GetOrCreateByAccessToken(nU.AccessToken, client)

assert.Nil(t, resultingUser)
assert.NotNil(t, err)
assert.Equal(t, http.StatusInternalServerError, err.Status())

是否有地方可以编写我自己的 .do() 方法版本，该方法将在响应中放入一些内容，从而导致 io.readall 返回错误？或者是否有更好的方法来仅使用我已经使用的预烘焙响应文本来实现错误？

正确答案

没有办法取代 do 方法，但有办法实现您的目标。

创建一个返回任意响应正文的 round tripper 类型：

type respondwithreader struct{ body io.reader }

func (rr respondwithreader) roundtrip(req *http.request) (*http.response, error) {
    return &http.response{
        proto:      "http/1.0",
        protomajor: 1,
        header:     make(http.header),
        close:      true,
        body:       ioutil.nopcloser(rr.body),
    }, nil

}

创建失败的 io.reader：

var errreadfail = errors.new("blah!")

type failreader int

func (failreader) read([]byte) (int, error) {
    return 0, errreadfail
}

将 stock 客户端与上面的传输器和读取器一起使用：

c := http.Client{Transport: respondWithReader{body: failReader(0)}}
resp, err := c.Get("http://whatever.com")
if err != nil {
    t.Error(err)
}
defer resp.Body.Close()

// ReadAll returns errReadFail
_, err = ioutil.ReadAll(resp.Body)
if err != errReadFail {
    t.Errorf("got err %v, expect %v", err, errReadFail)
}

Run the test on the Go playground。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~