1package mux
2
3import (
4	"context"
5	"net/http"
6	"testing"
7	"time"
8)
9
10func TestNativeContextMiddleware(t *testing.T) {
11	withTimeout := func(h http.Handler) http.Handler {
12		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
13			ctx, cancel := context.WithTimeout(r.Context(), time.Minute)
14			defer cancel()
15			h.ServeHTTP(w, r.WithContext(ctx))
16		})
17	}
18
19	r := NewRouter()
20	r.Handle("/path/{foo}", withTimeout(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
21		vars := Vars(r)
22		if vars["foo"] != "bar" {
23			t.Fatal("Expected foo var to be set")
24		}
25	})))
26
27	rec := NewRecorder()
28	req := newRequest("GET", "/path/bar")
29	r.ServeHTTP(rec, req)
30}
31