使用 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")
    }
}

以上是使用 httptest 單元測試 http 請求重試的詳細內容。更多資訊請關注PHP中文網其他相關文章！