Fix RawPath handling in addPrefix

middlewares/addPrefix.go

@@ -12,6 +12,9 @@ type AddPrefix struct {
 
 func (s *AddPrefix) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	r.URL.Path = s.Prefix + r.URL.Path
+	if r.URL.RawPath != "" {
+		r.URL.RawPath = s.Prefix + r.URL.RawPath
+	}
 	r.RequestURI = r.URL.RequestURI()
 	s.Handler.ServeHTTP(w, r)
 }

middlewares/addPrefix_test.go

@@ -9,21 +9,56 @@ import (
 )
 
 func TestAddPrefix(t *testing.T) {
-
+	tests := []struct {
-	path := "/bar"
+		desc            string
-	prefix := "/foo"
+		prefix          string
-
+		path            string
-	var expectedPath string
+		expectedPath    string
-	handler := &AddPrefix{
+		expectedRawPath string
-		Prefix: prefix,
+	}{
-		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		{
-			expectedPath = r.URL.Path
+			desc:         "regular path",
-		}),
+			prefix:       "/a",
+			path:         "/b",
+			expectedPath: "/a/b",
+		},
+		{
+			desc:            "raw path is supported",
+			prefix:          "/a",
+			path:            "/b%2Fc",
+			expectedPath:    "/a/b/c",
+			expectedRawPath: "/a/b%2Fc",
+		},
 	}
 
-	req := testhelpers.MustNewRequest(http.MethodGet, "http://localhost"+path, nil)
+	for _, test := range tests {
+		test := test
+		t.Run(test.desc, func(t *testing.T) {
+			t.Parallel()
 
-	handler.ServeHTTP(nil, req)
+			var actualPath, actualRawPath, requestURI string
+			handler := &AddPrefix{
+				Prefix: test.prefix,
+				Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+					actualPath = r.URL.Path
+					actualRawPath = r.URL.RawPath
+					requestURI = r.RequestURI
+				}),
+			}
 
-	assert.Equal(t, expectedPath, "/foo/bar", "Unexpected path.")
+			req := testhelpers.MustNewRequest(http.MethodGet, "http://localhost"+test.path, nil)
+
+			handler.ServeHTTP(nil, req)
+
+			assert.Equal(t, test.expectedPath, actualPath, "Unexpected path.")
+			assert.Equal(t, test.expectedRawPath, actualRawPath, "Unexpected raw path.")
+
+			expectedURI := test.expectedPath
+			if test.expectedRawPath != "" {
+				// go HTTP uses the raw path when existent in the RequestURI
+				expectedURI = test.expectedRawPath
+			}
+			assert.Equal(t, expectedURI, requestURI, "Unexpected request URI.")
+		})
+	}
 }