| 66 | } |
| 67 | |
| 68 | func TestXForwardForXRealIPPrecedence(t *testing.T) { |
| 69 | req, _ := http.NewRequest("GET", "/", nil) |
| 70 | req.Header.Add("X-Forwarded-For", "0.0.0.0") |
| 71 | req.Header.Add("X-Real-IP", "100.100.100.100") |
| 72 | w := httptest.NewRecorder() |
| 73 | |
| 74 | r := chi.NewRouter() |
| 75 | r.Use(RealIP) |
| 76 | |
| 77 | realIP := "" |
| 78 | r.Get("/", func(w http.ResponseWriter, r *http.Request) { |
| 79 | realIP = r.RemoteAddr |
| 80 | w.Write([]byte("Hello World")) |
| 81 | }) |
| 82 | r.ServeHTTP(w, req) |
| 83 | |
| 84 | if w.Code != 200 { |
| 85 | t.Fatal("Response Code should be 200") |
| 86 | } |
| 87 | |
| 88 | if realIP != "100.100.100.100" { |
| 89 | t.Fatal("Test get real IP precedence error.") |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | func TestInvalidIP(t *testing.T) { |
| 94 | req, _ := http.NewRequest("GET", "/", nil) |