mardi 14 mai 2019

Minimal http service test without resetting up routes?

Say I have a very simple Web service.

func main() {
    http.HandleFunc("/", sanityTest)
    log.Fatal(http.ListenAndServe(":8000", nil))
}

If I want to test it, I could minimally just have:

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

}

But that will result in a 404, since it doesn't know about the routes. So what I've seen main_test.go code do, is re-setup the handles in the test file's init, like so:

func init() {
    http.HandleFunc("/", sanityTest)
}

Which leads to duplication, and inevitably I have to create a function in main.go like:

func setupRoutes() {
        http.HandleFunc("/", sanityTest)
}

Which I find a little ugly. Am I missing a trick to instantiate the routes from main.go and avoid the init?

Aucun commentaire:

Enregistrer un commentaire