Function TestRouterHandleMethodOptions

router_test.go:925–991

(t *testing.T)

}

func TestRouterHandleMethodOptions(t *testing.T) {
	e := New()
	e.contextPathParamAllocSize.Store(1)
	r := e.router

	r.Add(Route{Method: http.MethodGet, Path: "/users", Handler: handlerFunc})
	r.Add(Route{Method: http.MethodPost, Path: "/users", Handler: handlerFunc})
	r.Add(Route{Method: http.MethodPut, Path: "/users/:id", Handler: handlerFunc})
	r.Add(Route{Method: http.MethodGet, Path: "/users/:id", Handler: handlerFunc})

	var testCases = []struct {
		name              string
		whenMethod        string
		whenURL           string
		expectAllowHeader string
		expectStatus      int
	}{
		{
			name:              "allows GET and POST handlers",
			whenMethod:        http.MethodOptions,
			whenURL:           "/users",
			expectAllowHeader: "OPTIONS, GET, POST",
			expectStatus:      http.StatusNoContent,
		},
		{
			name:              "allows GET and PUT handlers",
			whenMethod:        http.MethodOptions,
			whenURL:           "/users/1",
			expectAllowHeader: "OPTIONS, GET, PUT",
			expectStatus:      http.StatusNoContent,
		},
		{
			name:              "GET does not have allows header",
			whenMethod:        http.MethodGet,
			whenURL:           "/users",
			expectAllowHeader: "",
			expectStatus:      http.StatusOK,
		},
		{
			name:              "path with no handlers does not set Allows header",
			whenMethod:        http.MethodOptions,
			whenURL:           "/notFound",
			expectAllowHeader: "",
			expectStatus:      http.StatusNotFound,
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			req := httptest.NewRequest(tc.whenMethod, tc.whenURL, nil)
			rec := httptest.NewRecorder()
			c := e.NewContext(req, rec)

			h := r.Route(c)
			err := h(c)

			if tc.expectStatus >= 400 {
				assert.Error(t, err)
				he := err.(HTTPStatusCoder)