mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 02:46:30 +02:00
26 lines
729 B
Go
26 lines
729 B
Go
package httputil
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetClientIPAddress(t *testing.T) {
|
|
r1, err := http.NewRequest("GET", "https://example.com", nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "127.0.0.1", GetClientIPAddress(r1))
|
|
|
|
r2, err := http.NewRequest("GET", "https://example.com", nil)
|
|
require.NoError(t, err)
|
|
r2.RemoteAddr = "127.0.0.2:1234"
|
|
assert.Equal(t, "127.0.0.2", GetClientIPAddress(r2))
|
|
|
|
r3, err := http.NewRequest("GET", "https://example.com", nil)
|
|
require.NoError(t, err)
|
|
r3.RemoteAddr = "127.0.0.3:1234"
|
|
r3.Header.Set("X-Envoy-External-Address", "127.0.0.3")
|
|
assert.Equal(t, "127.0.0.3", GetClientIPAddress(r3))
|
|
}
|