From 9aa77ca769c689f3d1bf4ef8051bcd377f6fcaab Mon Sep 17 00:00:00 2001 From: Mat Ryer Date: Wed, 5 Jun 2013 12:07:38 -0600 Subject: [PATCH] Fixes #15 - writes 200 header if no specific header has been written --- http/test_response_writer.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/http/test_response_writer.go b/http/test_response_writer.go index 7aee1d6..2ff8f24 100644 --- a/http/test_response_writer.go +++ b/http/test_response_writer.go @@ -31,6 +31,11 @@ func (rw *TestResponseWriter) Header() http.Header { // Write writes the specified bytes to Output. func (rw *TestResponseWriter) Write(bytes []byte) (int, error) { + // assume 200 success if no header has been set + if rw.WrittenHeaderInt == 0 { + rw.WriteHeader(200) + } + // add these bytes to the output string rw.Output = rw.Output + string(bytes)