#!/usr/bin/python

from six.moves import urllib

def web_socket_do_extra_handshake(request):
    url_parts = urllib.parse.urlsplit(request.uri)
    request.extra_headers.append(('Set-Cookie', 'ws_test_'+(url_parts.query or '')+'=test; Path=/; HttpOnly\x0D\x0ASec-WebSocket-Origin: '+request.ws_origin))

def web_socket_transfer_data(request):
    # Expect close from user agent.
    request.ws_stream.receive_message()