Bladeren bron

test: add more tests (#1150)

Kevin Wan 3 jaren geleden
bovenliggende
commit
5636bf4955
2 gewijzigde bestanden met toevoegingen van 8 en 0 verwijderingen
  1. 1 0
      rest/httpx/util.go
  2. 7 0
      rest/httpx/util_test.go

+ 1 - 0
rest/httpx/util.go

@@ -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 - 0
rest/httpx/util_test.go

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