Function TestGroup_RouteNotFoundWithMiddleware

group_test.go:742–814

(t *testing.T)

}

func TestGroup_RouteNotFoundWithMiddleware(t *testing.T) {
	var testCases = []struct {
		expectBody             any
		name                   string
		whenURL                string
		expectCode             int
		givenCustom404         bool
		expectMiddlewareCalled bool
	}{
		{
			name:                   "ok, custom 404 handler is called with middleware",
			givenCustom404:         true,
			whenURL:                "/group/test3",
			expectBody:             "404 GET /group/*",
			expectCode:             http.StatusNotFound,
			expectMiddlewareCalled: true, // because RouteNotFound is added after middleware is added
		},
		{
			name:                   "ok, default group 404 handler is not called with middleware",
			givenCustom404:         false,
			whenURL:                "/group/test3",
			expectBody:             "404 GET /*",
			expectCode:             http.StatusNotFound,
			expectMiddlewareCalled: false, // because RouteNotFound is added before middleware is added
		},
		{
			name:                   "ok, (no slash) default group 404 handler is called with middleware",
			givenCustom404:         false,
			whenURL:                "/group",
			expectBody:             "404 GET /*",
			expectCode:             http.StatusNotFound,
			expectMiddlewareCalled: false, // because RouteNotFound is added before middleware is added
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {

			okHandler := func(c *Context) error {
				return c.String(http.StatusOK, c.Request().Method+" "+c.Path())
			}
			notFoundHandler := func(c *Context) error {
				return c.String(http.StatusNotFound, "404 "+c.Request().Method+" "+c.Path())
			}

			e := New()
			e.GET("/test1", okHandler)
			e.RouteNotFound("/*", notFoundHandler)

			g := e.Group("/group")
			g.GET("/test1", okHandler)

			middlewareCalled := false
			g.Use(func(next HandlerFunc) HandlerFunc {
				return func(c *Context) error {
					middlewareCalled = true
					return next(c)
				}
			})