我已經(jīng)創(chuàng)建了一些Go函數(shù)，這些函數(shù)對(duì)互聯(lián)網(wǎng)上的服務(wù)進(jìn)行HTTP GET調(diào)用并解析結(jié)果。我現(xiàn)在正在為這些函數(shù)編寫(xiě)測(cè)試用例。在我的測(cè)試用例中，我使用 go 包來(lái)模擬對(duì)這些外部服務(wù)的調(diào)用。以下是我的代碼。為簡(jiǎn)潔起見(jiàn)，有意刪除了錯(cuò)誤檢查。這是游樂(lè)場(chǎng)。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)}這是它產(chǎn)生的輸出：myResponse1:{"A":"B"}myResponse2:{"A":"B"}如您所見(jiàn)，我已經(jīng)創(chuàng)建了一些虛擬的HTTP響應(yīng)數(shù)據(jù)，當(dāng)您向發(fā)送HTTP請(qǐng)求時(shí)，它實(shí)際上會(huì)命中一個(gè)臨時(shí)的HTTP服務(wù)器，該服務(wù)器會(huì)用虛擬數(shù)據(jù)進(jìn)行響應(yīng)。涼！{"A":"B"}srv.URL當(dāng)您將第二個(gè) HTTP 請(qǐng)求發(fā)送到 時(shí)，它會(huì)再次使用相同的虛擬數(shù)據(jù)進(jìn)行響應(yīng)。但這就是我的問(wèn)題出現(xiàn)的地方。我希望臨時(shí)HTTP服務(wù)器在第二次和第三次收到請(qǐng)求時(shí)返回一些不同的數(shù)據(jù)。srv.URL{"C":"D"}{"E":"F"}如何更改函數(shù)的第一行，以便服務(wù)器在后續(xù) HTTP 調(diào)用中響應(yīng)我所需的數(shù)據(jù)？main()
1 回答
智慧大石
TA貢獻(xiàn)1946條經(jīng)驗(yàn) 獲得超3個(gè)贊
你可以使用一個(gè)黑客，如下（游樂(lè)場(chǎng)：這里)
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)
}
- 1 回答
- 0 關(guān)注
- 148 瀏覽
添加回答
舉報(bào)
0/150
提交
取消