File tree 2 files changed +5
-11
lines changed
2 files changed +5
-11
lines changed Original file line number Diff line number Diff line change 23
23
24
24
limitedReader struct {
25
25
BodyLimitConfig
26
- reader io.ReadCloser
27
- read int64
28
- context echo.Context
26
+ reader io.ReadCloser
27
+ read int64
29
28
}
30
29
)
31
30
@@ -80,7 +79,7 @@ func BodyLimitWithConfig(config BodyLimitConfig) echo.MiddlewareFunc {
80
79
81
80
// Based on content read
82
81
r := pool .Get ().(* limitedReader )
83
- r .Reset (req .Body , c )
82
+ r .Reset (req .Body )
84
83
defer pool .Put (r )
85
84
req .Body = r
86
85
@@ -102,9 +101,8 @@ func (r *limitedReader) Close() error {
102
101
return r .reader .Close ()
103
102
}
104
103
105
- func (r * limitedReader ) Reset (reader io.ReadCloser , context echo. Context ) {
104
+ func (r * limitedReader ) Reset (reader io.ReadCloser ) {
106
105
r .reader = reader
107
- r .context = context
108
106
r .read = 0
109
107
}
110
108
Original file line number Diff line number Diff line change @@ -56,9 +56,6 @@ func TestBodyLimit(t *testing.T) {
56
56
57
57
func TestBodyLimitReader (t * testing.T ) {
58
58
hw := []byte ("Hello, World!" )
59
- e := echo .New ()
60
- req := httptest .NewRequest (http .MethodPost , "/" , bytes .NewReader (hw ))
61
- rec := httptest .NewRecorder ()
62
59
63
60
config := BodyLimitConfig {
64
61
Skipper : DefaultSkipper ,
@@ -68,7 +65,6 @@ func TestBodyLimitReader(t *testing.T) {
68
65
reader := & limitedReader {
69
66
BodyLimitConfig : config ,
70
67
reader : io .NopCloser (bytes .NewReader (hw )),
71
- context : e .NewContext (req , rec ),
72
68
}
73
69
74
70
// read all should return ErrStatusRequestEntityTooLarge
@@ -78,7 +74,7 @@ func TestBodyLimitReader(t *testing.T) {
78
74
79
75
// reset reader and read two bytes must succeed
80
76
bt := make ([]byte , 2 )
81
- reader .Reset (io .NopCloser (bytes .NewReader (hw )), e . NewContext ( req , rec ) )
77
+ reader .Reset (io .NopCloser (bytes .NewReader (hw )))
82
78
n , err := reader .Read (bt )
83
79
assert .Equal (t , 2 , n )
84
80
assert .Equal (t , nil , err )
You can’t perform that action at this time.
0 commit comments