|
@@ -16,3 +16,10 @@ func TestGetRemoteAddr(t *testing.T) {
|
|
r.Header.Set(xForwardedFor, host)
|
|
r.Header.Set(xForwardedFor, host)
|
|
assert.Equal(t, host, GetRemoteAddr(r))
|
|
assert.Equal(t, host, GetRemoteAddr(r))
|
|
}
|
|
}
|
|
|
|
+
|
|
|
|
+func TestGetRemoteAddrNoHeader(t *testing.T) {
|
|
|
|
+ r, err := http.NewRequest(http.MethodGet, "/", strings.NewReader(""))
|
|
|
|
+ assert.Nil(t, err)
|
|
|
|
+
|
|
|
|
+ assert.True(t, len(GetRemoteAddr(r)) == 0)
|
|
|
|
+}
|