return ie.GetHeartbeatEvent().(*events.Heartbeat)
}
var _ = Describe("InstrumentedEmitter", func() {
var fakeEmitter *fake.FakeByteEmitter
var instrumentedEmitter emitter.InstrumentedEmitter
BeforeEach(func() {
fakeEmitter = fake.NewFakeByteEmitter()
instrumentedEmitter, _ = emitter.NewInstrumentedEmitter(fakeEmitter)
})
Describe("Delegators", func() {
It("delegates Close() to the concreteEmitter", func() {
instrumentedEmitter.Close()
Eventually(fakeEmitter.IsClosed).Should(BeTrue())
})
})
Describe("Emit()", func() {
var testData = []byte("hello")
It("calls the concrete emitter", func() {
Expect(fakeEmitter.Messages).To(HaveLen(0))
err := instrumentedEmitter.Emit(testData)
Expect(err).ToNot(HaveOccurred())
Expect(fakeEmitter.Messages).To(HaveLen(1))
Expect(fakeEmitter.Messages[0]).To(Equal(testData))
