使用 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中文网其他相关文章！
