def function2348(function1720, function596): function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596) function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '1')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596)
def test_protocol_version(message, transport): message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport) message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '1')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport)
def test_handshake_compress_bad_ext(message, transport): hdrs, sec_key = gen_ws_headers(compress=15, extension_text='bad') message.headers.extend(hdrs) status, headers, parser, writer, protocol, compress = do_handshake( message.method, message.headers, transport) headers = dict(headers) assert 'Sec-Websocket-Extensions' not in headers assert compress == 0
def function975(function1720, function596): 'Tests if the right protocol is selected given multiple' var68 = 'worse_proto' var688 = ['best', 'chat', 'worse_proto'] var777 = 'worse_proto,chat' function1720.headers.extend(function2240(var777)[0]) (var2450, var4433, var2450, var2450, var4712) = do_handshake(function1720.method, function1720.headers, function596, protocols=var688) assert (var4712 == var68)
def test_handshake_compress_multi_ext_bad(message, transport): hdrs, sec_key = gen_ws_headers(compress=15, extension_text='bad, permessage-deflate') message.headers.extend(hdrs) status, headers, parser, writer, protocol, compress = do_handshake( message.method, message.headers, transport) headers = dict(headers) assert 'Sec-Websocket-Extensions' in headers assert headers['Sec-Websocket-Extensions'] == 'permessage-deflate'
def test_handshake_compress_wbits(message, transport): hdrs, sec_key = gen_ws_headers(compress=9) message.headers.extend(hdrs) status, headers, parser, writer, protocol, compress = do_handshake( message.method, message.headers, transport) headers = dict(headers) assert 'Sec-Websocket-Extensions' in headers assert headers['Sec-Websocket-Extensions'] == ( 'permessage-deflate; server_max_window_bits=9') assert compress == 9
def function2558(function1720, function596): 'Tests if one protocol is returned by do_handshake' var1535 = 'chat' function1720.headers.extend(function2240(var1535)[0]) (var793, var2315, var793, var793, var1343) = do_handshake(function1720.method, function1720.headers, function596, protocols=[var1535]) assert (var1343 == var1535) var2315 = dict(var2315) assert (var2315['Sec-Websocket-Protocol'] == var1535)
def test_handshake(message, transport): hdrs, sec_key = gen_ws_headers() message.headers.extend(hdrs) status, headers, parser, writer, protocol = do_handshake( message.method, message.headers, transport) assert status == 101 assert protocol is None key = base64.b64encode(hashlib.sha1(sec_key.encode() + WS_KEY).digest()) headers = dict(headers) assert headers['Sec-Websocket-Accept'] == key.decode()
def function466(function1720, function596): (var2670, var597) = function2240() function1720.var4463.extend(var2670) (var623, var4463, var3037, var603, var3161) = do_handshake(function1720.method, function1720.var4463, function596) assert (var623 == 101) assert (var3161 is None) var4577 = base64.b64encode( hashlib.sha1((var597.encode() + WS_KEY)).digest()) var4463 = dict(var4463) assert (var4463['Sec-Websocket-Accept'] == var4577.decode())
def test_handshake_protocol_agreement(message, transport): '''Tests if the right protocol is selected given multiple''' best_proto = 'worse_proto' wanted_protos = ['best', 'chat', 'worse_proto'] server_protos = 'worse_proto,chat' message.headers.extend(gen_ws_headers(server_protos)[0]) _, resp_headers, _, _, protocol = do_handshake( message.method, message.headers, transport, protocols=wanted_protos) assert protocol == best_proto
def test_handshake_protocol_agreement(message, transport): '''Tests if the right protocol is selected given multiple''' best_proto = 'worse_proto' wanted_protos = ['best', 'chat', 'worse_proto'] server_protos = 'worse_proto,chat' message.headers.extend(gen_ws_headers(server_protos)[0]) _, resp_headers, _, _, protocol, _ = do_handshake( message.method, message.headers, transport, protocols=wanted_protos) assert protocol == best_proto
def test_handshake_compress_client_notakeover(message, transport): hdrs, sec_key = gen_ws_headers(compress=15, client_notakeover=True) message.headers.extend(hdrs) status, headers, parser, writer, protocol, compress = do_handshake( message.method, message.headers, transport) headers = dict(headers) assert 'Sec-Websocket-Extensions' in headers assert headers['Sec-Websocket-Extensions'] == ( 'permessage-deflate'), hdrs assert compress == 15
def function2787(arg2248, function1720, function596): 'Tests if a protocol mismatch handshake warns and returns None' var3592 = 'chat' function1720.headers.extend(function2240('test')[0]) with arg2248('aiohttp.websocket') as var1878: (var424, var424, var424, var424, var789) = do_handshake(function1720.method, function1720.headers, function596, protocols=[var3592]) assert (var789 is None) assert (var1878.records[( -1)].msg == 'Client protocols %r don’t overlap server-known ones %r')
def test_handshake(message, transport): hdrs, sec_key = gen_ws_headers() message.headers.extend(hdrs) status, headers, parser, writer, protocol = do_handshake( message.method, message.headers, transport) assert status == 101 assert protocol is None key = base64.b64encode( hashlib.sha1(sec_key.encode() + WS_KEY).digest()) headers = dict(headers) assert headers['Sec-Websocket-Accept'] == key.decode()
def test_handshake_protocol_unsupported(log, message, transport): '''Tests if a protocol mismatch handshake warns and returns None''' proto = 'chat' message.headers.extend(gen_ws_headers('test')[0]) with log('aiohttp.websocket') as ctx: _, _, _, _, protocol = do_handshake( message.method, message.headers, transport, protocols=[proto]) assert protocol is None assert (ctx.records[-1].msg == 'Client protocols %r don’t overlap server-known ones %r')
def test_handshake_protocol_unsupported(log, message, transport): '''Tests if a protocol mismatch handshake warns and returns None''' proto = 'chat' message.headers.extend(gen_ws_headers('test')[0]) with log('aiohttp.websocket') as ctx: _, _, _, _, protocol, _ = do_handshake( message.method, message.headers, transport, protocols=[proto]) assert protocol is None assert (ctx.records[-1].msg == 'Client protocols %r don’t overlap server-known ones %r')
def test_handshake_protocol(message, transport): '''Tests if one protocol is returned by do_handshake''' proto = 'chat' message.headers.extend(gen_ws_headers(proto)[0]) _, resp_headers, _, _, protocol = do_handshake( message.method, message.headers, transport, protocols=[proto]) assert protocol == proto # also test if we reply with the protocol resp_headers = dict(resp_headers) assert resp_headers['Sec-Websocket-Protocol'] == proto
def test_handshake_protocol(message, transport): '''Tests if one protocol is returned by do_handshake''' proto = 'chat' message.headers.extend(gen_ws_headers(proto)[0]) _, resp_headers, _, _, protocol, _ = do_handshake( message.method, message.headers, transport, protocols=[proto]) assert protocol == proto # also test if we reply with the protocol resp_headers = dict(resp_headers) assert resp_headers['Sec-Websocket-Protocol'] == proto
def function2444(function1720, function596): function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596) function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13'), ('Sec-Websocket-Key', '123')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596) var3913 = base64.b64encode(os.urandom(2)) function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13'), ('Sec-Websocket-Key', var3913.decode())]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596)
def test_protocol_key(message, transport): message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport) message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13'), ('Sec-Websocket-Key', '123')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport) sec_key = base64.b64encode(os.urandom(2)) message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'upgrade'), ('Sec-Websocket-Version', '13'), ('Sec-Websocket-Key', sec_key.decode())]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport)
def test_not_get(message, transport): with pytest.raises(http_exceptions.HttpProcessingError): do_handshake('POST', message.headers, transport)
def test_no_connection(message, transport): message.headers.extend([('Upgrade', 'websocket'), ('Connection', 'keep-alive')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport)
def test_no_upgrade(message, transport): with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(message.method, message.headers, transport)
def function2321(function1720, function596): function1720.headers.extend([('Upgrade', 'websocket'), ('Connection', 'keep-alive')]) with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596)
def function2181(function1720, function596): with pytest.raises(http_exceptions.HttpBadRequest): do_handshake(function1720.method, function1720.headers, function596)
def function1224(function1720, function596): with pytest.raises(http_exceptions.HttpProcessingError): do_handshake('POST', function1720.headers, function596)