Function TestCustomHTTPMethod

mux_test.go:1761–1805

(t *testing.T)

}

func TestCustomHTTPMethod(t *testing.T) {
	// first we must register this method to be accepted, then we
	// can define method handlers on the router below
	RegisterMethod("BOO")

	r := NewRouter()
	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("."))
	})

	// note the custom BOO method for route /hi
	r.MethodFunc("BOO", "/hi", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("custom method"))
	})

	ts := httptest.NewServer(r)
	defer ts.Close()

	if _, body := testRequest(t, ts, "GET", "/", nil); body != "." {
		t.Fatal(body)
	}
	if _, body := testRequest(t, ts, "BOO", "/hi", nil); body != "custom method" {
		t.Fatal(body)
	}

	var expectRoutes = map[string]string{
		"GET": "/",
		"BOO": "/hi",
	}
	Walk(r, func(method string, route string, handler http.Handler, _ ...func(http.Handler) http.Handler) error {
		r, ok := expectRoutes[method]
		if !ok {
			t.Fatalf("unexpected method %s", method)
		}
		if r != route {
			t.Fatalf("expected route %s, got %s", r, route)
		}
		delete(expectRoutes, method)

		return nil
	})
	if len(expectRoutes) != 0 {
		t.Fatalf("missing expected methods: %v", expectRoutes)
	}
}

func TestMuxMatch(t *testing.T) {
	r := NewRouter()