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 s t jd| jdd}d S Nzhttp://localhost:{0}FZverifyrequestsgetformatr/ r responser r r
test_index< s zHTTP01ServerTest.test_indexc C s t jd| jdd}d S Nzhttp://localhost:{0}/fooFr= r> rB 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)challrC
validation localhost)r/ r ZHTTP01Zresponse_and_validationr, r rI ZHTTP01Resourcer$ addrC Z
simple_verifyrJ Z
public_keyr/ r rN rJ rC rK rI resourcer r r _test_http01G s z HTTP01ServerTest._test_http01c C s d S r6 r r: r r r test_http01_foundT s z"HTTP01ServerTest.test_http01_foundc C s d S r6 r r: r r r test_http01_not_foundW s z&HTTP01ServerTest.test_http01_not_foundc C s ddl m} |dt dd}tj|jd}| t n}|d|j d f tj|j
d}| | d | }z|
tj
W n Y n0 W d n1 s0 Y W d n1 s0 Y d S )
Nr r! r g?)r$ timeoutr& rL r% g @)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; rD rF rQ rR rS rW 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)rL r )certschallenge_certsr& )
r Zload_pyopenssl_private_keyZ load_certr[ r\ r rZ r r0 r1 r2 r3 r4 )r rZ 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 s2 | j j d d \}}tjd||ddgd}d S )N rY r% s
acme-tls/1hostr/ rT Zalpn_protocols)r r r. r probe_sni)r r_ 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 )Nr] rY r% s bad-alpnr^ ) r r r. pytestraisesr Errorr r` )r r_ r/ r r r
test_bad_alpn s z!TLSALPN01ServerTest.test_bad_alpnN)r r r r r5 r; ra re r r r r rX q s
rX 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 rj s
z:BaseDualNetworkedServersTest.SingleProtocolServer.__init__)r r r r rj
__classcell__r r ro r SingleProtocolServer s rr zsocket.socket.bindc C sd ddl m} ddlm} t|d|_ttj }|t
j dt j
W d n1 sV0 Y d S )Nr )
EADDRINUSEBaseDualNetworkedServerszFake addr in use errorr )errnors r ru OSErrorZ side_effectrb rc r errorrf rr r r )r Z mock_bindrs ru exc_infor r r test_fail_to_bind s "z.BaseDualNetworkedServersTest.test_fail_to_bindc C sZ ddl m} |tjdtj}| }d }|D ]}|d }|r<|}q,|jD ] }| qHd S )Nr rt r r% ) r ru rf rr r r getsocknamesserversr )r ru r| Z socknamesZ prev_portZsocknamer/ r r r r test_ports_equal s
z-BaseDualNetworkedServersTest.test_ports_equalN) r r r r r Z TCPServerrr r Zpatchrz r} r r r r rf s
rf 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 s t jd| jdd}d S r< r> rB r r r rD s z)HTTP01DualNetworkedServersTest.test_indexc C s t jd| jdd}d S rE r> rB r r r rF s z'HTTP01DualNetworkedServersTest.test_404c C sf t jdd}|| j\}}ddlm} |j|||d}|rH| j| |j j
|j d| j | j
dS rG rM rO r r r rQ s z+HTTP01DualNetworkedServersTest._test_http01c C s d S r6 r r: r r r rR s z0HTTP01DualNetworkedServersTest.test_http01_foundc C s d S r6 r r: r r r rS s z4HTTP01DualNetworkedServersTest.test_http01_not_foundN) r r r r r5 r; rD rF rQ rR rS r r r r r~ s
r~ __main__r% )$r Z http.clientrV Z http_clientr r sysr0 typingr Zunittestr Zjosepyr( rb r? Zacmer r r Zacme._internal.testsr ZTestCaser r ZskipIfZ TLSALPN01 is_supportedrX rf r~ r exitmainargv__file__r r r r s. J3:0