mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-01 19:36:32 +02:00
33 lines
781 B
Go
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")
|
|
}
|
|
}
|