class Selenium::WebDriver::WebSocketConnection

def socket
  @socket ||= if URI(@url).scheme == 'wss'
                socket = TCPSocket.new(ws.host, ws.port)
                socket = OpenSSL::SSL::SSLSocket.new(socket, OpenSSL::SSL::SSLContext.new)
                socket.sync_close = true
                socket.connect
                socket
              else
                TCPSocket.new(ws.host, ws.port)
              end
end