Golang httptest 服务器循环依赖

Question

我想为一个函数编写一个测试

向 url1 发出 Get 请求，该请求检索 url2
向 url2 发出 Get 请求，return返回结果

但我不确定如何模拟 url2 的 return 值，因为我无法在服务器启动前获取 server.URL。但是在服务器启动后我无法更改处理程序。例如，下面的运行给出错误 Get /url2: unsupported protocol scheme ""

package main

import (
    "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 myFunc
func 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"))
    }
}))

Golang httptest 服务器循环依赖

Golang httptest server circular dependency

unit-testing

http

circular-dependency

go

httpserver