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

Question

我已经创建了一些Go函数，这些函数对互联网上的服务进行HTTP GET调用并解析结果。我现在正在为这些函数编写测试用例。在我的测试用例中，我使用 go 包来模拟对这些外部服务的调用。以下是我的代码。为简洁起见，有意删除了错误检查。这是游乐场。httptestpackage mainimport (    "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 main

import (

    "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)

}