func Initialize() { http.DefaultTransport = &http.Transport{Proxy: http.ProxyFromEnvironment} autowiredEmitter = nil origin := os.Getenv("DROPSONDE_ORIGIN") if len(origin) == 0 { log.Println("Failed to auto-initialize dropsonde: DROPSONDE_ORIGIN environment variable not set") return } destination = os.Getenv("DROPSONDE_DESTINATION") if len(destination) == 0 { log.Println("DROPSONDE_DESTINATION not set. Using " + defaultDestination) destination = defaultDestination } udpEmitter, err := emitter.NewUdpEmitter(destination) if err != nil { log.Printf("Failed to auto-initialize dropsonde: %v\n", err) return } hbEmitter, err := emitter.NewHeartbeatEmitter(udpEmitter, origin) if err != nil { log.Printf("Failed to auto-initialize dropsonde: %v\n", err) return } autowiredEmitter = emitter.NewEventEmitter(hbEmitter, origin) http.DefaultTransport = InstrumentedRoundTripper(http.DefaultTransport) }
) var _ = Describe("HeartbeatEmitter", func() { var ( wrappedEmitter *fake.FakeByteEmitter origin = "testHeartbeatEmitter/0" ) BeforeEach(func() { emitter.HeartbeatInterval = 10 * time.Millisecond wrappedEmitter = fake.NewFakeByteEmitter() }) Describe("NewHeartbeatEmitter", func() { It("requires non-nil args", func() { hbEmitter, err := emitter.NewHeartbeatEmitter(nil, origin) Expect(err).To(HaveOccurred()) Expect(err.Error()).To(Equal("wrappedEmitter is nil")) Expect(hbEmitter).To(BeNil()) }) It("starts periodic heartbeat emission", func() { hbEmitter, err := emitter.NewHeartbeatEmitter(wrappedEmitter, origin) Expect(err).NotTo(HaveOccurred()) Expect(hbEmitter).NotTo(BeNil()) Eventually(func() int { return len(wrappedEmitter.GetMessages()) }).Should(BeNumerically(">=", 2)) }) It("logs an error when heartbeat emission fails", func() { wrappedEmitter.ReturnError = errors.New("fake error")