我想為一個(gè)函數(shù)編寫一個(gè)測試向 url1 發(fā)出 Get 請(qǐng)求,該請(qǐng)求檢索 url2向 url2 發(fā)起 Get 請(qǐng)求,并返回結(jié)果但是我不確定如何模擬 url2 的返回值,因?yàn)槲覠o法在服務(wù)器啟動(dòng)之前獲取 server.URL。但是在服務(wù)器啟動(dòng)后我無法更改處理程序。例如,運(yùn)行下面給出錯(cuò)誤Get /url2: unsupported protocol scheme ""package mainimport ( "fmt" "io/ioutil" "net/http" "net/http/httptest")func myFunc(client *http.Client, url1 string) string { url2 := get(client, url1) return get(client, url2)}func get(client *http.Client, url string) string { resp, err := client.Get(url) if err != nil { fmt.Println(err) } body, err := ioutil.ReadAll(resp.Body) defer resp.Body.Close() if err != nil { fmt.Println(err) } return string(body)}// test myFuncfunc main() { srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.String() { case "/url1": w.Write([]byte("/url2")) // how to specify srv.URL+"/url2" here? case "/url2": w.Write([]byte("return data")) } })) defer srv.Close() myFunc(srv.Client(), srv.URL+"/url1")}來源:https ://onlinegdb.com/UpKlXfw45
1 回答

慕桂英3389331
TA貢獻(xiàn)2036條經(jīng)驗(yàn) 獲得超8個(gè)贊
在使用變量之前聲明 srv 變量。
var srv *httptest.Server
srv = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.URL.String() {
case "/url1":
w.Write([]byte(srv.URL + "/url2"))
case "/url2":
w.Write([]byte("return data"))
}
}))
- 1 回答
- 0 關(guān)注
- 113 瀏覽
添加回答
舉報(bào)
0/150
提交
取消