vendor/github.com/stretchr/testify/http/test_response_writer.go

   1 package http
   2
   3 import (
   4         "net/http"
   5 )
   6
   7 // TestResponseWriter DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead.
   8 type TestResponseWriter struct {
   9
  10         // StatusCode is the last int written by the call to WriteHeader(int)
  11         StatusCode int
  12
  13         // Output is a string containing the written bytes using the Write([]byte) func.
  14         Output string
  15
  16         // header is the internal storage of the http.Header object
  17         header http.Header
  18 }
  19
  20 // Header DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead.
  21 func (rw *TestResponseWriter) Header() http.Header {
  22
  23         if rw.header == nil {
  24                 rw.header = make(http.Header)
  25         }
  26
  27         return rw.header
  28 }
  29
  30 // Write DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead.
  31 func (rw *TestResponseWriter) Write(bytes []byte) (int, error) {
  32
  33         // assume 200 success if no header has been set
  34         if rw.StatusCode == 0 {
  35                 rw.WriteHeader(200)
  36         }
  37
  38         // add these bytes to the output string
  39         rw.Output = rw.Output + string(bytes)
  40
  41         // return normal values
  42         return 0, nil
  43
  44 }
  45
  46 // WriteHeader DEPRECATED: We recommend you use http://golang.org/pkg/net/http/httptest instead.
  47 func (rw *TestResponseWriter) WriteHeader(i int) {
  48         rw.StatusCode = i
  49 }