Skip to content

Commit

Permalink
More unit tests for ResponseWriter
Browse files Browse the repository at this point in the history
manucorporat committed Apr 8, 2015
1 parent a28104f commit c61c547
Showing 2 changed files with 6 additions and 1 deletion.
4 changes: 3 additions & 1 deletion response_writer.go
Original file line number Diff line number Diff line change
@@ -79,7 +79,9 @@ func (w *responseWriter) Written() bool {

// Implements the http.Hijacker interface
func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
w.size = 0 // this prevents Gin to write the HTTP headers
if w.size < 0 {
w.size = 0
}
return w.ResponseWriter.(http.Hijacker).Hijack()
}

3 changes: 3 additions & 0 deletions response_writer_test.go
Original file line number Diff line number Diff line change
@@ -15,6 +15,9 @@ import (
var _ ResponseWriter = &responseWriter{}
var _ http.ResponseWriter = &responseWriter{}
var _ http.ResponseWriter = ResponseWriter(&responseWriter{})
var _ http.Hijacker = ResponseWriter(&responseWriter{})
var _ http.Flusher = ResponseWriter(&responseWriter{})
var _ http.CloseNotifier = ResponseWriter(&responseWriter{})

func init() {
SetMode(TestMode)

0 comments on commit c61c547

Please sign in to comment.