var startFakeTrafficController = func() {
fakeHandler = &FakeHandler{innerHandler: handlers.NewWebsocketHandler(messagesToSend, 100*time.Millisecond)}
testServer = httptest.NewServer(fakeHandler)
endpoint = "ws://" + testServer.Listener.Addr().String()
appGuid = "app-guid"
}
Describe("Debug Printing", func() {
var debugPrinter *fakeDebugPrinter
BeforeEach(func() {
startFakeTrafficController()
debugPrinter = &fakeDebugPrinter{}
connection = consumer.New(endpoint, tlsSettings, consumerProxyFunc)
connection.SetDebugPrinter(debugPrinter)
})
It("includes websocket handshake", func() {
close(messagesToSend)
connection.Tail(appGuid, authToken)
Expect(debugPrinter.Messages[0].Body).To(ContainSubstring("Sec-WebSocket-Version: 13"))
})
It("does not include messages sent or received", func() {
messagesToSend <- marshalMessage(createMessage("hello", 0))
close(messagesToSend)
connection.Tail(appGuid, authToken)
