在Golang中处理 ioutil.ReadAll(response.Body) 的错误

在 Go 语言中处理 `ioutil.ReadAll(response.Body)` 错误时，使用 mock 技术可以模拟 `response.Body` 的行为，从而控制其返回的值和错误。通过实现 `io.ReadCloser` 接口，可以创建自定义的 mock body，在调用 `read()` 方法时触发特定错误。这允许开发者在测试中模拟错误场景，并验证代码在处理这些错误时的正确性。

由于某种原因，我似乎无法获取 ioutil.readall(res.body)，其中 res 是 res 返回的 *http.response，err := hc.do(redirectrequest) （对于 hc http.客户端、redirectrequest *http.request）。

迄今为止的测试策略

任何时候当我在 sut 中看到 hc.do 或 http.request 时，我的直觉是启动一个假服务器并将适当的应用程序状态指向它。对于本次测试，这样的服务器如下所示：

badServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { 
    // some stuff
    w.Write([some bad bytes])
}))
defer badServer.Close()

我似乎没有办法控制 res.body，这实际上是唯一阻止我 100% 测试完成的东西。

我尝试在 errorthrowingserver 的处理程序函数中，将 r.body 设置为存根 io.readcloser，该存根在调用 read() 时会引发错误，但这不会影响 res。

解决方案

你可以嘲笑身体。基本上 body 是一个 io.readcloser 接口，因此，您可以执行以下操作：

import (
    "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/mock"
)

type mockreadcloser struct {
    mock.mock
}

func (m *mockreadcloser) read(p []byte) (n int, err error) {
    args := m.called(p)
    return args.int(0), args.error(1)
}

func (m *mockreadcloser) close() error {
    args := m.called()
    return args.error(0)
}

func testtestingsomethingwithbodyerror(t *testing.t) {
    mockreadcloser := mockreadcloser{}
    // if read is called, it will return error
    mockreadcloser.on("read", mock.anythingoftype("[]uint8")).return(0, fmt.errorf("error reading"))
    // if close is called, it will return error
    mockreadcloser.on("close").return(fmt.errorf("error closing"))

    request := &http.request{
        // pass the mock address
        body: &mockreadcloser,
    }

    expected := "what you expected"
    result := yourmethod(request)

    assert.equal(t, expected, result)

    mockreadcloser.assertexpectations(t)
}

要停止阅读，您可以使用：

mockReadCloser.On("Read", mock.AnythingOfType("[]uint8")).Return(0, io.EOF).Once()

好了，本文到此结束，带大家了解了《在Golang中处理 ioutil.ReadAll(response.Body) 的错误》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！