100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
|
package ipx
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRemoteAddr(t *testing.T) {
|
||
|
|
request := &http.Request{
|
||
|
|
Header: map[string][]string{},
|
||
|
|
}
|
||
|
|
testCases := []struct {
|
||
|
|
tag string
|
||
|
|
headers map[string][]string
|
||
|
|
mustPublic bool
|
||
|
|
expected string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
"t1", map[string][]string{
|
||
|
|
"X-Real-IP": {"127.0.0.1"},
|
||
|
|
"X-Forwarded-For": {"127.0.0.1"},
|
||
|
|
}, false, "127.0.0.1",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"t2", map[string][]string{
|
||
|
|
"X-Real-IP": {"127.0.0.1:8080"},
|
||
|
|
"X-Forwarded-For": {"127.0.0.1:8080"},
|
||
|
|
}, false, "127.0.0.1",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"t3", map[string][]string{
|
||
|
|
"X-Real-IP": {"127.0.0.1"},
|
||
|
|
"X-Forwarded-For": {"127.0.0.1"},
|
||
|
|
}, true, "",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"t4", map[string][]string{
|
||
|
|
"X-Real-IP": {"127.0.0.1:8080"},
|
||
|
|
"X-Forwarded-For": {"127.0.0.1:8080"},
|
||
|
|
}, true, "",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"t5", map[string][]string{
|
||
|
|
"X-Real-IP": {"::1"},
|
||
|
|
"X-Forwarded-For": {"::1"},
|
||
|
|
}, true, "",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, testCase := range testCases {
|
||
|
|
request.Header = testCase.headers
|
||
|
|
addr := RemoteAddr(request, testCase.mustPublic)
|
||
|
|
assert.Equal(t, testCase.expected, addr, testCase.tag)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocalAddr(t *testing.T) {
|
||
|
|
ip := LocalAddr()
|
||
|
|
if ip == "" {
|
||
|
|
t.Error("LocalAddr() return empty value")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIsPrivate(t *testing.T) {
|
||
|
|
testCases := []struct {
|
||
|
|
tag string
|
||
|
|
ip string
|
||
|
|
expected bool
|
||
|
|
hasError bool
|
||
|
|
}{
|
||
|
|
{"t1", "127.0.0.1", true, false},
|
||
|
|
{"t2", "::1", true, false},
|
||
|
|
{"t3", "xxx", false, true},
|
||
|
|
}
|
||
|
|
for _, testCase := range testCases {
|
||
|
|
v, err := IsPrivate(testCase.ip)
|
||
|
|
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||
|
|
assert.Equal(t, testCase.hasError, err != nil, testCase.tag+" error")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIsPublic(t *testing.T) {
|
||
|
|
testCases := []struct {
|
||
|
|
tag string
|
||
|
|
ip string
|
||
|
|
expected bool
|
||
|
|
hasError bool
|
||
|
|
}{
|
||
|
|
{"t1", "127.0.0.1", false, false},
|
||
|
|
{"t2", "::1", false, false},
|
||
|
|
{"t3", "xxx", false, true},
|
||
|
|
{"t4", "120.228.142.126", true, false},
|
||
|
|
}
|
||
|
|
for _, testCase := range testCases {
|
||
|
|
v, err := IsPublic(testCase.ip)
|
||
|
|
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||
|
|
assert.Equal(t, testCase.hasError, err != nil, testCase.tag+" error")
|
||
|
|
}
|
||
|
|
}
|