feat: CSP Violation ( 感叹号图标被拦截 )

server/internal/handler/router_test.go

@@ -0,0 +1,59 @@
+package handler_test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+)
+
+func TestRouterStripSlashes(t *testing.T) {
+	r := chi.NewRouter()
+	r.Use(middleware.StripSlashes)
+
+	r.Route("/api/v1", func(r chi.Router) {
+		r.Post("/auth/login", func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusOK)
+			w.Write([]byte(`{"code":200}`))
+		})
+	})
+
+	tests := []struct {
+		name           string
+		method         string
+		url            string
+		expectedStatus int
+	}{
+		{
+			name:           "Login without trailing slash",
+			method:         "POST",
+			url:            "/api/v1/auth/login",
+			expectedStatus: http.StatusOK,
+		},
+		{
+			name:           "Login with trailing slash",
+			method:         "POST",
+			url:            "/api/v1/auth/login/",
+			expectedStatus: http.StatusOK,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			req, _ := http.NewRequest(tt.method, tt.url, nil)
+			rr := httptest.NewRecorder()
+			r.ServeHTTP(rr, req)
+
+			if rr.Code != tt.expectedStatus {
+				t.Errorf("expected status %v, got %v for %s", tt.expectedStatus, rr.Code, tt.url)
+			}
+			
+			// Check if it redirected (it shouldn't with StripSlashes middleware correctly placed)
+			if rr.Code == http.StatusPermanentRedirect || rr.Code == http.StatusMovedPermanently {
+				t.Errorf("got redirect %v for %s", rr.Code, tt.url)
+			}
+		})
+	}
+}