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 ZHTTP01_A http_achallZDNS01_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 d S Nr& r, r& r& r' test_more_info; s z AuthenticatorTest.test_more_infoc C s d S r1 r& r, r& r& r' test_get_chall_pref> s z%AuthenticatorTest.test_get_chall_prefc
C s 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 }t| jjD ]*\}\}}t| jj| j
| d
d }qd 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 r1 domain.0achallr& r& r' O z8AuthenticatorTest.test_script_perform.. tokenc s s | ] }|j V qd S r1 r5 r7 r& r& r' r: T r; ZCERTBOT_AUTH_OUTPUT )formatsys
executabler r r r6 Z
validationZ account_keyr r lenindexr challencode enumerater notificationcall_args_listtextwrapindentr! env)r" Z dns_expectedZ
http_expectediargs_Zneedler& r& r' test_script_performB s& z%AuthenticatorTest.test_script_performc C s, t | jjD ]\}\}}| j| }qd S r1 )rF r rG rH r )r" rL rM kwargsr9 r& r& r' test_manual_performd s
z%AuthenticatorTest.test_manual_performc C sh d tj| j_d| j_| j| j | jD ]6}| j |g t
|j t j
t jfrRt
|j t j
r,q,q,d S )Nz,{0} -c "import sys; sys.stdout.write('foo')"z # cleanup)r? r@ rA r r r r! Zperformr Zcleanup
isinstancerD r ZHTTP01ZDNS01)r" r9 r& r& r' test_cleanupn s
z AuthenticatorTest.test_cleanupc C s d| j _d S )Nz /bin/true)r r r, r& r& r' test_auth_hint_hook s z%AuthenticatorTest.test_auth_hint_hookc C s d S r1 r& r, r& r& r' test_auth_hint_no_hook s z(AuthenticatorTest.test_auth_hint_no_hook)__name__
__module__ __qualname____doc__r r- r0 r2 r3 rO rQ rS rT rU
__classcell__r& r& r$ r' r s "
r __main__r< )rY r@ rI Zunittestr r) Zacmer Zcertbotr Zcertbot.compatr r Z
certbot.testsr r r ZTempDirTestCaser rV exitmainargv__file__r& r& r& r' s