def connect(self):
super(WebSocketsClient, self).connect()
preamble = http.request_preamble("GET", "/")
self.wfile.write(preamble + "\r\n")
headers = websockets.client_handshake_headers()
self.client_nonce = headers.get_first("sec-websocket-key")
self.wfile.write(headers.format() + "\r\n")
self.wfile.flush()
resp = http.read_response(self.rfile, "get", None)
server_nonce = websockets.check_server_handshake(resp.headers)
if not server_nonce == websockets.create_server_nonce(self.client_nonce):
self.close()
def connect(self):
super(WebSocketsClient, self).connect()
preamble = http.request_preamble("GET", "/")
self.wfile.write(preamble + "\r\n")
headers = websockets.client_handshake_headers()
self.client_nonce = headers.get_first("sec-websocket-key")
self.wfile.write(headers.format() + "\r\n")
self.wfile.flush()
resp = http.read_response(self.rfile, "get", None)
server_nonce = websockets.check_server_handshake(resp.headers)
if not server_nonce == websockets.create_server_nonce(
self.client_nonce):
self.close()
def test_check_server_handshake(self):
headers = websockets.server_handshake_headers("key")
assert websockets.check_server_handshake(headers)
headers["Upgrade"] = ["not_websocket"]
assert not websockets.check_server_handshake(headers)
def test_check_server_handshake(self):
headers = websockets.server_handshake_headers("key")
assert websockets.check_server_handshake(headers)
headers["Upgrade"] = ["not_websocket"]
assert not websockets.check_server_handshake(headers)
