1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-07-11 22:08:14 +02:00
pomerium/internal/middleware/responsewriter/wrap_writer_test.go
2019-06-10 23:01:30 -04:00

33 lines
781 B
Go

package responsewriter
import (
"net/http/httptest"
"testing"
)
func TestFlushWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
f := &flushWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
f.Flush()
if !f.wroteHeader {
t.Fatal("want Flush to have set wroteHeader=true")
}
}
func TestHttpFancyWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
f := &httpFancyWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
f.Flush()
if !f.wroteHeader {
t.Fatal("want Flush to have set wroteHeader=true")
}
}
func TestHttp2FancyWriterRemembersWroteHeaderWhenFlushed(t *testing.T) {
f := &http2FancyWriter{basicWriter{ResponseWriter: httptest.NewRecorder()}}
f.Flush()
if !f.wroteHeader {
t.Fatal("want Flush to have set wroteHeader=true")
}
}