如何使用 httptest 模拟响应正文读取错误？

使用 httptest 模拟响应正文读取时的错误

使用 httptest 测试 HTTP 客户端时，可能需要在响应正文中模拟错误阅读。

考虑以下消耗响应主体的包装函数：

<code class="go">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
}</code>

要测试此函数，可以使用 httptest 设置假服务器：

<code class="go">package req_test

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

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")
    }

}</code>

要强制读取错误，从文档中了解 Response.Body 的行为至关重要：

// Body represents the response body.
//
// ...
// If the network connection fails or the server terminates the response, Body.Read calls return an error.

因此，模拟错误的一种简单方法是从测试处理程序创建无效的 HTTP 响应。例如，谎报内容长度可能会导致客户端出现意外的 EOF 错误。

此类处理程序的示例：

<code class="go">handler := func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Length", "1")
}</code>

以上是如何使用 httptest 模拟响应正文读取错误？的详细内容。更多信息请关注PHP中文网其他相关文章！