40 lines
891 B
Go
40 lines
891 B
Go
package httpx
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestDecodeJSONRejectsOversizeBody(t *testing.T) {
|
|
t.Parallel()
|
|
payload := `{"password":"` + strings.Repeat("a", 2<<20) + `"}`
|
|
req := httptest.NewRequest(http.MethodPost, "/x", strings.NewReader(payload))
|
|
var dst struct {
|
|
Password string `json:"password"`
|
|
}
|
|
err := DecodeJSON(req, &dst)
|
|
if err == nil {
|
|
t.Fatal("want error for 2 MiB JSON body")
|
|
}
|
|
if !IsBodyTooLarge(err) {
|
|
t.Fatalf("want IsBodyTooLarge, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDecodeJSONAcceptsSmallBody(t *testing.T) {
|
|
t.Parallel()
|
|
req := httptest.NewRequest(http.MethodPost, "/x", bytes.NewReader([]byte(`{"password":"ok"}`)))
|
|
var dst struct {
|
|
Password string `json:"password"`
|
|
}
|
|
if err := DecodeJSON(req, &dst); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if dst.Password != "ok" {
|
|
t.Fatalf("password=%q", dst.Password)
|
|
}
|
|
}
|