模拟 Go 测试中的 HTTP 请求 504 超时错误的方法

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《模拟 Go 测试中的 HTTP 请求 504 超时错误的方法》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我正在尝试向 go 中的库添加 timeout 选项，并编写了以下测试来模拟该行为。

func TestClientTimeout(t *testing.T) {
    backend := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        d := map[string]interface{}{
            "id":    "12",
            "scope": "test-scope",
        }

        time.Sleep(100 * time.Millisecond)
        e := json.NewEncoder(w)
        err := e.Encode(&d)
        if err != nil {
            t.Error(err)
        }
        w.WriteHeader(http.StatusOK)
    }))

    url := backend.URL
    ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
    defer cancel()
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        t.Error("Request error", err)
    }

    resp, err := http.DefaultClient.Do(req.WithContext(ctx))
    if err != nil {
        t.Error("Response error", err)
    }

    defer resp.Body.Close()

    t.Log(">>>>>>>Response is: ", resp)
}

但我总是遇到以下错误，而不是 http.statusgatewaytimeout

=== 运行测试客户端超时 --- 失败：testclienttimeout（0.05s） client_test.go:37: 请求之前的时间戳 2018-07-13 09:10:14.936898 +0200 cest m=+0.002048937 client_test.go:40：响应错误获取http://127.0.0.1:49597：超出上下文截止日期 恐慌：运行时错误：无效的内存地址或零指针取消引用[已恢复] 恐慌：运行时错误：无效的内存地址或零指针取消引用

如何修复此测试，以返回带有 http.statusgatewaytimeout(504) 状态代码的响应？

解决方案

您收到错误 context 最后期限超出 的原因是因为请求客户端的 context.context 上的超时比服务器端处理程序中的超时短。这意味着 context.context 以及客户端 http.defaultclient 在写入任何响应之前都会放弃。

panic: 运行时错误：无效的内存地址... 是因为您推迟关闭响应的正文，但如果客户端返回错误，则响应为 nil。

这里响应为nil，如果错误非nil，则将t.error更改为t.fatal

resp, err := http.defaultclient.do(req.withcontext(ctx))
if err != nil {
    // this should be t.fatal, or don't do the body close if there's an error
    t.error("response error", err)
}

defer resp.body.close()

找到问题的真正根源，http.statusgatewaytimeout 是服务器端超时，这意味着创建的任何超时都必须在服务器端。客户端 http.defaultclient 永远不会创建它自己的服务器错误响应代码。

要创建服务器端超时，您可以将处理程序函数包装在 http.timeouthandler 中：

handlerfunc := http.handlerfunc(func(w http.responsewriter, r *http.request) {
    d := map[string]interface{}{
        "id":    "12",
        "scope": "test-scope",
    }

    time.sleep(100 * time.millisecond)
    e := json.newencoder(w)
    err := e.encode(&d)
    if err != nil {
        t.error(err)
    }
    w.writeheader(http.statusok)
})

backend := httptest.newserver(http.timeouthandler(handlerfunc, 20*time.millisecond, "server timeout"))

但是，这将创建 503 - service unavailable 错误响应代码。

了解 504 的重要一点是，这是一个“网关”或“代理”错误响应代码。这意味着该代码不太可能来自实际处理请求的服务器。此代码更常见于负载均衡器和代理。

504 网关超时 服务器在充当网关或代理时，没有从它需要访问的上游服务器收到及时的响应以完成请求。

您已经使用 httptest.newserver(...) 在测试方法中模拟了 http.server，因此您可以在处理函数中手动返回 http.statusgatewaytimeout 响应状态。

handlerFunc := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusGatewayTimeout)
})

backend := httptest.NewServer(handlerFunc)

好了，本文到此结束，带大家了解了《模拟 Go 测试中的 HTTP 请求 504 超时错误的方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！