MockRouter.go

type MockRouter struct {
    mu sync.Mutex
    r *mux.Router
}

func (mr *MockRouter) Swap(r *mux.Router) {
    mr.mu.Lock()
    mr.r = r
    mr.mu.Unlock()
}

func (mr *MockRouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    mr.r.ServeHTTP(w, r)
}