func (w *WebsocketServer) streamLogs(appId string, ws *gorilla.Conn) {
w.logger.Debugf("WebsocketServer: Requesting a wss sink for app %s", appId)
websocketSink := websocket.NewWebsocketSink(
appId,
w.logger,
ws,
w.bufferSize,
)
w.sinkManager.RegisterSink(websocketSink)
defer w.sinkManager.UnregisterSink(websocketSink)
go ws.ReadMessage()
server.NewKeepAlive(ws, w.keepAliveInterval).Run()
}
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)
})
Describe("Identifier", func() {
It("returns the remote address", func() {
Expect(websocketSink.Identifier()).To(Equal("RemoteAddressString"))
})
})
Describe("AppId", func() {
It("returns the application id", func() {
Expect(websocketSink.AppId()).To(Equal("appId"))
})
})
Describe("ShouldReceiveErrors", func() {