Question

我想做一个简单的服务器，它可以接收来自服务的Web挂钩请求并为我处理。为了好玩，我想在Go中构建它，因为这听起来像是一种不错的语言，而且这是一个简单的项目。

服务器似乎运行正常，但是我无法使单元测试正常工作。经过检查，似乎每个网址都给出了404。我在做什么错了？

main.go

package main

import (
    "fmt"
    "log"
    "net/http"
)

func pingHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "{\"check\": \"online\"}")
}

func main() {
    // Start server
    log.Print("Starting server")
    http.HandleFunc("/ping", pingHandler)
    log.Fatal(http.ListenAndServe(":7080", nil))
}

main_test.go

package main

import (
    "fmt"
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestPingHandler(t *testing.T) {
    req, err := http.NewRequest("GET", "/ping", nil)
    if err != nil {
        t.Fatal(err)
    }
    rr := httptest.NewRecorder()
    http.DefaultServeMux.ServeHTTP(rr, req)

    status := rr.Code
    fmt.Println(status)
}

Answer 1

您的处理程序已在main中注册，但是在运行单元测试时不会调用main。因此，当您尝试通过DefaultMux进行测试时，没有注册任何处理程序，并且得到404。但是，通常，您测试的是处理程序，而不是多路复用器。因此，而不是这一行：

http.DefaultServeMux.ServeHTTP(rr, req)

您应该测试：

pingHandler(rr, req)

即使未执行main来注册处理程序，这仍然可以工作，因为您现在正在直接测试处理程序。

您还应该使用httptest.NewRequest创建测试请求； http.NewRequest用于创建要在Client中使用的请求。

Golang单元测试HTTP处理程序

1 个答案: