mirror of https://github.com/stretchr/testify.git
added comments for the test response writer
parent
0b90ca8b73
commit
870c90a638
|
@ -4,12 +4,21 @@ import (
|
||||||
"net/http"
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// TestResponseWriter is a http.ResponseWriter object that keeps track of all activity
|
||||||
|
// allowing you to make assertions about how it was used.
|
||||||
type TestResponseWriter struct {
|
type TestResponseWriter struct {
|
||||||
|
|
||||||
|
// WrittenHeaderInt is the last int written by the call to WriteHeader(int)
|
||||||
WrittenHeaderInt int
|
WrittenHeaderInt int
|
||||||
Output string
|
|
||||||
header http.Header
|
// Output is a string containing the written bytes using the Write([]byte) func.
|
||||||
|
Output string
|
||||||
|
|
||||||
|
// header is the internal storage of the http.Header object
|
||||||
|
header http.Header
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Header gets the http.Header describing the headers that were set in this response.
|
||||||
func (rw *TestResponseWriter) Header() http.Header {
|
func (rw *TestResponseWriter) Header() http.Header {
|
||||||
|
|
||||||
if rw.header == nil {
|
if rw.header == nil {
|
||||||
|
@ -18,6 +27,8 @@ func (rw *TestResponseWriter) Header() http.Header {
|
||||||
|
|
||||||
return rw.header
|
return rw.header
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Write writes the specified bytes to Output.
|
||||||
func (rw *TestResponseWriter) Write(bytes []byte) (int, error) {
|
func (rw *TestResponseWriter) Write(bytes []byte) (int, error) {
|
||||||
|
|
||||||
// add these bytes to the output string
|
// add these bytes to the output string
|
||||||
|
@ -27,6 +38,8 @@ func (rw *TestResponseWriter) Write(bytes []byte) (int, error) {
|
||||||
return 0, nil
|
return 0, nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WriteHeader stores the HTTP status code in the WrittenHeaderInt.
|
||||||
func (rw *TestResponseWriter) WriteHeader(i int) {
|
func (rw *TestResponseWriter) WriteHeader(i int) {
|
||||||
rw.WrittenHeaderInt = i
|
rw.WrittenHeaderInt = i
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue