在 Golang 中测试 HTTP 路由

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《在 Golang 中测试 HTTP 路由》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到golang等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

问题内容

我正在使用 Gorilla mux 和 net/http 包创建一些路由，如下所示

package routes

//some imports

//some stuff

func AddQuestionRoutes(r *mux.Router) {
    s := r.PathPrefix("/questions").Subrouter()
    s.HandleFunc("/{question_id}/{question_type}", getQuestion).Methods("GET")
    s.HandleFunc("/", postQuestion).Methods("POST")
    s.HandleFunc("/", putQuestion).Methods("PUT")
    s.HandleFunc("/{question_id}", deleteQuestion).Methods("DELETE")
}

我正在尝试编写一个测试来测试这些路线。例如，我正在尝试GET专门测试路线以获取400返回，因此我有以下测试代码。

package routes

//some imports

var m *mux.Router
var req *http.Request
var err error
var respRec *httptest.ResponseRecorder

func init() {
    //mux router with added question routes
    m = mux.NewRouter()
    AddQuestionRoutes(m)

    //The response recorder used to record HTTP responses
    respRec = httptest.NewRecorder()
}

func TestGet400(t *testing.T) {
    //Testing get of non existent question type
    req, err = http.NewRequest("GET", "/questions/1/SC", nil)
    if err != nil {
        t.Fatal("Creating 'GET /questions/1/SC' request failed!")
    }

    m.ServeHTTP(respRec, req)

    if respRec.Code != http.StatusBadRequest {
        t.Fatal("Server error: Returned ", respRec.Code, " instead of ", http.StatusBadRequest)
    }
}

但是，当我运行这个测试时，我得到一个404可以想象的，因为请求没有被正确路由。？

当我从浏览器测试这个 GET 路由时，它确实返回 a400所以我确定测试的设置方式存在问题。

正确答案

这里使用 init() 是值得怀疑的。它仅作为程序初始化的一部分执行一次。相反，也许是这样的：

func setup() {
    //mux router with added question routes
    m = mux.NewRouter()
    AddQuestionRoutes(m)

    //The response recorder used to record HTTP responses
    respRec = httptest.NewRecorder()
}

func TestGet400(t *testing.T) {
    setup()
    //Testing get of non existent question type
    req, err = http.NewRequest("GET", "/questions/1/SC", nil)
    if err != nil {
        t.Fatal("Creating 'GET /questions/1/SC' request failed!")
    }

    m.ServeHTTP(respRec, req)

    if respRec.Code != http.StatusBadRequest {
        t.Fatal("Server error: Returned ", respRec.Code, " instead of ", http.StatusBadRequest)
    }
}

在每个适当的测试用例开始时调用 setup() 。您的原始代码与其他测试共享相同的 respRec，这可能会污染您的测试结果。

如果您需要一个提供更多功能（如 setup/teardown 固定装置）的测试框架，请参阅gocheck等包。

今天关于《在 Golang 中测试 HTTP 路由》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！