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)