U
|
Z ±dÆ ã @ s d dl Z d dlmZ d dlZd dlZd dlmZmZmZ d dl Zddl
|
m Z m Z ddl mZmZmZ dd ZejG d
|
d d ejjZd d Zdd ZdS )é N)Úpartial)ÚAF_INETÚ SOCK_STREAMÚ IPPROTO_TCPé )Ú
|
client_ctxÚ
|
SERVER_CTXé )Úopen_ssl_over_tcp_streamÚopen_ssl_over_tcp_listenersÚserve_ssl_over_tcpc
|
à sf | 4 I d H J z,| d¡I d H }|s&q8| |¡I d H qW n tjk
|
rP Y nX W 5 Q I d H R X d S )Ni' )Ú receive_someÚsend_allÚtrioÚBrokenResourceError)ÚstreamÚdata© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_highlevel_ssl_helpers.pyÚ echo_handler s r c @ s$ e Zd Ze ¡ Zdd Zdd ZdS )ÚFakeHostnameResolverc Ç s t ttd| jfgS )NÚ )r r r Úsockaddr©ÚselfÚargsr r r Ú getaddrinfo% s z FakeHostnameResolver.getaddrinfoc Ç s t d S )N)ÚNotImplementedErrorr r r r Ú getnameinfo( s z FakeHostnameResolver.getnameinfoN)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibr r r r r r r r ! s r c à s
|
t ¡ 4 I d H è}| tttdtdd¡I d H \}|4 I d H ¨ |jj ¡ }t
|
|}t j |¡ t ddI d H }|4 I d H , t t j¡ | ¡ I d H W 5 Q R X W 5 Q I d H R X t dd| dI d H }|4 I d H , t t j¡ | ¡ I d H W 5 Q R X W 5 Q I d H R X t dd| dI d H }|4 I d H R t|t jsDt|jdksTt| d¡I d H | d ¡I d H dks~tW 5 Q I d H R X |jrtt dd| d
|
d d I d H }|4 I d H |jsÌtW 5 Q I d H R X |j ¡ W 5 Q I d H R X W 5 Q I d H R X d S ) Nr ú 127.0.0.1©Úhostztrio-test-1.example.orgéP zxyzzy.example.org)Ú ssl_contextó xr T)r( Úhttps_compatibleZhappy_eyeballs_delay)r Z open_nurseryÚstartr r r r Útransport_listenerÚsocketÚ getsocknamer Z set_custom_hostname_resolverr
|
ÚpytestZraisesr Ú do_handshakeÚ
|
isinstanceZ SSLStreamÚAssertionErrorÚserver_hostnamer r Ú_https_compatibleZ cancel_scopeÚcancel)r ZnurseryÚlistenerr Zhostname_resolverr r r r Ú1test_open_ssl_over_tcp_stream_and_everything_else. sR ÿ ( ÿ ( ÿ * ú r7 c
|
à s¾ t dtddI d H \} | 4 I d H L t| tjs4t| j}t|tjsJt|j ¡ d dks`t| j
|
rjtW 5 Q I d H R X t dtdddI d H \} | 4 I d H | j
|
sªtW 5 Q I d H R X d S )Nr r$ r% T)r& r* ) r r r1 r Z SSLListenerr2 r, ZSocketListenerr- r. r4 )r6 Útlr r r Ú test_open_ssl_over_tcp_listenersc s ÿr9 )r/ Ú functoolsr r" r Z trio.socketr r r Z trio.testingZtest_sslr r Z_highlevel_ssl_helpersr
|
r r r ÚsÚabcZHostnameResolverr r7 r9 r r r r Ú<module> s 5
|
