Function TestGroupSetNotFoundHandle

group_test.go:10–87

TestGroupSetNotFoundHandle tests the SetNotFoundHandle functionality

(t *testing.T)

// TestGroupSetNotFoundHandle tests the SetNotFoundHandle functionality
func TestGroupSetNotFoundHandle(t *testing.T) {
	tests := []struct {
		name           string
		groupPrefix    string
		requestPath    string
		expectedBody   string
		shouldUseGroup bool
	}{
		{
			name:           "Group 404 - API endpoint not found",
			groupPrefix:    "/api",
			requestPath:    "/api/users",
			expectedBody:   "API 404",
			shouldUseGroup: true,
		},
		{
			name:           "Group 404 - Similar prefix should not match",
			groupPrefix:    "/api",
			requestPath:    "/api_v2/users",
			expectedBody:   "Global 404",
			shouldUseGroup: false,
		},
		{
			name:           "Global 404 - No matching group",
			groupPrefix:    "/api",
			requestPath:    "/web/index",
			expectedBody:   "Global 404",
			shouldUseGroup: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Create app
			app := New()
			
			// Set global 404 handler
			app.SetNotFoundHandle(func(ctx Context) {
				ctx.WriteString("Global 404")
			})
			
			// Create group with custom 404 handler
			group := app.HttpServer.Group(tt.groupPrefix)
			group.SetNotFoundHandle(func(ctx Context) {
				ctx.WriteString(tt.expectedBody)
			})
			
			// Add a valid route to group
			group.GET("/exists", func(ctx Context) error {
				return ctx.WriteString("OK")
			})
			
			// Create context
			context := &HttpContext{
				response: &Response{},
				request: &Request{
					Request: &http.Request{
						URL:    &url.URL{Path: tt.requestPath},