socket-server-helper.go

func (s *someSuite) newServer(handler http.Handler) (*httptest.Server, func()) {
	done := make(chan struct{})
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		s.NotPanics(func() { handler.ServeHTTP(w, r) })
		close(done)
	}))
	closeFunc := func() {
		server.Close()
		<-done
	}
	return server, closeFunc
}