package main import ( "net/http" "net/http/httptest" "regexp" "strings" "sync" "github.com/go-check/check" ) type handlerFunc func(w http.ResponseWriter, r *http.Request) type testRegistry struct { server *httptest.Server hostport string handlers map[string]handlerFunc mu sync.Mutex } func (tr *testRegistry) registerHandler(path string, h handlerFunc) { tr.mu.Lock() defer tr.mu.Unlock() tr.handlers[path] = h } func newTestRegistry(c *check.C) (*testRegistry, error) { testReg := &testRegistry{handlers: make(map[string]handlerFunc)} ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { url := r.URL.String() var matched bool var err error for re, function := range testReg.handlers { matched, err = regexp.MatchString(re, url) if err != nil { c.Fatalf("Error with handler regexp") return } if matched { function(w, r) break } } if !matched { c.Fatal("Unable to match", url, "with regexp") } })) testReg.server = ts testReg.hostport = strings.Replace(ts.URL, "http://", "", 1) return testReg, nil }