mirror of
https://github.com/labstack/echo.git
synced 2024-11-24 08:22:21 +02:00
141 lines
3.1 KiB
Go
141 lines
3.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"testing"
|
|
|
|
"github.com/labstack/echo"
|
|
"github.com/labstack/echo/test"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type closeNotifyingRecorder struct {
|
|
*test.ResponseRecorder
|
|
closed chan bool
|
|
}
|
|
|
|
func newCloseNotifyingRecorder() *closeNotifyingRecorder {
|
|
return &closeNotifyingRecorder{
|
|
test.NewResponseRecorder(),
|
|
make(chan bool, 1),
|
|
}
|
|
}
|
|
|
|
func (c *closeNotifyingRecorder) close() {
|
|
c.closed <- true
|
|
}
|
|
|
|
func (c *closeNotifyingRecorder) CloseNotify() <-chan bool {
|
|
return c.closed
|
|
}
|
|
|
|
func TestGzip(t *testing.T) {
|
|
e := echo.New()
|
|
req := test.NewRequest(echo.GET, "/", nil)
|
|
rec := test.NewResponseRecorder()
|
|
c := echo.NewContext(req, rec, e)
|
|
// Skip if no Accept-Encoding header
|
|
h := NewGzip().Handle(echo.HandlerFunc(func(c echo.Context) error {
|
|
c.Response().Write([]byte("test")) // For Content-Type sniffing
|
|
return nil
|
|
}))
|
|
h.Handle(c)
|
|
assert.Equal(t, "test", rec.Body.String())
|
|
|
|
req = test.NewRequest(echo.GET, "/", nil)
|
|
req.Header().Set(echo.AcceptEncoding, "gzip")
|
|
rec = test.NewResponseRecorder()
|
|
c = echo.NewContext(req, rec, e)
|
|
|
|
// Gzip
|
|
h.Handle(c)
|
|
// assert.Equal(t, http.StatusOK, rec.Status())
|
|
assert.Equal(t, "gzip", rec.Header().Get(echo.ContentEncoding))
|
|
assert.Contains(t, rec.Header().Get(echo.ContentType), echo.TextPlain)
|
|
r, err := gzip.NewReader(rec.Body)
|
|
defer r.Close()
|
|
if assert.NoError(t, err) {
|
|
buf := new(bytes.Buffer)
|
|
buf.ReadFrom(r)
|
|
assert.Equal(t, "test", buf.String())
|
|
}
|
|
}
|
|
|
|
// func TestGzipFlush(t *testing.T) {
|
|
// res := test.NewResponseRecorder()
|
|
// buf := new(bytes.Buffer)
|
|
// w := gzip.NewWriter(buf)
|
|
// gw := gzipWriter{Writer: w, ResponseWriter: res}
|
|
//
|
|
// n0 := buf.Len()
|
|
// if n0 != 0 {
|
|
// t.Fatalf("buffer size = %d before writes; want 0", n0)
|
|
// }
|
|
//
|
|
// if err := gw.Flush(); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
//
|
|
// n1 := buf.Len()
|
|
// if n1 == 0 {
|
|
// t.Fatal("no data after first flush")
|
|
// }
|
|
//
|
|
// gw.Write([]byte("x"))
|
|
//
|
|
// n2 := buf.Len()
|
|
// if n1 != n2 {
|
|
// t.Fatalf("after writing a single byte, size changed from %d to %d; want no change", n1, n2)
|
|
// }
|
|
//
|
|
// if err := gw.Flush(); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
//
|
|
// n3 := buf.Len()
|
|
// if n2 == n3 {
|
|
// t.Fatal("Flush didn't flush any data")
|
|
// }
|
|
// }
|
|
|
|
// func TestGzipCloseNotify(t *testing.T) {
|
|
// rec := newCloseNotifyingRecorder()
|
|
// buf := new(bytes.Buffer)
|
|
// w := gzip.NewWriter(buf)
|
|
// gw := gzipWriter{Writer: w, ResponseWriter: rec}
|
|
// closed := false
|
|
// notifier := gw.CloseNotify()
|
|
// rec.close()
|
|
//
|
|
// select {
|
|
// case <-notifier:
|
|
// closed = true
|
|
// case <-time.After(time.Second):
|
|
// }
|
|
//
|
|
// assert.Equal(t, closed, true)
|
|
// }
|
|
//
|
|
// func BenchmarkGzip(b *testing.B) {
|
|
// b.StopTimer()
|
|
// b.ReportAllocs()
|
|
//
|
|
// h := func(c echo.Context) error {
|
|
// c.Response().Write([]byte("test")) // For Content-Type sniffing
|
|
// return nil
|
|
// }
|
|
// req, _ := http.NewRequest(echo.GET, "/", nil)
|
|
// req.Header().Set(echo.AcceptEncoding, "gzip")
|
|
//
|
|
// b.StartTimer()
|
|
//
|
|
// for i := 0; i < b.N; i++ {
|
|
// e := echo.New()
|
|
// res := test.NewResponseRecorder()
|
|
// c := echo.NewContext(req, res, e)
|
|
// Gzip()(h)(c)
|
|
// }
|
|
//
|
|
// }
|