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)
}