a
}|g @ s d Z ddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddl
m Z ddl
m Z ddl
mZ dd l
mZ G d
d d ejZed kreeejd
d eg dS )z*Tests for certbot._internal.plugins.manual N)mock)
challenges)errors)
filesystem)os) acme_util)utilc sh e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
ZS )AuthenticatorTestz9Tests for certbot._internal.plugins.manual.Authenticator.c s t t }| | _| |j tj | _
tj | _ tj
| _| j
| j | jg| _dD ]}ttj| j| qVtjdd d ddtj| jdtj| jdtj| jdtj| jdtj| jdd
| _dd
lm} || jd d | _d S )
N)
config_dirwork_dirZ in_progressr Fr
r
backup_dirtemp_checkpoint_dirZ
in_progess)
Z http01_portmanual_auth_hookmanual_cleanup_hooknoninteractive_modevalidate_hooksr
r r r
Zin_progress_dir)
AuthenticatorZmanual)name) supersetUp test_utilZpatch_display_utilstartmock_get_displayZ
addCleanupstopr HTTP01_A http_achallDNS01_A
dns_achallZ DNS01_A_2Z dns_achall_2achallsr mkdirr pathjoinZtempdirr Z MagicMockconfigZ certbot._internal.plugins.manualr auth)selfZget_display_patchdr __class__ O/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/manual_test.pyr s.
zAuthenticatorTest.setUpc C sB d| j _ttj | j W d n1 s40 Y d S NT)r" r pytestraisesr Z PluginErrorr# preparer$ r( r( r) #test_prepare_no_hook_noninteractive0 s z5AuthenticatorTest.test_prepare_no_hook_noninteractivec C sT t jt j| j_d| j_tt j
| j W d n1 sF0 Y d S r* )
r r abspathsepr" r r r+ r, r ZHookCommandNotFoundr# r- r. r( r( r) test_prepare_bad_hook5 s z'AuthenticatorTest.test_prepare_bad_hookc C s t | j tsJ d S N)
isinstancer# Z more_infostrr. r( r( r) test_more_info; s z AuthenticatorTest.test_more_infoc C s | j dtjtjgks J d S )Nz example.org)r# Zget_chall_prefr HTTP01DNS01r. r( r( r) test_get_chall_pref> s
z%AuthenticatorTest.test_get_chall_prefc
C sz d tj| j_d | jjd| j| jjd dd | j
D t | j
| j
| j d }d | j
j| j
jd| j
| j
jd d d | j
D t | j
| j
| j
d }| j| j
d
d | j
D ksJ | jj| j d |ksJ | jj| j
d |ks
J | jjt | j
ks&J t| jjD ]>\}\}}t| jj| j
| d d
}||d v s6J q6d S )Na {0} -c "from certbot.compat import os;print(os.environ.get('CERTBOT_DOMAIN'));print(os.environ.get('CERTBOT_TOKEN', 'notoken'));print(os.environ.get('CERTBOT_VALIDATION', 'novalidation'));print(os.environ.get('CERTBOT_ALL_DOMAINS'));print(os.environ.get('CERTBOT_REMAINING_CHALLENGES'));"z{0}
{1}
{2}
{3}
{4}Znotoken,c s s | ] }|j V qd S r3 domain.0achallr( r( r) O z8AuthenticatorTest.test_script_perform.. tokenc s s | ] }|j V qd S r3 r; r= r( r( r) r@ T rA c S s g | ]}| |jqS r( Zresponse account_keyr= r( r( r)
X rA z9AuthenticatorTest.test_script_perform..CERTBOT_AUTH_OUTPUT r )formatsys
executabler" r r r<
validationrE r! r lenindexr challencoder# performenvr notification
call_count enumeratecall_args_listtextwrapindent)r$ Z dns_expectedZ
http_expectediargs_Zneedler( r( r) test_script_performB s:
z%AuthenticatorTest.test_script_performc C s | j | jdd | jD ks"J | jjt| jks\}\}}| j| }| |j
|d v szJ |d du sLJ qLd S )Nc S s g | ]}| |jqS r( rD r= r( r( r) rF f rA z9AuthenticatorTest.test_manual_perform..r wrapF) r# rQ r r rS rT rM rU rV rL rE )r$ rY rZ kwargsr? r( r( r) test_manual_performd s
z%AuthenticatorTest.test_manual_performc C s d tj| j_d| j_| j| j | jD ]}| j |g t
j d dksPJ t
j d |j ksdJ t
|jtjtjfrt
j d ||jksJ t
|jtjrt
j d |jdksJ q,dt
j vs,J q,d S ) Nz,{0} -c "import sys; sys.stdout.write('foo')"z # cleanuprG ZfooZCERTBOT_DOMAINZCERTBOT_VALIDATIONZ
CERTBOT_TOKENrC )rI rJ rK r" r r r# rQ r Zcleanupr environr< r4 rO r r7 r8 rL rE rP )r$ r? r( r( r) test_cleanupn s$
z AuthenticatorTest.test_cleanupc C s@ d| j _| jtjtjgdks$J | jtjgdks s