net/http testing library doesnt supprots CloseNotify, so don't panic on context.StreamWriter

This commit is contained in:
Gerasimos (Makis) Maropoulos 2017-01-06 15:57:09 +02:00
parent d060a73559
commit 7df9b2b437

View File

@ -231,7 +231,11 @@ func (w *responseWriter) Flush() {
// is a problem, use HTTP/2 or only use CloseNotify on methods // is a problem, use HTTP/2 or only use CloseNotify on methods
// such as POST. // such as POST.
func (w *responseWriter) CloseNotify() <-chan bool { func (w *responseWriter) CloseNotify() <-chan bool {
return w.ResponseWriter.(http.CloseNotifier).CloseNotify() if notifier, supportsCloseNotify := w.ResponseWriter.(http.CloseNotifier); supportsCloseNotify {
return notifier.CloseNotify()
}
ch := make(chan bool, 1)
return ch
} }
// clone returns a clone of this response writer // clone returns a clone of this response writer