func hubMaintainer(logger lager.Logger, desiredHub, actualHub events.Hub) ifrit.RunFunc {
return func(signals <-chan os.Signal, ready chan<- struct{}) error {
logger := logger.Session("hub-maintainer")
close(ready)
logger.Info("started")
defer logger.Info("finished")
<-signals
err := desiredHub.Close()
if err != nil {
logger.Error("error-closing-desired-hub", err)
}
err = actualHub.Close()
if err != nil {
logger.Error("error-closing-actual-hub", err)
}
return nil
}
}
Expect(err).NotTo(HaveOccurred())
// emit event so hub sees closed source and drops subscription
hub.Emit(eventfakes.FakeEvent{})
})
It("calls the callback with a zero count", func() {
Eventually(counts).Should(Receive(BeZero()))
})
})
Context("when the hub is closed", func() {
BeforeEach(func() {
Eventually(counts).Should(Receive())
err := hub.Close()
Expect(err).NotTo(HaveOccurred())
})
It("calls the callback with a zero count", func() {
Eventually(counts).Should(Receive(BeZero()))
})
})
})
})
It("fans-out events emitted to it to all subscribers", func() {
source1, err := hub.Subscribe()
Expect(err).NotTo(HaveOccurred())
source2, err := hub.Subscribe()
Expect(err).NotTo(HaveOccurred())
handler *handlers.EventHandler
eventStreamDone chan struct{}
server *httptest.Server
)
BeforeEach(func() {
logger = lagertest.NewTestLogger("test")
desiredHub = events.NewHub()
actualHub = events.NewHub()
handler = handlers.NewEventHandler(desiredHub, actualHub)
eventStreamDone = make(chan struct{})
})
AfterEach(func() {
desiredHub.Close()
actualHub.Close()
server.Close()
})
var ItStreamsEventsFromHub = func(hubRef *events.Hub) {
Describe("Streaming Events", func() {
var hub events.Hub
var response *http.Response
BeforeEach(func() {
hub = *hubRef
})
JustBeforeEach(func() {
var err error