@@ -17,6 +17,7 @@ package gzip
17
17
import (
18
18
"compress/gzip"
19
19
"fmt"
20
+ "log"
20
21
"net/http"
21
22
"net/http/httptest"
22
23
"testing"
@@ -77,7 +78,9 @@ func TestResponseFilterWriter(t *testing.T) {
77
78
for i , ts := range tests {
78
79
server .Next = httpserver .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) (int , error ) {
79
80
w .Header ().Set ("Content-Length" , fmt .Sprint (len (ts .body )))
80
- w .Write ([]byte (ts .body ))
81
+ if _ , err := w .Write ([]byte (ts .body )); err != nil {
82
+ log .Println ("[ERROR] failed to write response: " , err )
83
+ }
81
84
return 200 , nil
82
85
})
83
86
@@ -86,7 +89,9 @@ func TestResponseFilterWriter(t *testing.T) {
86
89
87
90
w := httptest .NewRecorder ()
88
91
89
- server .ServeHTTP (w , r )
92
+ if _ , err := server .ServeHTTP (w , r ); err != nil {
93
+ log .Println ("[ERROR] unable to serve a gzipped response: " , err )
94
+ }
90
95
91
96
resp := w .Body .String ()
92
97
@@ -109,15 +114,19 @@ func TestResponseGzippedOutput(t *testing.T) {
109
114
110
115
server .Next = httpserver .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) (int , error ) {
111
116
w .Header ().Set ("Content-Encoding" , "gzip" )
112
- w .Write ([]byte ("gzipped" ))
117
+ if _ , err := w .Write ([]byte ("gzipped" )); err != nil {
118
+ log .Println ("[ERROR] failed to write response: " , err )
119
+ }
113
120
return 200 , nil
114
121
})
115
122
116
123
r := urlRequest ("/" )
117
124
r .Header .Set ("Accept-Encoding" , "gzip" )
118
125
119
126
w := httptest .NewRecorder ()
120
- server .ServeHTTP (w , r )
127
+ if _ , err := server .ServeHTTP (w , r ); err != nil {
128
+ log .Println ("[ERROR] unable to serve a gzipped response: " , err )
129
+ }
121
130
resp := w .Body .String ()
122
131
123
132
if resp != "gzipped" {
0 commit comments