无需重置路由的最小 http 服务测试？

Question

假设我有一个非常简单的 Web 服务。func main() {    http.HandleFunc("/", sanityTest)    log.Fatal(http.ListenAndServe(":8000", nil))}如果我想测试它，我至少可以拥有：func ExampleTest() {        server := httptest.NewServer(http.DefaultServeMux)        defer server.Close()        resp, err := http.Get(server.URL)        if err != nil {                log.Fatal(err)        }        body, _ := ioutil.ReadAll(resp.Body)        fmt.Println(resp.StatusCode)        fmt.Println(resp.Header.Get("Content-Type"))        fmt.Println(string(body))        // Output:        // 200        // text/plain; charset=utf-8        // OK}但这将导致 404，因为它不知道路由。所以我看到 main_test.go 代码所做的是在测试文件的 init 中重新设置句柄，如下所示：func init() {    http.HandleFunc("/", sanityTest)}这会导致重复，不可避免地我必须在 main.go 中创建一个函数，例如：func setupRoutes() {        http.HandleFunc("/", sanityTest)}我觉得有点难看。我是否缺少从 main.go 实例化路由并避免 init 的技巧？

Answer 1

您可以在测试和 main.go 文件之间重复使用路由，如果您想在处理程序中模拟某些内容，这也很有帮助（在router()下面的 func 中添加一个新参数）

主要去：

func sanityTest(w http.ResponseWriter, r *http.Request) {

    fmt.Fprintf(w, "%s", "sanity test")

}

func router() *http.ServeMux {

    h := http.NewServeMux()

    h.HandleFunc("/", sanityTest)

    return h

}

func main() {

    http.ListenAndServe(":8080", router())

}

main_test.go：

func TestSanity(t *testing.T) {

    tests := []struct {

        name string

        uri  string

        want string

    }{

        {"1", "/", "sanity test"},

    }

    ts := httptest.NewServer(router())

    defer ts.Close()

    for _, tt := range tests {

        t.Run(tt.name, func(t *testing.T) {

            url := ts.URL + tt.uri

            resp, _ := http.Get(url)

            respBody, _ := ioutil.ReadAll(resp.Body)

            resp.Body.Close()

            got := string(respBody)

            if got != tt.want {

                t.Errorf("got %s, Want %s", got, tt.want)

            }

        })

    }

}