如何在读取响应正文时强制出错

本篇文章向大家介绍《如何在读取响应正文时强制出错》，主要包括golang，具有一定的参考价值，需要的朋友可以参考一下。

问题内容

我已经在 go 中编写了 http 客户端包装器，我需要对其进行彻底测试。我正在使用包装器中的 ioutil.ReadAll 读取响应正文。我在弄清楚如何在 httptest 的帮助下强制读取响应正文失败时遇到了一些麻烦。

package req

func GetContent(url string) ([]byte, error) {
    response, err := httpClient.Get(url)
    // some header validation goes here
    body, err := ioutil.ReadAll(response.Body)
    defer response.Body.Close()

    if err != nil {
        errStr := fmt.Sprintf("Unable to read from body %s", err)
        return nil, errors.New(errStr)
    }

    return body, nil
}

我假设我可以这样设置一个假服务器：

package req_test

func Test_GetContent_RequestBodyReadError(t *testing.T) {

    handler := func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
    }

    ts := httptest.NewServer(http.HandlerFunc(handler))
    defer ts.Close()

    _, err := GetContent(ts.URL)

    if err != nil {
        t.Log("Body read failed as expected.")
    } else {
        t.Fatalf("Method did not fail as expected")
    }

}

我假设我需要修改 ResposeWriter。现在，我有什么办法可以修改 responseWriter 从而强制包装器中的 ioutil.ReadAll 失败？

我意识到您似乎认为它是[这篇文章](https://stackoverflow.com/questions/45126312/how-do-i-test- an-error-on-reading-from-a-request- body)的副本，尽管您可能相信或可能是这样，但将其标记为副本并不能真正帮助我。在这种情况下，“重复”帖子的答案中提供的代码对我来说意义不大。

正确答案

检查文档Response.Body以查看何时读取它可能会返回错误：

// Body represents the response body.
//
// The response body is streamed on demand as the Body field
// is read. If the network connection fails or the server
// terminates the response, Body.Read calls return an error.
//
// The http Client and Transport guarantee that Body is always
// non-nil, even on responses without a body or responses with
// a zero-length body. It is the caller's responsibility to
// close Body. The default HTTP client's Transport may not
// reuse HTTP/1.x "keep-alive" TCP connections if the Body is
// not read to completion and closed.
//
// The Body is automatically dechunked if the server replied
// with a "chunked" Transfer-Encoding.
Body io.ReadCloser

最简单的方法是从测试处理程序生成无效的 HTTP 响应。

怎么做？方法有很多，一个简单的就是在内容长度上“撒谎”：

handler := func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Length", "1")
}

这个处理程序告诉它有 1 个字节的正文，但实际上它没有发送。因此，在另一端（客户端）尝试从中读取 1 个字节时，显然不会成功，并且会导致以下错误：

Unable to read from body unexpected EOF

如果您需要模拟从请求正文（而不是从响应正文）读取错误，请参阅相关问题：[How do I test an error on reading from a request body?](https://stackoverflow.com/questions/45126312/how-do-i-test-an- error-on-reading-from-a-request-body/45126402#45126402)

好了，本文到此结束，带大家了解了《如何在读取响应正文时强制出错》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！