Function TestMux

handler_test.go:36–136

(t *testing.T)

}

func TestMux(t *testing.T) {
	// Start the servers
	srv1 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch r.URL.Path {
		case "/robots.txt":
			w.Write([]byte(`
User-agent: *
Disallow: /deny
`))
			return
		case "/204":
			w.Header().Set("Content-Type", "text/plain")
			w.WriteHeader(204)
			w.Write(nil)
			return
		case "/4xx":
			w.Header().Set("Content-Type", "text/plain")
			w.WriteHeader(404)
			w.Write(nil)
		case "/json":
			w.Header().Set("Content-Type", "application/json")
		default:
			w.Header().Set("Content-Type", "text/html; charset=utf-8")
		}
		w.Write([]byte("1"))
	}))
	defer srv1.Close()
	srv2 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("ok"))
	}))
	defer srv2.Close()
	srvu, err := url.Parse(srv1.URL)
	if err != nil {
		t.Fatal(err)
	}
	srv1Host := srvu.Host
	srvu, err = url.Parse(srv2.URL)
	if err != nil {
		t.Fatal(err)
	}
	srv2Host := srvu.Host

	// List of test cases
	cases := []struct {
		url   string
		trace string
	}{
		0: {srv2.URL + "/none", "d"}, // no specific handler, use default
		1: {srv1.URL + "/json", "j"}, // json-specific handler
		2: {srv1.URL + "/deny", "a"}, // ErrDisallowed, use any errors handler
		3: {srv1.URL + "/a", "g"},    // GET text/html
		4: {srv1.URL + "/204", "s"},  // status 204
		5: {srv1.URL + "/4xx", "r"},  // status range 4xx
		6: {srv1.URL + "/b", "p"},    // path-specific handler
		7: {srv1.URL + "/baba", "q"}, // path-specific handler
		8: {srv1.URL + "/b/c", "p"},  // path-specific handler
		9: {srv2.URL + "/zz", "r"},   // custom predicate
	}