"github.com/cloudfoundry/dropsonde/emitter/fake" "github.com/cloudfoundry/dropsonde/events" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) func getHeartbeatEvent(ie emitter.InstrumentedEmitter) *events.Heartbeat { 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() {
import ( "github.com/cloudfoundry/dropsonde/emitter" "github.com/cloudfoundry/dropsonde/emitter/fake" "github.com/cloudfoundry/dropsonde/events" "github.com/cloudfoundry/dropsonde/factories" "github.com/gogo/protobuf/proto" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var _ = Describe("EventEmitter", func() { Describe("Emit", func() { Context("without an origin", func() { It("returns an error", func() { innerEmitter := fake.NewFakeByteEmitter() eventEmitter := emitter.NewEventEmitter(innerEmitter, "") testEvent := factories.NewHeartbeat(1, 2, 3) err := eventEmitter.Emit(testEvent) Expect(err).To(HaveOccurred()) Expect(err.Error()).To(ContainSubstring("Wrap: ")) }) }) It("marshals events and delegates to the inner emitter", func() { innerEmitter := fake.NewFakeByteEmitter() origin := "fake-origin" eventEmitter := emitter.NewEventEmitter(innerEmitter, origin)
"github.com/cloudfoundry/dropsonde/events" "github.com/cloudfoundry/dropsonde/factories" "github.com/gogo/protobuf/proto" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var _ = Describe("HeartbeatResponder", func() { var ( wrappedEmitter *fake.FakeByteEmitter origin = "testHeartbeatResponder/0" ) BeforeEach(func() { wrappedEmitter = fake.NewFakeByteEmitter() }) Describe("NewHeartbeatResponder", func() { It("requires non-nil args", func() { heartbeatResponder, err := emitter.NewHeartbeatResponder(nil, origin) Expect(err).To(HaveOccurred()) Expect(err.Error()).To(Equal("wrappedEmitter is nil")) Expect(heartbeatResponder).To(BeNil()) }) }) Describe("Emit", func() { var ( heartbeatResponder emitter.RespondingByteEmitter testData = []byte("hello")