U
|
Z ±dÞF ã @ sv d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z m Z m Z mZ ejdk rnd dlmZ dd Zdd Zd
|
d Zd d Zdd Zdd Zdd ZejddG dd dejjZG dd dej j ej j Z dddd d d Z d d Z!d!d" Z"d#d$ Z#d%d& Z$d'd( Z%d)d* Z&d+d, Z'd-d. Z(d/d0 Z)d1d2 Z*d3d4 Z+d5d6 Z,d7d8 Z-d9d: Z.d;d< Z/d=d> Z0dS )?é N)ÚAF_INETÚAF_INET6Ú SOCK_STREAMÚ IPPROTO_TCP)Ú reorder_for_rfc_6555_section_5_4Ú close_allÚopen_tcp_streamÚformat_host_port)é é )ÚBaseExceptionGroupc
|
C sà G dd d} G dd d}| }t }| |¡ W 5 Q R X |jsHt| }t t¡& t }| |¡ tW 5 Q R X W 5 Q R X |jst| }t t¡. t }| | ¡ | |¡ W 5 Q R X W 5 Q R X |jsÜtd S )Nc @ s e Zd ZdZdd ZdS )z test_close_all.<locals>.CloseMeFc S s
|
d| _ d S ©NT©Úclosed©Úself© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_highlevel_open_tcp_stream.pyÚclose s z%test_close_all.<locals>.CloseMe.closeN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r ÚCloseMe s r c @ s e Zd Zdd ZdS )z#test_close_all.<locals>.CloseKillerc S s t d S ©N)ÚOSErrorr r r r r s z)test_close_all.<locals>.CloseKiller.closeN)r r r r r r r r Ú CloseKiller s r )r Úaddr ÚAssertionErrorÚpytestÚraisesÚ RuntimeErrorr )r r ÚcÚto_closer r r Útest_close_all s$
|
|
|
r# c C s dd } dd }| |fD ]d}|d|d|dg}t | ||d|d|dgksXt|dg}t | ||dgkstq| d|d| d|dg}t|}t | ||ks¶t| d| d| d|d|dg}t | || d|d| d| d|dgks td S )Nc S s t ttdd | ¡dffS )NÚ z 10.0.0.{}éP )r r r Úformat©Úir r r Úfake44 s ûz4test_reorder_for_rfc_6555_section_5_4.<locals>.fake4c S s t ttdd | ¡dffS )Nr$ z::{}r% )r r r r& r' r r r Úfake6= s z4test_reorder_for_rfc_6555_section_5_4.<locals>.fake6r é é )r r Úlist)r) r* ZfakeÚtargetsÚorigr r r Ú%test_reorder_for_rfc_6555_section_5_43 s
|
"r0 c C sp t dddkstt dddks$tt dddks6tt dddksHtt d d
|
d ksZtt d d
|
d ksltd S ) Nú 127.0.0.1r% z 127.0.0.1:80s 127.0.0.1z example.comi» zexample.com:443s example.comú::1Úhttpz
|
[::1]:https ::1)r r r r r r Útest_format_host_portX s r4 c à s t j ¡ } | d¡I d H | ¡ \}}| d¡ td|I d H }| ¡ I d H \}}| d¡I d H | d¡I d H dkszt |
|
¡ I d H | ¡ | ¡ d S )N©r1 r r+ r1 ó x) ÚtrioÚsocketÚbindÚ getsocknameÚlistenr ÚacceptZsend_allÚrecvr Úacloser )Z listen_sockÚ_Z listen_portÚ client_streamÚ server_sockr r r Ú*test_open_tcp_stream_real_socket_smoketestb s
|
|
rB c à sP t t¡ td dI d H W 5 Q R X t t¡ tddI d H W 5 Q R X d S )Nr% r1 s 80)r r Ú
|
ValueErrorr Ú TypeErrorr r r r Ú%test_open_tcp_stream_input_validationq s rE c
|
C s` t ¡ N} z| d¡ W n tk
|
r8 Y W 5 Q R £ dS X | ¡ d dkW 5 Q R £ S Q R X d S )N)ú 127.0.0.2r Fr rF )r8 r9 r r: )Úsr r r Úcan_bind_127_0_0_2x s
|
rH c à sx t j ¡ b} | d¡I d H | ¡ t r2d}nd}t| ¡ d|iI d H 4 I d H z}|j ¡ d |ksnttt jdr|j t jj
|
t jj ¡st| ¡ I d H \}}| ¡ I d H | ¡ |d |ksÌtW 5 Q I d H R X t t¡ t| ¡ ddiI d H W 5 Q R X t| ¡ ddiI d H 4 I d H 4}| ¡ I d H \}}| ¡ ||j ¡ ksZtW 5 Q I d H R X W 5 Q R X d S ) Nr5 rF r1 Ú local_addressr ÚIP_BIND_ADDRESS_NO_PORTz::z0.0.0.0)r7 r8 r9 r; rH r r: r ÚhasattrÚ
|
getsockoptÚ
|
IPPROTO_IPrJ r< r> r r r r )ZlistenerrI r@ rA Z remote_addrr r r Útest_local_address_real sB ÿÿ ÿ "ÿÿrN F)Úeqc @ s e Zd Ze ¡ Ze ¡ Ze ¡ Ze ¡ ZejddZ ejddZ
|
ejddZ ejddZ ejddZ dd Zdd Zdd ZdS )
|
Ú
|
FakeSocketN)ÚdefaultFc à s |d | _ |d | _| j | jjks&t| | jj| j < t ¡ | jj| j < | jj| j \}}t |¡I d H |dkrxt
|
d|dkrd| _ d| _ d S )Nr r+ ÚerrorZsorryÚpostconnect_failT) ÚipÚportÚscenarioÚsocketsr r7 Ú current_timeÚ connect_timesÚip_dictÚsleepr ÚfailingÚ succeeded)r ÚsockaddrÚdelayÚresultr r r ÚconnectÀ s
|
|
zFakeSocket.connectc C s
|
d| _ d S r r r r r r r Î s zFakeSocket.closec O s | j r
|
td S r )r\ ÚKeyboardInterrupt)r ÚargsÚkwargsr r r Ú
|
setsockoptÒ s zFakeSocket.setsockopt)r r r ÚattrÚibrV ÚfamilyÚtypeÚprotorT rU r] r r\ ra r re r r r r rP ³ s rP c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚScenarioc C s dd |D }t t|t |ks&ti }|D ].\}}}d|ksDt|dksPt||f||< q.|| _|| _|| _|| _d| _i | _i | _ d S )Nc S s g | ]\}}}|qS r r )Ú.0rT r? r r r Ú
|
<listcomp>Ü s z%Scenario.__init__.<locals>.<listcomp>r )rR ÚsuccessrS )
|
ÚlenÚsetr rU Úip_orderrZ Úsupported_familiesÚ socket_countrW rY ) r rU Úip_listrr rq rZ rT r_ r` r r r Ú__init__Ú s zScenario.__init__c C s. || j krtd| jd7 _t| |||S )Nz%pretending not to support this familyr+ )rr r rs rP )r rh ri rj r r r r8 ì s
|
zScenario.socketc C s@ d|kr t jj}|| jddf}nt jj}|| jf}|ttd|fS )Nú:r r$ )r7 r8 r rU r r r )r rT rh r^ r r r Ú_ip_to_gai_entryò s
|
zScenario._ip_to_gai_entryc sf |dks t | jkst |tjjks*t |tjjks:t |dksFt |dksRt fdd jD S )Ns test.example.comr c s g | ]} |¡qS r )rw )rl rT r r r rm s z(Scenario.getaddrinfo.<locals>.<listcomp>)r rU r7 r8 Ú AF_UNSPECr rq )r ÚhostrU rh ri rj Úflagsr r r Ú getaddrinfoû s zScenario.getaddrinfoc à s t d S r )ÚNotImplementedError)r r^ rz r r r Ú getnameinfo s zScenario.getnameinfoc C s\ | j t| jkst| j ¡ D ]8\}}|| jks4t||k rF|jsFt|j| jks tq d S r )rs ro rW r ÚitemsrZ r rU )r r] rT r8 r r r Úcheck s
|
zScenario.checkN) r r r ru r8 rw r{ r} r r r r r rk Ù s rk Tr )Úipv4_supportedÚipv6_supportedÚ expect_errorc
|
|
Ë sØ t }|r| tjj¡ |r*| tjj¡ t| ||}tj |¡ tj |¡ z8t d| f|I d H }|dkspt
|
| |j¡ |j|fW S t
|
k
|
r Y n8 |k
|
rÒ } z| d ¡ | |f W Y ¢S d } ~ X Y nX d S )Nztest.example.comr ) rp r r7 r8 r r rk Z set_custom_hostname_resolverZset_custom_socket_factoryr r r )
|
rU rt r r r rd rr rV ÚstreamÚexcr r r Ú run_scenario s$
|
r
c à s8 t ddgI d H \}}|jdks$tt ¡ dks4td S )Nr% )ú1.2.3.4ç°rhí|¿?rn r r ©r
rT r r7 rX ©Úautojump_clockÚsockrV r r r Útest_one_host_quick_success< s r c à s8 t ddgI d H \}}|jdks$tt ¡ dks4td S )NéQ )r éd rn r r r r r r r Útest_one_host_slow_successB s r c à s< t ddgtdI d H \}}t|ts(tt ¡ dks8td S )NéR )r r rR ©r r ©r
r Ú
|
isinstancer r7 rX ©r r rV r r r Útest_one_host_quick_failH s ÿr c à s< t ddgtdI d H \}}t|ts(tt ¡ dks8td S )NéS )r r rR r r r r r r r Útest_one_host_slow_failP s ÿr c à s, t ddgtdI d H \}}t|ts(td S )Nr )r r+ rS r )r
rb r r r r r r Ú"test_one_host_failed_after_connectX s ÿr c à sR t ddddgI d H \}}|jdks(tt ¡ dks8t|jddd d
|
ksNtd S ) Nr% ©ú1.1.1.1r+ rn ©ú2.2.2.2r+ rn ©ú3.3.3.3çÉ?rn r gffffffæ?r ç Ð?ç à?©r r r ©r
rT r r7 rX rY r r r r Útest_basic_fallthrough` s ýþýr¤ c à sP t ddddgI d H \}}|jdks(tt ¡ dks8t|jddd ksLtd S )
|
Nr% r )r ç¹?rn r r gffffffÖ?r r )r r r£ r r r r Útest_early_successs s ýþþr¦ c à sV t ddddgddI d H \}}|jdks,tt ¡ dks<t|jd dd
|
d ksRtd S ) Nr% r r r gÍÌÌÌÌÌÜ?©Úhappy_eyeballs_delayr r+ r gÍÌÌÌÌÌì?r¢ r£ r r r r Útest_custom_delay s ýù ýr© c à sV t dddddgI d H \}}|jdks*tt ¡ dks:t|jdd d
|
d d ksRtd S ) Nr% ©r r¥ rR ©r r rR ©r é
|
rn )ú4.4.4.4r rn r® gé?r r¥ ç433333Ó?çá?©r r r r® r£ r r r r Útest_custom_errors_expedite s ûþ
|
ür² c à s~ t dddddgtdI d H \}}t|ts.tt|jts>tt|jjdksRtt ¡ dksbt|j
|
d d
|
d d d ksztd S )Nr% rª r« )r r rR )r® r rR r é g$@r r¥ r¯ r° r± ) r
r r r Ú __cause__r ro Ú
|
exceptionsr7 rX rY r r r r Ú test_all_fail® s& üø
|
ür¶ c à s¦ t ddddddgddI d H \}}|jd jr2t|jd
|
jsZ|jd jsZ|jd jsZt|jd jrjt|jdksxtt ¡ dkst|jddddddks¢td S )Nr% )r r¡ rR ©r r rn )r é rn )r® é rn )ú5.5.5.5r¡ rR r+ r§ r r r r® rº )r r r® g %@r r¡ ç ø?g @g @)r r r r® rº )r
rW r] r rT r7 rX rY r r r r Útest_multi_successÅ s6 û÷
|
ÿ
|
þ
|
ýûr¼ c à sT t ddddgddI d H \}}|jdks,tt ¡ dks<t|jd dd
|
ksPtd S ) Nr% )r r rR r )ú::3r¡ rn r+ r§ r½ r» r )r r½ r£ r r r r Útest_does_reorderã s û÷ þr¾ c à sX t dddddgdddI d H \}}|jd ks0tt ¡ d
|
ks@t|jd d d ksTtd S )Nr% )r2 r rn )r r rn )r½ r¥ rn )r® r rn r+ F)r¨ r r½ çñ?r ç ð?)r2 r½ r£ r r r r Útest_handles_no_ipv4÷ s üõ þrÁ c à sX t dddddgdddI d H \}}|jd ks0tt ¡ d
|
ks@t|jd d d ksTtd S )Nr% )r2 r rn r· )r½ r rn )r® r¥ rn r+ F)r¨ r r® r¿ r rÀ )r r® r£ r r r r Útest_handles_no_ipv6 s üõ þr c à s, t dg tdI d H \}}dt|ks(td S )Nr% r zno results found)r
r Ústrr r r r r Ú test_no_hosts# s rÄ c à sd t d¡*}tdddddgtdI d H \}}|W 5 Q R X |jsDtt ¡ dksTt|jdd d S )
|
Né r% )r r rn r· r¬ )r® r rn r F)r] )r7 Z move_on_afterr
r Zcancelled_caughtr rX r )r Z cancel_scoper rV r r r Ú test_cancel( s üø
|
rÆ )1r Úsysr8 rf r7 Z trio.socketr r r r Z trio._highlevel_open_tcp_streamr r r r Ú version_infoZexceptiongroupr r# r0 r4 rB rE rH rN rG Ú
|
SocketTyperP ÚabcZ SocketFactoryZHostnameResolverrk r
r r r r r r¤ r¦ r© r² r¶ r¼ r¾ rÁ r rÄ rÆ r r r r Ú<module> sL
|
%
|
2
|
%Fï )
|
