Function TestRouteHeaders

middleware/route_headers_test.go:10–185

(t *testing.T)

)

func TestRouteHeaders(t *testing.T) {
	t.Run("empty router should call next handler exactly once", func(t *testing.T) {
		var callCount atomic.Int32

		hr := RouteHeaders()

		handler := hr.Handler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			callCount.Add(1)
			w.WriteHeader(http.StatusOK)
		}))

		req := httptest.NewRequest("GET", "/", nil)
		rec := httptest.NewRecorder()

		handler.ServeHTTP(rec, req)

		if callCount.Load() != 1 {
			t.Errorf("expected next handler to be called exactly once, but was called %d times", callCount.Load())
		}
	})

	t.Run("matching header should route to correct middleware", func(t *testing.T) {
		var matchedRoute string

		hr := RouteHeaders().
			Route("Host", "example.com", func(next http.Handler) http.Handler {
				return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
					matchedRoute = "example.com"
					next.ServeHTTP(w, r)
				})
			}).
			Route("Host", "other.com", func(next http.Handler) http.Handler {
				return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
					matchedRoute = "other.com"
					next.ServeHTTP(w, r)
				})
			})

		handler := hr.Handler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.WriteHeader(http.StatusOK)
		}))

		req := httptest.NewRequest("GET", "/", nil)
		req.Host = "example.com"
		req.Header.Set("Host", "example.com")
		rec := httptest.NewRecorder()

		handler.ServeHTTP(rec, req)

		if matchedRoute != "example.com" {
			t.Errorf("expected matched route to be 'example.com', got '%s'", matchedRoute)
		}
	})

	t.Run("wildcard pattern should match", func(t *testing.T) {
		var matched bool

		hr := RouteHeaders().