28 lines
774 B
Go
28 lines
774 B
Go
|
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)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
