如何使用 Go 的 http 测试模拟多个不同的 HTTP 响应？

Question

我已经创建了一些Go函数，这些函数对互联网上的服务进行HTTP GET调用并解析结果。

我现在正在为这些函数编写测试用例。在我的测试用例中，我使用 go 包来模拟对这些外部服务的调用。以下是我的代码。为简洁起见，有意删除了错误检查。这是游乐场。httptest

package main

import (

    "fmt"

    "io"

    "context"

    "net/http"

    "net/http/httptest"

)

func handlerResponse() http.Handler {

    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

        w.WriteHeader(http.StatusOK)

        w.Write([]byte(`{"A":"B"}`))

    })

}

func buildMyRequest(ctx context.Context, url string) *http.Request {

    request, _ := http.NewRequestWithContext(ctx, "GET", url, nil)

    return request

}

func myPrint(response *http.Response) {

    b := make([]byte, 60000)

    for {

        _, err := response.Body.Read(b)

        if err == io.EOF {

            break

        }

    }

    fmt.Println(string(b))

}

func main() {

    srv := httptest.NewServer(handlerResponse())            

    client := http.Client{}

    myResponse1, _ := client.Do(buildMyRequest(context.Background(), srv.URL))

    fmt.Println("myResponse1:")

    myPrint(myResponse1)

    

    myResponse2, _ := client.Do(buildMyRequest(context.Background(), srv.URL))

    fmt.Println("myResponse2:")

    myPrint(myResponse2)

}

这是它产生的输出：

myResponse1:

{"A":"B"}

myResponse2:

{"A":"B"}

如您所见，我已经创建了一些虚拟的HTTP响应数据，当您向发送HTTP请求时，它实际上会命中一个临时的HTTP服务器，该服务器会用虚拟数据进行响应。凉！{"A":"B"}srv.URL

当您将第二个 HTTP 请求发送到 时，它会再次使用相同的虚拟数据进行响应。但这就是我的问题出现的地方。我希望临时HTTP服务器在第二次和第三次收到请求时返回一些不同的数据。srv.URL{"C":"D"}{"E":"F"}

如何更改函数的第一行，以便服务器在后续 HTTP 调用中响应我所需的数据？main()

Answer 1

你可以使用一个黑客，如下（游乐场：这里)package mainimport (    "fmt"    "io"    "context"    "net/http"    "net/http/httptest"    "sync")type responseWriter struct{   resp map[int]string   count int   lock *sync.Mutex}func NewResponseWriter()*responseWriter{   r := new(responseWriter)   r.lock = new(sync.Mutex)   r.resp = map[int]string{    0: `{"E":"F"}`,    1: `{"A":"B"}`,    2: `{"C":"D"}`,   }   r.count = 0   return r}func (r *responseWriter)GetResp()string{   r.lock.Lock()   defer r.lock.Unlock()   r.count ++   return r.resp[r.count%3]}func handlerResponse(rr *responseWriter) http.Handler {    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {        w.WriteHeader(http.StatusOK)        w.Write([]byte(rr.GetResp()))    })}func buildMyRequest(ctx context.Context, url string) *http.Request {    request, _ := http.NewRequestWithContext(ctx, "GET", url, nil)    return request}func myPrint(response *http.Response) {    b := make([]byte, 60000)    for {        _, err := response.Body.Read(b)        if err == io.EOF {            break        }    }    fmt.Println(string(b))}func main() {        rr := NewResponseWriter()    srv := httptest.NewServer(handlerResponse(rr))      client := http.Client{}    myResponse1, err := client.Do(buildMyRequest(context.Background(), srv.URL))    if err != nil{       fmt.Println(err)       return    }        defer myResponse1.Body.Close()    fmt.Println("myResponse1:")    myPrint(myResponse1)        myResponse2, err := client.Do(buildMyRequest(context.Background(), srv.URL))    if err != nil{       fmt.Println(err)       return    }        defer myResponse2.Body.Close()    fmt.Println("myResponse2:")    myPrint(myResponse2)}