mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-02-02 16:28:39 +08:00
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package httpx
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetRemoteAddr(t *testing.T) {
|
|
host := "8.8.8.8"
|
|
r, err := http.NewRequest(http.MethodGet, "/", strings.NewReader(""))
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(xForwardedFor, host)
|
|
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)
|
|
}
|
|
|
|
func TestGetFormValues_TooManyValues(t *testing.T) {
|
|
form := url.Values{}
|
|
|
|
// Add more values than the limit
|
|
for i := 0; i < maxFormParamCount+10; i++ {
|
|
form.Add("param", fmt.Sprintf("value%d", i))
|
|
}
|
|
|
|
// Create a new request with the form data
|
|
req, err := http.NewRequest("POST", "/test", strings.NewReader(form.Encode()))
|
|
assert.NoError(t, err)
|
|
|
|
// Set the content type for form data
|
|
req.Header.Set(ContentType, "application/x-www-form-urlencoded")
|
|
|
|
_, err = GetFormValues(req)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "too many form values")
|
|
}
|