1 files changed, 7 insertions, 38 deletions

diff --git a/vendor/github.com/prometheus/common/route/route_test.go b/vendor/github.com/prometheus/common/route/route_test.go
index e7b1cba33..a9bb20996 100644
--- a/vendor/github.com/prometheus/common/route/route_test.go
+++ b/vendor/github.com/prometheus/common/route/route_test.go
@@ -1,16 +1,13 @@
 package route
 
 import (
-	"fmt"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-
-	"golang.org/x/net/context"
 )
 
 func TestRedirect(t *testing.T) {
-	router := New(nil).WithPrefix("/test/prefix")
+	router := New().WithPrefix("/test/prefix")
 	w := httptest.NewRecorder()
 	r, err := http.NewRequest("GET", "http://localhost:9090/foo", nil)
 	if err != nil {
@@ -29,47 +26,19 @@ func TestRedirect(t *testing.T) {
 	}
 }
 
-func TestContextFunc(t *testing.T) {
-	router := New(func(r *http.Request) (context.Context, error) {
-		return context.WithValue(context.Background(), "testkey", "testvalue"), nil
-	})
-
-	router.Get("/test", func(w http.ResponseWriter, r *http.Request) {
-		want := "testvalue"
-		got := Context(r).Value("testkey")
+func TestContext(t *testing.T) {
+	router := New()
+	router.Get("/test/:foo/", func(w http.ResponseWriter, r *http.Request) {
+		want := "bar"
+		got := Param(r.Context(), "foo")
 		if want != got {
 			t.Fatalf("Unexpected context value: want %q, got %q", want, got)
 		}
 	})
 
-	r, err := http.NewRequest("GET", "http://localhost:9090/test", nil)
+	r, err := http.NewRequest("GET", "http://localhost:9090/test/bar/", nil)
 	if err != nil {
 		t.Fatalf("Error building test request: %s", err)
 	}
 	router.ServeHTTP(nil, r)
 }
-
-func TestContextFnError(t *testing.T) {
-	router := New(func(r *http.Request) (context.Context, error) {
-		return context.Background(), fmt.Errorf("test error")
-	})
-
-	router.Get("/test", func(w http.ResponseWriter, r *http.Request) {})
-
-	r, err := http.NewRequest("GET", "http://localhost:9090/test", nil)
-	if err != nil {
-		t.Fatalf("Error building test request: %s", err)
-	}
-	w := httptest.NewRecorder()
-	router.ServeHTTP(w, r)
-
-	if w.Code != http.StatusBadRequest {
-		t.Fatalf("Unexpected response status: got %q, want %q", w.Code, http.StatusBadRequest)
-	}
-
-	want := "Error creating request context: test error\n"
-	got := w.Body.String()
-	if want != got {
-		t.Fatalf("Unexpected response body: got %q, want %q", got, want)
-	}
-}