simplify code with http.Flusher type conversion (#325)

* simplify code with http.Flusher type conversion

* simplify code with http.Flusher type conversion, better version
This commit is contained in:
Kevin Wan
2020-12-29 10:25:55 +08:00
committed by GitHub
parent 0bd2a0656c
commit e97e1f10db
8 changed files with 29 additions and 32 deletions

View File

@@ -92,9 +92,8 @@ func TestCryptionHandlerFlush(t *testing.T) {
req := httptest.NewRequest(http.MethodGet, "/any", nil)
handler := CryptionHandler(aesKey)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(respText))
flusher, ok := w.(http.Flusher)
assert.Equal(t, ok, true)
assert.True(t, ok)
flusher.Flush()
}))
recorder := httptest.NewRecorder()