diff options
author | Chris <ccbrown112@gmail.com> | 2017-08-08 17:40:46 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-08-08 17:40:46 -0500 |
commit | 6b741c4cea36f54b8f20c4a3e5871f00123db185 (patch) | |
tree | a10eeddd277c4eba48a27431ead816561a20f159 /vendor/github.com/stretchr/testify/http/test_response_writer.go | |
parent | c42510a7f4cf87d1dd7f7339633f06df86f25cce (diff) | |
download | chat-6b741c4cea36f54b8f20c4a3e5871f00123db185.tar.gz chat-6b741c4cea36f54b8f20c4a3e5871f00123db185.tar.bz2 chat-6b741c4cea36f54b8f20c4a3e5871f00123db185.zip |
testify (#7116)
Diffstat (limited to 'vendor/github.com/stretchr/testify/http/test_response_writer.go')
-rw-r--r-- | vendor/github.com/stretchr/testify/http/test_response_writer.go | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/vendor/github.com/stretchr/testify/http/test_response_writer.go b/vendor/github.com/stretchr/testify/http/test_response_writer.go new file mode 100644 index 000000000..5c3f813fa --- /dev/null +++ b/vendor/github.com/stretchr/testify/http/test_response_writer.go @@ -0,0 +1,49 @@ +package http + +import ( + "net/http" +) + +// TestResponseWriter DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead. +type TestResponseWriter struct { + + // StatusCode is the last int written by the call to WriteHeader(int) + StatusCode int + + // 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 DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead. +func (rw *TestResponseWriter) Header() http.Header { + + if rw.header == nil { + rw.header = make(http.Header) + } + + return rw.header +} + +// Write DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead. +func (rw *TestResponseWriter) Write(bytes []byte) (int, error) { + + // assume 200 success if no header has been set + if rw.StatusCode == 0 { + rw.WriteHeader(200) + } + + // add these bytes to the output string + rw.Output = rw.Output + string(bytes) + + // return normal values + return 0, nil + +} + +// WriteHeader DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead. +func (rw *TestResponseWriter) WriteHeader(i int) { + rw.StatusCode = i +} |