如何在读取响应正文时强加错误

小编典典

如何在读取响应正文时强加错误

我已经在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失败吗？

我意识到您似乎认为这是这篇文章的副本，尽管您可能认为是这样，也可能是这样，但仅将其标记为副本并不能真正帮助我。在这种情况下，“重复”帖子中答案中提供的代码对我来说意义不大。

阅读 296

2020-07-02

共1个答案

小编典典

查看的文档，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

如果您需要模拟从请求正文（而不是从响应正文）读取错误

2020-07-02