卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章35208本站已运行394

使用 httptest 单元测试 http 请求重试

使用 httptest 单元测试 http 请求重试

php小编百草为大家介绍了一种名为"httptest"的单元测试工具,它能够帮助开发者在进行http请求重试时进行测试。这个工具不仅可以模拟各种http请求,还能够自动重试请求,以确保代码的稳定性和可靠性。使用httptest工具,开发者可以轻松进行http请求的单元测试,并且可以方便地对请求进行重试,以应对网络不稳定或者其他异常情况。这种工具的使用能够大大提升开发效率,减少出错的可能性,是每个php开发者都值得尝试的工具。

问题内容

我正在努力为名为 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")
    }
}
卓越飞翔博客
上一篇: helm :将 YAML 转换为 JSON 时出错:yaml:第 xx 行:未找到预期的密钥
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏