123456789101112131415161718192021222324252627 |
- package utils
- import (
- "net/http/httptest"
- "testing"
- )
- func TestGetIPAddressFromRequest(t *testing.T) {
- expectedResult := "203.0.113.195"
- request := httptest.NewRequest("GET", "/", nil)
- request.RemoteAddr = "203.0.113.195:41237"
- //First Test without X-FORWARDED-FOR header
- result := GetIPAddressFromRequest(request)
- if result != expectedResult {
- t.Errorf("Remote address only test failed: Expected %s, got %s", expectedResult, result)
- }
- //Test with X-FORWARDED-FOR header
- request.Header.Set("X-FORWARDED-FOR header", "203.0.113.195:41237, 198.51.100.100:38523, 140.248.67.176:12345")
- result = GetIPAddressFromRequest(request)
- if result != expectedResult {
- t.Errorf("X-FORWARD-FOR header test failed: Expected %s, got %s", expectedResult, result)
- }
- }
|