Wie schreibe ich Test mit httprouter

Question

Ich schreibe Tests für einen einfachen REST-Service in GoLang. Aber, weil ich julienschmidt/httprouter als Routing-Bibliothek verwende. Ich kämpfe darum, wie man einen Test schreibt.

main.go

package main 

func main() { 
    router := httprouter.New() 
    bookController := controllers.NewBookController() 
    router.GET("/book/:id", bookController.GetBook) 
    http.ListenAndServe(":8080", router) 
} 

Controller

package controllers 

type BookController struct {} 

func NewBookController *BookController { 
    return &BookController() 
} 

func (bc BookController) GetBook(w http.ResponseWriter, r *http.Request, p httprouter.Params) { 
    fmt.Fprintf(w,"%s", p) 
} 

Meine Frage ist: Wie kann diese während testen GetBook weder Httphandler noch HttpHandle

Wenn ich einen traditionellen Handler, der Test wird so einfach sein

func TestGetBook(t *testing.T) { 
    req, _ := http.NewRequest("GET", "/book/sampleid", nil) 
    rr := httptest.NewRecorder() 
    handler := controllers.NewBookController().GetBook 
    handler.ServeHTTP(rr,req) 
    if status := rr.code; status != http.StatusOK { 
    t.Errorf("Wrong status") 
    } 
} 

Das Problem ist, httprouter ist nicht Handler oder handlefunc. So bin ich jetzt stecken

Answer 1

Nur Spin einen neuen Router für jeden Test und registrieren Sie dann den Handler im Test, dann übergeben Sie die Testanforderung an den Router, nicht der Handler, so dass der Router die Pfadparameter analysieren und übergeben kann sie an den Handler.

func TestGetBook(t *testing.T) { 
    handler := controllers.NewBookController() 
    router := httprouter.New() 
    router.GET("/book/:id", handler.GetBook) 

    req, _ := http.NewRequest("GET", "/book/sampleid", nil) 
    rr := httptest.NewRecorder() 

    router.ServeHTTP(rr, req) 
    if status := rr.code; status != http.StatusOK { 
     t.Errorf("Wrong status") 
    } 
} 

Answer 2

Sie benötigen Handler wickeln, so dass es als http.HandlerFunc zugegriffen werden kann:

func TestGetBook(t *testing.T) { 
    req, _ := http.NewRequest("GET", "/book/sampleid", nil) 
    rr := httptest.NewRecorder() 
    handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { 
     controllers.NewBookController().GetBook(w, r, httprouter.Params{}) 
    }) 
    handler.ServeHTTP(rr,req) 
    if status := rr.code; status != http.StatusOK { 
     t.Errorf("Wrong status") 
    } 
} 

Wenn Ihr Handler Parameter erfordert, werden Sie entweder sie von der Anforderung manuell zu analysieren, oder gib sie nur als drittes Argument an.