利用 httptest 对 http 请求进行重试单元测试

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《利用 httptest 对 http 请求进行重试单元测试》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

我正在努力为名为 HttpRequest 的组件编写单元测试，该组件包装 HTTP 请求并处理响应解组。最近，我向该组件添加了一项功能，允许它在第一次尝试时遇到“连接被拒绝”错误时重试 HTTP 请求。

要使用 HttpRequest 组件，我这样调用一次：user, err := HttpRequest[User](config)。 config 参数包含执行请求的所有必要信息，例如 URL、方法、超时、重试次数和请求正文。它还将响应正文解组为指定类型的实例（在本例中为 User）

当我尝试测试初始请求失败并出现“连接被拒绝”错误但第二次尝试成功的场景时，就会出现问题。重试发生在组件内部，因此我只对组件进行一次调用。

我发现为这种情况创建单元测试具有挑战性，因为为了使请求因“连接被拒绝”而失败，被调用的端口上不需要有侦听器。问题是，当使用 httptest 时，它总是在创建实例时侦听端口，即使使用 httptest.NewUnstartedServer 也是如此。因此，在创建 httptest 实例后，我的客户端代码中将永远不会遇到“连接被拒绝”错误。

但是，在创建 httptest 实例之前，我不知道它将侦听哪个端口。 httptest 总是选择一个随机端口，并且无法以编程方式指定一个端口。这意味着我无法在创建 httptest 实例之前进行 HttpRequest 调用。

有人对如何有效地对这种场景进行单元测试有任何想法吗？

正确答案

newunstartedserver 非常简单：

func newunstartedserver(handler http.handler) *server {
    return &server{
        listener: newlocallistener(),
        config:   &http.server{handler: handler},
    }
}

如果自己选择一个端口适合您，您可以这样做：

func mynewunstartedserver(port int, handler http.handler) *httptest.server {
    addr := fmt.sprintf("127.0.0.1:%d", port)
    l, err := net.listen("tcp", addr)
    if err != nil {
        addr = fmt.sprintf("[::1]::%d", port)
        if l, err = net.listen("tcp6", addr); err != nil {
            panic(fmt.sprintf("httptest: failed to listen on a port: %v", err))
        }
    }
    return &httptest.server{
        listener: l,
        config:   &http.server{handler: handler},
    }
}

创建监听器的代码修改自httptest.newlocallistener。

另一个选择是实现 http.roundtripper 接口并使用此 roundtripper 创建一个 http.client。以下是从 net/ 复制的示例http/client_test.go：

type recordingTransport struct {
    req *Request
}

func (t *recordingTransport) RoundTrip(req *Request) (resp *Response, err error) {
    t.req = req
    return nil, errors.New("dummy impl")
}

func TestGetRequestFormat(t *testing.T) {
    setParallel(t)
    defer afterTest(t)
    tr := &recordingTransport{}
    client := &Client{Transport: tr}
    url := "http://dummy.faketld/"
    client.Get(url) // Note: doesn't hit network
    if tr.req.Method != "GET" {
        t.Errorf("expected method %q; got %q", "GET", tr.req.Method)
    }
    if tr.req.URL.String() != url {
        t.Errorf("expected URL %q; got %q", url, tr.req.URL.String())
    }
    if tr.req.Header == nil {
        t.Errorf("expected non-nil request Header")
    }
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。