return fake.writeDeadline } type fakeCounter struct { incrementCalls chan struct{} } func (f *fakeCounter) Increment() { f.incrementCalls <- struct{}{} } var _ = Describe("WebsocketSink", func() { var ( 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")) }) })
fake.messages = append(fake.messages, data) return nil } func (fake *fakeMessageWriter) ReadMessages() [][]byte { fake.RLock() defer fake.RUnlock() return fake.messages } var _ = Describe("WebsocketSink", func() { var ( logger *gosteno.Logger websocketSink *websocket.WebsocketSink fakeWebsocket *fakeMessageWriter ) BeforeEach(func() { logger = loggertesthelper.Logger() fakeWebsocket = &fakeMessageWriter{} websocketSink = websocket.NewWebsocketSink("appId", logger, fakeWebsocket, 10, "dropsonde-origin") }) Describe("Identifier", func() { It("returns the remote address", func() { Expect(websocketSink.Identifier()).To(Equal("client-address")) }) })