func closeHub(logger lager.Logger, hub event.Hub) ifrit.Runner {
return ifrit.RunFunc(func(signals <-chan os.Signal, ready chan<- struct{}) error {
logger.Info("starting")
defer logger.Info("finished")
close(ready)
logger.Info("started")
<-signals
logger.Info("shutting-down")
hub.Close()
return nil
})
}
handler *handlers.EventStreamHandler
server *httptest.Server
)
BeforeEach(func() {
hub = event.NewHub()
logger = lager.NewLogger("test")
logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.DEBUG))
handler = handlers.NewEventStreamHandler(hub, logger)
})
AfterEach(func() {
hub.Close()
if server != nil {
server.Close()
}
})
Describe("EventStream", func() {
var (
response *http.Response
eventStreamDone chan struct{}
)
BeforeEach(func() {
eventStreamDone = make(chan struct{})
server = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {