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))