You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.4 KiB
91 lines
2.4 KiB
package xhandler
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"context"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTimeoutHandler(t *testing.T) {
|
|
ctx := context.WithValue(context.Background(), contextKey, "value")
|
|
xh := TimeoutHandler(time.Second)(&handler{})
|
|
h := New(ctx, xh)
|
|
w := httptest.NewRecorder()
|
|
r, err := http.NewRequest("GET", "http://example.com/foo", nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
h.ServeHTTP(w, r)
|
|
assert.Equal(t, "value with deadline", w.Body.String())
|
|
}
|
|
|
|
type closeNotifyWriter struct {
|
|
*httptest.ResponseRecorder
|
|
closed bool
|
|
}
|
|
|
|
func (w *closeNotifyWriter) CloseNotify() <-chan bool {
|
|
notify := make(chan bool, 1)
|
|
if w.closed {
|
|
// return an already "closed" notifier
|
|
notify <- true
|
|
}
|
|
return notify
|
|
}
|
|
|
|
// NOTE THIS CURRENTLY FAILS
|
|
// func TestCloseHandlerClientClose(t *testing.T) {
|
|
// ctx := context.WithValue(context.Background(), contextKey, "value")
|
|
// xh := CloseHandler(&handler{})
|
|
// h := New(ctx, xh)
|
|
// w := &closeNotifyWriter{ResponseRecorder: httptest.NewRecorder(), closed: true}
|
|
// r, err := http.NewRequest("GET", "http://example.com/foo", nil)
|
|
// if err != nil {
|
|
// log.Fatal(err)
|
|
// }
|
|
// h.ServeHTTP(w, r)
|
|
// assert.Equal(t, "value canceled", w.Body.String())
|
|
//}
|
|
|
|
func TestCloseHandlerRequestEnds(t *testing.T) {
|
|
ctx := context.WithValue(context.Background(), contextKey, "value")
|
|
xh := CloseHandler(&handler{})
|
|
h := New(ctx, xh)
|
|
w := &closeNotifyWriter{ResponseRecorder: httptest.NewRecorder(), closed: false}
|
|
r, err := http.NewRequest("GET", "http://example.com/foo", nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
h.ServeHTTP(w, r)
|
|
assert.Equal(t, "value", w.Body.String())
|
|
}
|
|
|
|
func TestIf(t *testing.T) {
|
|
trueHandler := HandlerFuncC(func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/true", r.URL.Path)
|
|
})
|
|
falseHandler := HandlerFuncC(func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
|
|
assert.NotEqual(t, "/true", r.URL.Path)
|
|
})
|
|
ctx := context.WithValue(context.Background(), contextKey, "value")
|
|
xh := If(
|
|
func(ctx context.Context, w http.ResponseWriter, r *http.Request) bool {
|
|
return r.URL.Path == "/true"
|
|
},
|
|
func(next HandlerC) HandlerC {
|
|
return trueHandler
|
|
},
|
|
)(falseHandler)
|
|
h := New(ctx, xh)
|
|
r, _ := http.NewRequest("GET", "http://example.com/true", nil)
|
|
h.ServeHTTP(nil, r)
|
|
r, _ = http.NewRequest("GET", "http://example.com/false", nil)
|
|
h.ServeHTTP(nil, r)
|
|
}
|