瀏覽代碼

test: add more tests (#1150)

Kevin Wan 3 年之前
父節點
當前提交
5636bf4955
共有 2 個文件被更改,包括 8 次插入0 次删除
  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)
+}