如何测试从请求正文中读取的错误？

今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《如何测试从请求正文中读取的错误？》，主要内容是讲解golang等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

问题内容

我正在为 golang 中的 http 处理程序编写单元测试。在查看此代码覆盖率报告时，我遇到了以下问题：从请求中读取请求正文时，ioutil.ReadAll可能会返回我需要处理的错误。然而，当我为我的处理程序编写单元测试时，我不知道如何以一种会触发此类错误的方式向我的处理程序发送请求（内容的过早结束似乎不会产生这样的错误，但会在解组身体）。这就是我想要做的：

package demo

import (
    "bytes"
    "io/ioutil"
    "net/http"
    "net/http/httptest"
    "testing"
)

func HandlePostRequest(w http.ResponseWriter, r *http.Request) {
    body, bytesErr := ioutil.ReadAll(r.Body)
    if bytesErr != nil {
        // intricate logic goes here, how can i test it?
        http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
        return
    }
    defer r.Body.Close()
    // continue...
}

func TestHandlePostRequest(t *testing.T) {
    ts := httptest.NewServer(http.HandlerFunc(HandlePostRequest))
    data, _ := ioutil.ReadFile("testdata/fixture.json")
    res, err := http.Post(ts.URL, "application/json", bytes.NewReader(data))
    // continue...
}

我怎样才能写一个测试用例HandlePostRequest也涵盖bytesErr不存在的情况nil？

正确答案

你可以创建和使用http.Request你伪造的，在读取它的主体时故意返回一个错误。您不一定需要一个全新的请求，一个有缺陷的主体就足够了（这是一个io.ReadCloser）。

最简单的实现方法是使用httptest.NewRequest()可以传递一个io.Reader值的函数，该值将用作io.ReadCloser请求正文（包装为 an ）。

io.Reader这是一个在尝试读取时故意返回错误的示例：

type errReader int

func (errReader) Read(p []byte) (n int, err error) {
    return 0, errors.New("test error")
}

将涵盖您的错误案例的示例：

func HandlePostRequest(w http.ResponseWriter, r *http.Request) {
    defer r.Body.Close()
    body, err := ioutil.ReadAll(r.Body)
    if err != nil {
        fmt.Printf("Error reading the body: %v\n", err)
        return
    }
    fmt.Printf("No error, body: %s\n", body)
}

func main() {
    testRequest := httptest.NewRequest(http.MethodPost, "/something", errReader(0))
    HandlePostRequest(nil, testRequest)
}

输出（在Go Playground上试试）：

Error reading the body: test error

如果您需要模拟从响应正文（而不是请求正文）读取错误，请参阅相关问题：[How to force error on reading response body](https://stackoverflow.com/questions/53171123/how-to-force-error-on- reading-response-body/53173459#53173459)

终于介绍完啦！小伙伴们，这篇关于《如何测试从请求正文中读取的错误？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！