a
}|g @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlZddlm Z ddl
Z ddl Z
ddlZddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ G d
d dejZdd Z G dd dejZ e dkr
e e ej!dd e"g dS )z/Tests for certbot._internal.plugins.standalone. N)Dict)Set)Tuple)mock)
challenges)
standalone) achallenges)errors) acme_util)utilc @ s@ e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dS )ServerManagerTestz=Tests for certbot._internal.plugins.standalone.ServerManager.c C s, ddl m} i | _i | _|| j| j| _d S )Nr )
ServerManager)$certbot._internal.plugins.standaloner
ZcertsZhttp_01_resourcesmgr)selfr
r S/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/standalone_test.pysetUp s zServerManagerTest.setUpc C s d S Nr r r r r test_init s zServerManagerTest.test_initc C s2 | j jd|d}| d d }| j j|d d S )Nr portchallenge_type )r )r run getsocknamesstop)r r serverr r r r _test_run_stop$ s z ServerManagerTest._test_run_stopc C s | tj d S r )r r HTTP01r r r r test_run_stop_http_01+ s z'ServerManagerTest.test_run_stop_http_01c C sD | j jdtjd}| d d }| j j|tjd}| j | d S )Nr r r )r r r r r r )r r r server2r r r test_run_idempotent. s
z%ServerManagerTest.test_run_idempotentc C s t t j}|d | d }t }z|d|f W n tyN Y n0 ttj" | j j
|t j d W d n1 s0 Y |
|
d S )N r r r% )r )socketZAF_INET6bind getsocknameOSErrorpytestraisesr StandaloneBindErrorr r r r close)r Z some_serverr Zmaybe_another_serverr r r test_run_bind_error7 s
$z%ServerManagerTest.test_run_bind_errorN)
__name__
__module__ __qualname____doc__r r r r! r# r. r r r r r s r c C s4 t t jt jd} | d | d }| |S )z%Gets an open port number from the OS.r r$ r )r& ZAF_INETZ SOCK_STREAMr' r( r- )Z open_socketr r r r
get_open_portH s
r3 c @ s e Zd ZdZdd Zdd Zdd Zdd Ze d
d Z
e d d
Z dd Z dd Z
dd Zdd Zedd Zdd Zdd Zd S ) AuthenticatorTestz=Tests for certbot._internal.plugins.standalone.Authenticator.c C s< ddl m} tjt d| _|| jdd| _t | j_d S )Nr )
Authenticator) http01_portr )name)r r5 r MagicMockr3 configauthservers)r r5 r r r r T s zAuthenticatorTest.setUpc C s d S r r r r r r test_more_info[ s z AuthenticatorTest.test_more_infoc C s d S r r r r r r test_get_chall_pref^ s z%AuthenticatorTest.test_get_chall_prefc C s& | }| j|}dd |D }d S )Nc S s g | ]}| |jqS r )response account_key).0Zachallr r r
f z2AuthenticatorTest.test_perform..) _get_achallsr: perform)r Zachallsr> Zexpectedr r r test_performb s z AuthenticatorTest.test_performc C s^ | }t j}ttj|dd}|gdt g | jjj_ |j
}d|_ | |
| d S )Nerrno T)rG
EADDRINUSEr r, r r8 r: r; r side_effectyesno return_valuerE _assert_correct_yesno_call)r mock_get_utility mock_utilityencountered_errnoerror
mock_yesnor r r test_perform_eaddrinuse_retryi s z/AuthenticatorTest.test_perform_eaddrinuse_retryc C s\ | }|j }d|_tj}ttj | | W d n1 sD0 Y | | d S )NF)
rL rM rG rJ r* r+ r PluginError
_fail_performrN )r rO rP rS rQ r r r test_perform_eaddrinuse_no_retryu s (z2AuthenticatorTest.test_perform_eaddrinuse_no_retryc C s |j \}}d S r )Z call_args)r rS Z
yesno_argsZ yesno_kwargsr r r rN s
z,AuthenticatorTest._assert_correct_yesno_callc C s@ t j}ttj | | W d n1 s20 Y d S r )rG ZEACCESr* r+ r rU rV r rQ r r r test_perform_eacces s z%AuthenticatorTest.test_perform_eaccesc C s@ t j}ttj | | W d n1 s20 Y d S r )rG ZENOTCONNr* r+ r r, rV rX r r r $test_perform_unexpected_socket_error s z6AuthenticatorTest.test_perform_unexpected_socket_errorc C s4 t tj|dd}|| jjj_| j| d S )NrF rH )
r r, r r8 r: r; r rK rD rC )r rQ rR r r r rV s z AuthenticatorTest._fail_performc C s. d}t jtd}tjtj||d}|gS )Ns localhostzrsa512_key.pem)Zchallbdomainr? ) joseZJWKload test_utilZ load_vectorr Z"KeyAuthorizationAnnotatedChallenger
ZHTTP01_P)clsr[ keyZhttp_01r r r rC s z AuthenticatorTest._get_achallsc C s ddd| j jj_| j jd d | j jd ddg | j dg | j jj d ddi| j jj_| j dg | j dg | j jj
d d S ) NZserver1r" )r rI Zchall1Zchall2Zchall3r rI ) r: r; ZrunningrM ZservedaddupdateZcleanupr Zassert_called_once_withZassert_called_withr r r r test_cleanup s z AuthenticatorTest.test_cleanupc C s d| j _d | j _d| j _d S )NZ80z 127.0.0.1)r9 r6 Zhttp01_addressr r r r test_auth_hint s z AuthenticatorTest.test_auth_hintN)r/ r0 r1 r2 r r< r= rE r^ Zpatch_display_utilrT rW rN rY rZ rV classmethodrC rc rd r r r r r4 Q s"
r4 __main__r )#r2 rG r& systypingr r r Zunittestr Zjosepyr\ ZOpenSSL.cryptoZOpenSSLr* Zacmer r Zacme_standaloneZcertbotr r Z
certbot.testsr
r r^ ZTestCaser r3 r4 r/ exitmainargv__file__r r r r s, 1 n