You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package httphandler
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMaxBytesHandler(t *testing.T) {
|
|
maxb := MaxBytesHandler(10)
|
|
handler := maxb(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {}))
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "http://localhost",
|
|
bytes.NewBufferString("123456789012345"))
|
|
resp := httptest.NewRecorder()
|
|
handler.ServeHTTP(resp, req)
|
|
assert.Equal(t, http.StatusRequestEntityTooLarge, resp.Code)
|
|
|
|
req = httptest.NewRequest(http.MethodPost, "http://localhost", bytes.NewBufferString("12345"))
|
|
resp = httptest.NewRecorder()
|
|
handler.ServeHTTP(resp, req)
|
|
assert.Equal(t, http.StatusOK, resp.Code)
|
|
}
|
|
|
|
func TestMaxBytesHandlerNoLimit(t *testing.T) {
|
|
maxb := MaxBytesHandler(-1)
|
|
handler := maxb(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {}))
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "http://localhost",
|
|
bytes.NewBufferString("123456789012345"))
|
|
resp := httptest.NewRecorder()
|
|
handler.ServeHTTP(resp, req)
|
|
assert.Equal(t, http.StatusOK, resp.Code)
|
|
}
|