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"))
})
})