Golang httptest 服务器循环依赖

Question

我想为一个函数编写一个测试向 url1 发出 Get 请求，该请求检索 url2向 url2 发起 Get 请求，并返回结果但是我不确定如何模拟 url2 的返回值，因为我无法在服务器启动之前获取 server.URL。但是在服务器启动后我无法更改处理程序。例如，运行下面给出错误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

Answer 1

在使用变量之前声明 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"))

    }

}))