logger *gosteno.Logger websocketSink *websocket.WebsocketSink fakeWebsocket *fakeMessageWriter writeTimeout time.Duration ) BeforeEach(func() { logger = loggertesthelper.Logger() fakeWebsocket = &fakeMessageWriter{} writeTimeout = 5 * time.Second websocketSink = websocket.NewWebsocketSink("appId", logger, fakeWebsocket, 10, writeTimeout, "dropsonde-origin") }) Describe("Identifier", func() { It("returns the remote address", func() { Expect(websocketSink.Identifier()).To(Equal("client-address")) }) }) Describe("StreamId", func() { It("returns the application id", func() { Expect(websocketSink.AppID()).To(Equal("appId")) }) }) Describe("ShouldReceiveErrors", func() { It("returns true", func() { Expect(websocketSink.ShouldReceiveErrors()).To(BeTrue()) }) })