Method WriteHeader

recorder_test.go:149–175

WriteHeader implements http.ResponseWriter.

(code int)

// WriteHeader implements http.ResponseWriter.
func (rw *ResponseRecorder) WriteHeader(code int) {
	if rw.wroteHeader {
		return
	}

	checkWriteHeaderCode(code)

	if rw.ClientTrace != nil && code >= 100 && code < 200 {
		if code == 100 {
			rw.ClientTrace.Got100Continue()
		}
		// treat 101 as a terminal status, see issue 26161
		if code != http.StatusSwitchingProtocols {
			if err := rw.ClientTrace.Got1xxResponse(code, textproto.MIMEHeader(rw.HeaderMap)); err != nil {
				panic(err)
			}
			return
		}
	}

	rw.Code = code
	rw.wroteHeader = true
	if rw.HeaderMap == nil {
		rw.HeaderMap = make(http.Header)
	}
	rw.snapHeader = rw.HeaderMap.Clone()
}

// Flush implements http.Flusher. To test whether Flush was
// called, see rw.Flushed.