zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: add more tests (#1150)

Browse Source
This commit is contained in:
Kevin Wan
2021-10-20 17:50:01 +08:00
committed by GitHub
parent a944a7fd7e
commit 5636bf4955
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
rest/httpx/util.go
View File

@@ -10,5 +10,6 @@ func GetRemoteAddr(r *http.Request) string {
if len(v) > 0 { if len(v) > 0 {
return v return v
} }
return r.RemoteAddr return r.RemoteAddr
} }

7
rest/httpx/util_test.go
View File

@@ -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)
}