21 lines
362 B
Go
21 lines
362 B
Go
package mock_http
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
)
|
|
|
|
type Recorder struct {
|
|
*httptest.ResponseRecorder
|
|
err error
|
|
}
|
|
|
|
func NewRecorder(err error) *Recorder {
|
|
return &Recorder{ResponseRecorder: httptest.NewRecorder(), err: err}
|
|
}
|
|
|
|
func (r *Recorder) Write(buf []byte) (int, error) {
|
|
if r.err != nil {
|
|
return 0, r.err
|
|
}
|
|
return r.ResponseRecorder.Write(buf)
|
|
}
|