1 package middleware
2
3 import (
4 "net/http/httptest"
5 "testing"
6 )
7
8 func TestFlushWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
9 f := &flushWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
10 f.Flush()
11
12 if !f.wroteHeader {
13 t.Fatal("want Flush to have set wroteHeader=true")
14 }
15 }
16
17 func TestHttpFancyWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
18 f := &httpFancyWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
19 f.Flush()
20
21 if !f.wroteHeader {
22 t.Fatal("want Flush to have set wroteHeader=true")
23 }
24 }
25
26 func TestHttp2FancyWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
27 f := &http2FancyWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
28 f.Flush()
29
30 if !f.wroteHeader {
31 t.Fatal("want Flush to have set wroteHeader=true")
32 }
33 }
34
View as plain text