context_test.go

package tigertonic

import (
	"net/http"
	"testing"
)

func TestContext(t *testing.T) {
	ch := make(chan bool)
	mux := NewTrieServeMux()
	mux.HandleFunc("GET", "/1", func(w http.ResponseWriter, r *http.Request) {
		if 1 != len(contexts) {
			t.Error(contexts)
		}
		<-ch
		w.WriteHeader(http.StatusNoContent)
	})
	mux.HandleFunc("GET", "/2", func(w http.ResponseWriter, r *http.Request) {
		if 2 != len(contexts) {
			t.Error(contexts)
		}
		ch <- true
		w.WriteHeader(http.StatusNoContent)
	})
	handler := WithContext(mux, testContext{})
	go func() {
		w := &testResponseWriter{}
		r, _ := http.NewRequest("GET", "http://example.com/1", nil)
		handler.ServeHTTP(w, r)
		w2 := &testResponseWriter{}
		r2, _ := http.NewRequest("GET", "http://example.com/2", nil)
		handler.ServeHTTP(w2, r2)
	}()
}

type testContext struct {
	Foo string
	Bar int
}