diff --git a/session_go120.go b/session_go120.go new file mode 100644 index 0000000..3223f25 --- /dev/null +++ b/session_go120.go @@ -0,0 +1,18 @@ +//go:build go1.20 +// +build go1.20 + +package scs + +import ( + "bufio" + "net" + "net/http" +) + +func (sw *sessionResponseWriter) Flush() { + http.NewResponseController(sw.ResponseWriter).Flush() +} + +func (sw *sessionResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { + return http.NewResponseController(sw.ResponseWriter).Hijack() +} diff --git a/session_go120_test.go b/session_go120_test.go new file mode 100644 index 0000000..0468ced --- /dev/null +++ b/session_go120_test.go @@ -0,0 +1,61 @@ +//go:build go1.20 +// +build go1.20 + +package scs + +import ( + "fmt" + "net/http" + "testing" + "time" +) + +func TestFlusher(t *testing.T) { + t.Parallel() + + sessionManager := New() + sessionManager.Lifetime = 500 * time.Millisecond + + mux := http.NewServeMux() + + mux.HandleFunc("/get", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + _, ok := w.(http.Flusher) + + fmt.Fprint(w, ok) + })) + + ts := newTestServer(t, sessionManager.LoadAndSave(mux)) + defer ts.Close() + + ts.execute(t, "/put") + + _, body := ts.execute(t, "/get") + if body != "true" { + t.Errorf("want %q; got %q", "true", body) + } +} + +func TestHijacker(t *testing.T) { + t.Parallel() + + sessionManager := New() + sessionManager.Lifetime = 500 * time.Millisecond + + mux := http.NewServeMux() + + mux.HandleFunc("/get", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + _, ok := w.(http.Hijacker) + + fmt.Fprint(w, ok) + })) + + ts := newTestServer(t, sessionManager.LoadAndSave(mux)) + defer ts.Close() + + ts.execute(t, "/put") + + _, body := ts.execute(t, "/get") + if body != "true" { + t.Errorf("want %q; got %q", "true", body) + } +}