func newFakeClient(proto, addr string) *fakeclient.FakeClient { c := fakeclient.FakeClient{} c.SchemeReturns(proto) c.AddressReturns(addr) return &c }
"github.com/cloudfoundry/dropsonde/metricbatcher" "github.com/cloudfoundry/dropsonde/metrics" "github.com/cloudfoundry/dropsonde/signature" "github.com/cloudfoundry/gosteno" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var sharedSecret = []byte("secret") var _ = Describe("DopplerForwarder", func() { var ( sender *fake.FakeMetricSender clientPool *fakes.FakeClientPool client *fakeclient.FakeClient logger *gosteno.Logger forwarder *dopplerforwarder.DopplerForwarder envelope *events.Envelope doneChan chan struct{} bufferSize uint ) BeforeEach(func() { sender = fake.NewFakeMetricSender() metrics.Initialize(sender, metricbatcher.New(sender, time.Millisecond*10)) client = &fakeclient.FakeClient{} clientPool = &fakes.FakeClientPool{} clientPool.RandomClientReturns(client, nil) logger = loggertesthelper.Logger() loggertesthelper.TestLoggerSink.Clear()
clientFactory.newClient = func(string) (clientpool.Client, error) { return nil, errors.New("boom") } }) It("does not include the client", func() { pool.SetLegacy(allLegacy, nil) Expect(pool.Clients()).To(BeEmpty()) Expect(loggertesthelper.TestLoggerSink.LogContents()).To(ContainSubstring("Invalid url")) }) }) }) }) Context("when a server no longer exists", func() { var fakeClient *fakeclient.FakeClient BeforeEach(func() { fakeClient = newFakeClient("udp", "host:port") clientFactory.newClient = func(string) (clientpool.Client, error) { return fakeClient, nil } }) It("the client is stopped", func() { s := map[string]string{ "a": "udp://host:port", } pool.Set(s, nil) Expect(fakeClient.CloseCallCount()).To(Equal(0))
"github.com/cloudfoundry/dropsonde/metric_sender/fake" "github.com/cloudfoundry/dropsonde/metricbatcher" "github.com/cloudfoundry/dropsonde/metrics" "github.com/cloudfoundry/dropsonde/signature" "github.com/cloudfoundry/gosteno" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var sharedSecret = []byte("secret") var _ = Describe("DopplerForwarder", func() { var ( sender *fake.FakeMetricSender clientPool *fakes.FakeClientPool client *fakeclient.FakeClient logger *gosteno.Logger forwarder *dopplerforwarder.DopplerForwarder envelope *events.Envelope ) BeforeEach(func() { sender = fake.NewFakeMetricSender() metrics.Initialize(sender, metricbatcher.New(sender, time.Millisecond*10)) client = &fakeclient.FakeClient{} clientPool = &fakes.FakeClientPool{} clientPool.RandomClientReturns(client, nil) logger = loggertesthelper.Logger() loggertesthelper.TestLoggerSink.Clear()