a
}|g& @ s4 d Z ddlmZ ddlZddlZddlZddlZddlm Z ddl
Z
ddl
m Z ddl Z
ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ G d d
d
e
jZG d d d e
jZe
ej d
G dd de
jZ G dd de
jZ G dd de
jZ e dkr0e e!ej"dd e#g dS )zTests for acme.standalone. N)Set)mock)
challenges) crypto_util)errors) test_utilc @ s e Zd ZdZdd Zdd ZdS )
TLSServerTestz$Tests for acme.standalone.TLSServer.c C s( ddl m} |dtjdd}| d S )Nr TLSServer r T)bind_and_activate)acme.standaloner
socketserverBaseRequestHandler server_closeselfr
server r H/usr/lib/python3.9/site-packages/acme/_internal/tests/standalone_test.py test_bind s
zTLSServerTest.test_bindc C s0 t jr,ddlm} |dtjddd}| d S )Nr r r T)r
ipv6)socketZhas_ipv6r r
r r r r r r r test_ipv6 s
zTLSServerTest.test_ipv6N)__name__
__module__ __qualname____doc__r r r r r r r s r c @ sP e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )HTTP01ServerTestz'Tests for acme.standalone.HTTP01Server.c C sj t jtd| _t | _ddlm } |d| jd| _
| j
j d | _
tj| j
jd| _| j d S )Nrsa1024_key.pemr HTTP01Serverr resources target)joseJWKloadr load_vector account_keysetr$ r r" r r getsocknameport threadingThread
serve_foreverthreadstart)r r" r r r setUp+ s zHTTP01ServerTest.setUpc C s" | j | j | j d S Nr shutdownr3 joinr r r r r tearDown7 s
zHTTP01ServerTest.tearDownc C s2 t jd| jdd}|jdks$J |js.J d S Nzhttp://localhost:{0}FZverifyz'ACME client standalone challenge solverrequestsgetformatr/ textokr responser r r
test_index< s
zHTTP01ServerTest.test_indexc C s* t jd| jdd}|jtjks&J d S Nzhttp://localhost:{0}/fooFr= r? r@ rA r/ Z status_code http_clientZ NOT_FOUNDrD r r r test_404B s zHTTP01ServerTest.test_404c C sf t jdd}|| j\}}ddlm} |j|||d}|rH| j| |j j
|j d| j | j
dS Ns xxxxxxxxxxxxxxxx)tokenr )HTTP01RequestHandler)challrE
validation localhost)r/ r ZHTTP01Zresponse_and_validationr, r rM ZHTTP01Resourcer$ addrE Z
simple_verifyrN Z
public_keyr/ r rR rN rE rO rM resourcer r r _test_http01G s z HTTP01ServerTest._test_http01c C s | j ddsJ d S NTrR rU r: r r r test_http01_foundT s z"HTTP01ServerTest.test_http01_foundc C s | j ddrJ d S NFrW rX r: r r r test_http01_not_foundW s z&HTTP01ServerTest.test_http01_not_foundc C s ddl m} |dt dd}tj|jd}| t z}|d|j d f tj|j
d}| | d | }z|
tj
W n Y n0 |rJ d
W d n1 s0 Y W d n1 s0 Y d S ) Nr r! r g?)r$ timeoutr& rP r% g @z"Server shutdown should not be hung)r r" r- r0 r1 r2 r4 r Zconnectr. r8 r9 is_aliveZ SHUT_RDWR)r r" r Z
server_threadclientZ stop_threadZis_hungr r r test_timely_shutdownZ s
z%HTTP01ServerTest.test_timely_shutdownN) r r r r r5 r; rF rJ rU rY r[ r_ r r r r r ' s
r zpyOpenSSL too oldc @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) TLSALPN01ServerTestz)Test for acme.standalone.TLSALPN01Server.c C st dt dt dfi| _dt dt dfi| _ddlm} |d| j| jd | _tj | jj
d
| _ | j d S ) N localhostzrsa2048_key.pemzrsa2048_cert.pemzrsa4096_key.pemzrsa4096_cert.pemr )TLSALPN01Server)rP r )certschallenge_certsr& )
r Zload_pyopenssl_private_keyZ load_certrc rd r rb r r0 r1 r2 r3 r4 )r rb r r r r5 u s zTLSALPN01ServerTest.setUpc C s" | j | j | j d S r6 r7 r: r r r r; s
z TLSALPN01ServerTest.tearDownc C sT | j j d d \}}tjd||ddgd}t|t| jd d ksPJ d S )N ra r% s
acme-tls/1hostr/ r\ Zalpn_protocols)r r r. r probe_snir( ZComparableX509rd )r rg r/ Zcertr r r test_challenge_certs s z(TLSALPN01ServerTest.test_challenge_certsc C s^ | j j d d \}}ttj& tjd||ddgd W d n1 sP0 Y d S )Nre ra r% s bad-alpnrf ) r r r. pytestraisesr Errorr rh )r rg r/ r r r
test_bad_alpn s z!TLSALPN01ServerTest.test_bad_alpnN)r r r r r5 r; ri rm r r r r r` q s
r` c @ s< e Zd ZdZG dd dejZeddd Z dd Z
d S )
BaseDualNetworkedServersTestz2Test for acme.standalone.BaseDualNetworkedServers.c s e Zd ZdZ fddZ ZS )z1BaseDualNetworkedServersTest.SingleProtocolServerzUServer that only serves on a single protocol. FreeBSD has this behavior for AF_INET6.c s | dd}|r"tj| _d|d< ntj| _t j|i | |rttdd}| j|tj d z|
| W n | Y n0 d S )Nr Fr
Z IPPROTO_IPV6) r% )
popr ZAF_INET6Zaddress_familyZAF_INETsuper__init__getattrZ
setsockoptZ IPV6_V6ONLYZ server_bindZserver_activater )r argskwargsr level __class__r r rr s
z:BaseDualNetworkedServersTest.SingleProtocolServer.__init__)r r r r rr
__classcell__r r rw r SingleProtocolServer s rz zsocket.socket.bindc C st ddl m} ddlm} t|d|_ttj }|t
j dt j
W d n1 sV0 Y |jj |kspJ d S )Nr )
EADDRINUSEBaseDualNetworkedServerszFake addr in use errorr )errnor{ r r} OSErrorZ side_effectrj rk r errorrn rz r r value)r Z mock_bindr{ r} exc_infor r r test_fail_to_bind s "z.BaseDualNetworkedServersTest.test_fail_to_bindc C sf ddl m} |tjdtj}| }d }|D ] }|d }|rH||ksHJ |}q,|jD ] }| qTd S )Nr r| r r% ) r r} rn rz r r getsocknamesserversr )r r} r Z socknamesZ prev_portZsocknamer/ r r r r test_ports_equal s
z-BaseDualNetworkedServersTest.test_ports_equalN) r r r r r Z TCPServerrz r Zpatchr r r r r r rn s
rn c @ sH e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dS ) HTTP01DualNetworkedServersTestz5Tests for acme.standalone.HTTP01DualNetworkedServers.c C sZ t jtd| _t | _ddlm } |d| jd| _
| j
d d | _ | j
d S )Nr r )HTTP01DualNetworkedServersr r# r% )r( r) r* r r+ r, r- r$ r r r r r/ r2 )r r r r r r5 s z$HTTP01DualNetworkedServersTest.setUpc C s | j d S r6 )r Zshutdown_and_server_closer: r r r r; s z'HTTP01DualNetworkedServersTest.tearDownc C s2 t jd| jdd}|jdks$J |js.J d S r< r> rD r r r rF s
z)HTTP01DualNetworkedServersTest.test_indexc C s* t jd| jdd}|jtjks&J d S rG rH rD r r r rJ s z'HTTP01DualNetworkedServersTest.test_404c C sf t jdd}|| j\}}ddlm} |j|||d}|rH| j| |j j
|j d| j | j
dS rK rQ rS r r r rU s z+HTTP01DualNetworkedServersTest._test_http01c C s | j ddsJ d S rV rX r: r r r rY s z0HTTP01DualNetworkedServersTest.test_http01_foundc C s | j ddrJ d S rZ rX r: r r r r[ s z4HTTP01DualNetworkedServersTest.test_http01_not_foundN) r r r r r5 r; rF rJ rU rY r[ r r r r r s
r __main__r% )$r Z http.clientr^ rI r r sysr0 typingr Zunittestr Zjosepyr( rj r? Zacmer r r Zacme._internal.testsr ZTestCaser r ZskipIfZ TLSALPN01 is_supportedr` rn r r exitmainargv__file__r r r r s. J3:0