35 lines
604 B
Go
35 lines
604 B
Go
package libhttp
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func newMockResponseWriter() mockResponseWriter {
|
|
return mockResponseWriter{
|
|
body: strings.Builder{},
|
|
status: 200,
|
|
header: http.Header{},
|
|
}
|
|
}
|
|
|
|
type mockResponseWriter struct {
|
|
body strings.Builder
|
|
status int
|
|
header http.Header
|
|
}
|
|
|
|
var _ http.ResponseWriter = (*mockResponseWriter)(nil)
|
|
|
|
func (m *mockResponseWriter) Header() http.Header {
|
|
return m.header
|
|
}
|
|
|
|
func (m *mockResponseWriter) Write(bytes []byte) (int, error) {
|
|
return m.body.Write(bytes)
|
|
}
|
|
|
|
func (m *mockResponseWriter) WriteHeader(statusCode int) {
|
|
m.status = statusCode
|
|
}
|