mirror of
https://github.com/caddyserver/caddy.git
synced 2025-05-24 02:02:26 -04:00
6fde3632ef
The vendor/ folder was created with the help of @FiloSottile's gvt and vendorcheck. Any dependencies of Caddy plugins outside this repo are not vendored. We do not remove any unused, vendored packages because vendorcheck -u only checks using the current build configuration; i.e. packages that may be imported by files toggled by build tags of other systems. CI tests have been updated to ignore the vendor/ folder. When Go 1.9 is released, a few of the go commands should be revised to again use ./... as it will ignore the vendor folder by default.
36 lines
910 B
Go
36 lines
910 B
Go
package limits
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
|
)
|
|
|
|
func TestBodySizeLimit(t *testing.T) {
|
|
var (
|
|
gotContent []byte
|
|
gotError error
|
|
expectContent = "hello"
|
|
)
|
|
l := Limit{
|
|
Next: httpserver.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
gotContent, gotError = ioutil.ReadAll(r.Body)
|
|
return 0, nil
|
|
}),
|
|
BodyLimits: []httpserver.PathLimit{{Path: "/", Limit: int64(len(expectContent))}},
|
|
}
|
|
|
|
r := httptest.NewRequest("GET", "/", strings.NewReader(expectContent+expectContent))
|
|
l.ServeHTTP(httptest.NewRecorder(), r)
|
|
if got := string(gotContent); got != expectContent {
|
|
t.Errorf("expected content[%s], got[%s]", expectContent, got)
|
|
}
|
|
if gotError != httpserver.ErrMaxBytesExceeded {
|
|
t.Errorf("expect error %v, got %v", httpserver.ErrMaxBytesExceeded, gotError)
|
|
}
|
|
}
|