github.com/lingyao2333/mo-zero@v1.4.1/rest/httpx/util_test.go (about)

     1  package httpx
     2  
     3  import (
     4  	"net/http"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestGetRemoteAddr(t *testing.T) {
    12  	host := "8.8.8.8"
    13  	r, err := http.NewRequest(http.MethodGet, "/", strings.NewReader(""))
    14  	assert.Nil(t, err)
    15  
    16  	r.Header.Set(xForwardedFor, host)
    17  	assert.Equal(t, host, GetRemoteAddr(r))
    18  }
    19  
    20  func TestGetRemoteAddrNoHeader(t *testing.T) {
    21  	r, err := http.NewRequest(http.MethodGet, "/", strings.NewReader(""))
    22  	assert.Nil(t, err)
    23  
    24  	assert.True(t, len(GetRemoteAddr(r)) == 0)
    25  }