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
certshttp_01_resourcesmgr)selfr
r S/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/standalone_test.pysetUp s zServerManagerTest.setUpc C s( | j j| ju sJ | j j| ju s$J d S N)r r r r r r r test_init s zServerManagerTest.test_initc C sZ | j jd|d}| d d }| j ||iks6J | j j|d | j i ksVJ d S )Nr portchallenge_type )r )r run getsocknamesrunningstop)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 sx | j jdtjd}| d d }| j j|tjd}| j ||iksJJ ||u sVJ | j | | j i kstJ d S )Nr r 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 | j
i ksJ | | d S )N r r r( )r )socketZAF_INET6bind getsocknameOSErrorpytestraisesr StandaloneBindErrorr r 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& r1 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+ r0 )Z open_socketr r r r
get_open_portH s
r6 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 r8 r MagicMockr6 configauthservers)r r8 r r r r T s zAuthenticatorTest.setUpc C s t | j tsJ d S r )
isinstancer= Z more_infostrr r r r test_more_info[ s z AuthenticatorTest.test_more_infoc C s | j jd dtjgksJ d S )N)domain)r= Zget_chall_prefr r# r r r r test_get_chall_pref^ s z%AuthenticatorTest.test_get_chall_prefc C s2 | }| j|}dd |D }||ks.J d S )Nc S s g | ]}| |jqS r )response account_key).0Zachallr r r
f z2AuthenticatorTest.test_perform..) _get_achallsr= perform)r ZachallsrD 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)rM
EADDRINUSEr r/ r r; r= r> r side_effectyesno return_valuerK _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)
rR rS rM rP r- r. r PluginError
_fail_performrT )r rU rV rY rW r r r test_perform_eaddrinuse_no_retryu s (z2AuthenticatorTest.test_perform_eaddrinuse_no_retryc C s. |j \}}d|d v sJ |ddr*J d S )Nzin user defaultT)Z call_argsget)r rY Z
yesno_argsZ yesno_kwargsr r r rT s
z,AuthenticatorTest._assert_correct_yesno_callc C s@ t j}ttj | | W d n1 s20 Y d S r )rM ZEACCESr- r. r r[ r\ r rW 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 )rM ZENOTCONNr- r. r r/ r\ r` r r r $test_perform_unexpected_socket_error s z6AuthenticatorTest.test_perform_unexpected_socket_errorc C s4 t tj|dd}|| jjj_| j| d S )NrL rN )
r r/ r r; r= r> r rQ rJ rI )r rW rX r r r r\ s z AuthenticatorTest._fail_performc C s. d}t jtd}tjtj||d}|gS )Ns localhostzrsa512_key.pem)ZchallbrB rE ) joseZJWKload test_utilZ load_vectorr Z"KeyAuthorizationAnnotatedChallenger
ZHTTP01_P)clsrB keyZhttp_01r r r rI s z AuthenticatorTest._get_achallsc C s ddd| j jj_| j jd d | j jd ddg | j dg | j jt ddhdksdJ | j jj
d d di| j jj_| j dg | j jt dhdksJ d| j jj j ksJ | j dg | j jt t dksJ | j jj d d S )
Nserver1r% )r rO Zchall1Zchall2Zchall3)rh r% r rO )
r= r> r rS ZservedaddupdateZcleanupsetr Zassert_called_once_withZ
call_countZassert_called_withr r r r test_cleanup s,
z AuthenticatorTest.test_cleanupc C sD d| j _d | j _d| jg v s$J d| j _d| jg v s@J d S )NZ80z
on port 80z 127.0.0.1zon 127.0.0.1:80)r< r9 Zhttp01_addressr= Z auth_hintr r r r test_auth_hint s
z AuthenticatorTest.test_auth_hintN)r2 r3 r4 r5 r rA rC rK re Zpatch_display_utilrZ r] rT ra rb r\ classmethodrI rl rm r r r r r7 Q s"
r7 __main__r )#r5 rM r) systypingr r r Zunittestr Zjosepyrc ZOpenSSL.cryptoZOpenSSLr- Zacmer r Zacme_standaloneZcertbotr r Z
certbot.testsr
r re ZTestCaser r6 r7 r2 exitmainargv__file__r r r r s, 1 n