aboutsummaryrefslogtreecommitdiff
path: root/src/vendor/github.com/rs/xhandler/middleware_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/vendor/github.com/rs/xhandler/middleware_test.go')
-rw-r--r--src/vendor/github.com/rs/xhandler/middleware_test.go88
1 files changed, 0 insertions, 88 deletions
diff --git a/src/vendor/github.com/rs/xhandler/middleware_test.go b/src/vendor/github.com/rs/xhandler/middleware_test.go
deleted file mode 100644
index 51306e3..0000000
--- a/src/vendor/github.com/rs/xhandler/middleware_test.go
+++ /dev/null
@@ -1,88 +0,0 @@
-package xhandler
-
-import (
- "log"
- "net/http"
- "net/http/httptest"
- "testing"
- "time"
-
- "github.com/stretchr/testify/assert"
- "golang.org/x/net/context"
-)
-
-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
-}
-
-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)
-}