a
}|gE @ sj d Z ddlZddlZddlmZ ddlmZ ddlZddlmZ ddlm Z ddl
m Z ddl
m Z ddl
m Z dd lmZ eed
d d ZG d
d dejZG dd dejZG dd dejZG dd deZG dd deZG dd deZG dd deZ G dd d e Z G d d d e Z G d d d ejZ d!d" Z e!d#krfe"e#ej$d$d e%g dS )%z"Tests for certbot._internal.hooks. N)python_version_tuple)mock)errors)util)
filesystem)os)List)majorminorc C s@ t t d }t t d }|| k r(dS || kr4dS ||k S d S )Nr TF)intr )r r
ZpymajorZpyminor r
E/usr/lib/python3.9/site-packages/certbot/_internal/tests/hook_test.pypyver_lt s r c @ s. e Zd ZdZedd Zeddd ZdS )ValidateHooksTestz1Tests for certbot._internal.hooks.validate_hooks.c O s ddl m} ||i |S )Nr )validate_hooks)certbot._internal.hooksr )clsargskwargsr r
r
r _call s zValidateHooksTest._callz%certbot._internal.hooks.validate_hookc C sR t }| | dd |jD }h dt|d d ks>J d|d ksNJ d S )Nc S s g | ]}|d d qS )r r r
).0callr
r
r
) z-ValidateHooksTest.test_it..> ZdeployZpostZprerenew)r Z MagicMockr call_args_listset)selfZmock_validate_hookconfigtypesr
r
r test_it$ s
zValidateHooksTest.test_itN) __name__
__module__ __qualname____doc__ classmethodr r patchr" r
r
r
r r s
r c @ sH e Zd ZdZedd Zdd Zeddd Z ed d
d Z
d S )
ValidateHookTestz0Tests for certbot._internal.hooks.validate_hook.c O s ddl m} ||i |S )Nr )
validate_hook)r r* )r r r r* r
r
r r 2 s zValidateHookTest._callc
C s t dv t jdddH ttj | dd W d n1 sJ0 Y W d n1 sh0 Y W d n1 s0 Y d S )N.certbot._internal.hooks.plug_util.path_surgeryz0certbot._internal.hooks.filesystem.is_executableF) return_valueZdummyfoo)r r( pytestraisesr HookCommandNotFoundr r r
r
r test_hook_not_executable7 s z)ValidateHookTest.test_hook_not_executablez'certbot._internal.hooks.util.exe_existsc C sv d|_ tdH}ttj | dd W d n1 s@0 Y W d n1 s^0 Y |jsrJ d S )NFr+ r- bar) r, r r( r. r/ r r0 r called)r Zmock_exe_existsZmock_psr
r
r test_not_foundA s
Hz ValidateHookTest.test_not_foundz certbot._internal.hooks._progc C s | d d |jdu sJ d S )Nr- F)r r4 )r Z mock_progr
r
r
test_unsetI s zValidateHookTest.test_unsetN) r# r$ r% r& r' r r2 r r( r5 r6 r
r
r
r r) / s
r) c @ s( e Zd ZdZedd Zedd ZdS )HookTestz!Common base class for hook tests.c O s t dS )z7Calls the method being tested with the given arguments.N)NotImplementedError)r r r r
r
r r R s zHookTest._callc O sD t d&}d|_| j|i | W d n1 s60 Y |S )zCalls self._call after mocking out certbot.compat.misc.execute_command_status.
The mock execute object is returned rather than the return value
of self._call.
*certbot.compat.misc.execute_command_statusr r; N)r r( r, r )r r r mock_executer
r
r _call_with_mock_executeW s .z HookTest._call_with_mock_executeN)r# r$ r% r& r' r r= r
r
r
r r7 O s
r7 c s e Zd ZdZedd Z fddZ fddZdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd Zdd Z ZS ) PreHookTestz+Tests for certbot._internal.hooks.pre_hook.c O s ddl m} ||i |S )Nr )pre_hook)r r? )r r r r? r
r
r r h s zPreHookTest._callc sJ t d| j_t| jj tj | jjd| _
t | j
| d S Nr- r3 )
supersetUpr r? r makedirsZrenewal_pre_hooks_dirr pathjoindir_hook create_hook_reset_pre_hook_alreadyr1 __class__r
r rB m s
zPreHookTest.setUpc s | t d S N)rH rA tearDownr1 rI r
r rL x s zPreHookTest.tearDownc C s ddl m} | d S )Nr )executed_pre_hooks)r rM clear)r rM r
r
r rH } s z#PreHookTest._reset_pre_hook_alreadyc C s d| j _| d S )Ncertonlyr verb_test_nonrenew_commonr1 r
r
r
test_certonly s zPreHookTest.test_certonlyc C s d| j _| d S )NrunrP r1 r
r
r test_run s zPreHookTest.test_runc C s. | | j}|jd| jjtjd | d S )Npre-hookenv)r= r assert_called_once_withr? r ANY_test_no_executions_commonr r< r
r
r rR s z!PreHookTest._test_nonrenew_commonc C st d | j _d| j _t| j td }| | j }W d n1 sH0 Y |j du s`J |j
j du spJ d S )Nr certbot._internal.hooks.loggerF) r r? rQ r removerF r r( r= r4 infor Z mock_loggerr< r
r
r
test_no_hooks s *zPreHookTest.test_no_hooksc C s6 d| j _| | j }|jd| j jtjd | d S )NFrV rW )r directory_hooksr= rY r? r rZ r[ r\ r
r
r test_renew_disabled_dir_hooks s z)PreHookTest.test_renew_disabled_dir_hooksc C sJ d| j _| | j }|jd| jtjd |jd| j jtjd | d S Nr rV rW )
r rQ r= assert_any_callrF r rZ assert_called_withr? r[ r\ r
r
r test_renew_no_overlap s
z!PreHookTest.test_renew_no_overlapc C s> | j | j_d| j_| | j}|jd| j tjd | d S rd ) rF r r? rQ r= rY r rZ r[ r\ r
r
r test_renew_with_overlap s
z#PreHookTest.test_renew_with_overlapc C sT t d }| | j}W d n1 s,0 Y |jdu sDJ |jjsPJ d S )Nr] F)r r( r= r r4 r_ r` r
r
r r[ s *z&PreHookTest._test_no_executions_common)r# r$ r% r& r' r rB rL rH rS rU rR ra rc rg rh r[
__classcell__r
r
rI r r> e s
r> c s e Zd ZdZedd Z fddZ fddZdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Z ZS ) PostHookTestz,Tests for certbot._internal.hooks.post_hook.c O s ddl m} ||i |S )Nr ) post_hook)r rk )r r r rk r
r
r r s zPostHookTest._callc sJ t d| j_t| jj tj | jjd| _
t | j
| d S )Nr3 r- )
rA rB r rk r rC Zrenewal_post_hooks_dirr rD rE rF rG _reset_post_hook_eventuallyr1 rI r
r rB s
zPostHookTest.setUpc s | t d S rK )rl rA rL r1 rI r
r rL s zPostHookTest.tearDownc C s ddl m} |d d = d S Nr )
post_hooksr rn r rn r
r
r rl s z(PostHookTest._reset_post_hook_eventuallyc C sF dD ]<}|| j _| | j g }|jd| j jtjd | rJ qd S )NrO rT post-hookrW )r rQ r= rY rk r rZ _get_eventually)r rQ r< r
r
r test_certonly_and_run_with_hook s
z,PostHookTest.test_certonly_and_run_with_hookc C s> d | j _dD ],}|| j _| | j g jr,J | r J q d S )Nrq )r rk rQ r= r4 rs )r rQ r
r
r #test_cert_only_and_run_without_hook s
z0PostHookTest.test_cert_only_and_run_without_hookc C s4 d| j _| | j dgj}|jd d dks0J d S )NrO success.orgrX RENEWED_DOMAINS)r rQ r= call_argsr )r r r
r
r test_renew_env s zPostHookTest.test_renew_envc C s d| j _| | j jg d S )NF)r rb _test_renew_commonrk r1 r
r
r rc s z*PostHookTest.test_renew_disabled_dir_hooksc C s d | j _| | jg d S rK )r rk rz rF r1 r
r
r test_renew_no_config_hook s z&PostHookTest.test_renew_no_config_hookc C s t | j | | jjg d S rK )r r^ rF rz r rk r1 r
r
r test_renew_no_dir_hook s z#PostHookTest.test_renew_no_dir_hookc C s" d | j _t| j | g d S rK )r rk r r^ rF rz r1 r
r
r test_renew_no_hooks s z PostHookTest.test_renew_no_hooksc C s< | j | jjg}| | d| j_|| jj | | d S )Nbaz)rF r rk rz append)r expectedr
r
r rg s
z"PostHookTest.test_renew_no_overlapc C s | j | j_| | j g d S rK )rF r rk rz r1 r
r
r rh s
z$PostHookTest.test_renew_with_overlapc C s8 d| j _tdD ]"}| | j g | |ksJ qd S )Nr )r rQ ranger rs )r r _r
r
r rz s z PostHookTest._test_renew_commonc C s ddl m} |S rm ro rp r
r
r rs s z PostHookTest._get_eventually)r# r$ r% r& r' r rB rL rl rt ru ry rc r{ r| r} rg rh rz rs ri r
r
rI r rj s"
rj c sT e Zd ZdZedd Zdd Z fddZdd Zd
d Z d d
Z
dd Z Z S )RunSavedPostHooksTestz7Tests for certbot._internal.hooks.run_saved_post_hooks.c O sF ddl m} d|v r |d n|d }d|v r4|d n|d }|||S )Nr )run_saved_post_hooksrenewed_domainsfailed_domainsr )r r )r r r r r r r
r
r r s zRunSavedPostHooksTest._callc O sH d}t j|| jd | j|i |W d S 1 s:0 Y dS )zCall run_saved_post_hooks but mock out execute and eventually
certbot._internal.hooks.post_hooks is replaced with
self.eventually. The mock execute object is returned rather than
the return value of run_saved_post_hooks.
z"certbot._internal.hooks.post_hooks)newN)r r(
eventuallyr= )r r r Zeventually_pathr
r
r &_call_with_mock_execute_and_eventually s z.execute_side_effectr9 N)r r( Z side_effectr )r r r r r< r
r r r= B s .z'RenewalHookTest._call_with_mock_executec s t dd dD | _d S )Nc S s h | ]}|t jvr|qS r
)r r )r varr
r
r _ s
z(RenewalHookTest.setUp..)rw r )rA rB
vars_to_clearr1 rI r
r rB ] s
zRenewalHookTest.setUpc s( | j D ]}tj|d qt d S rK )r r r poprA rL )r r rI r
r rL d s
zRenewalHookTest.tearDown)r# r$ r% r& r= rB rL ri r
r
rI r r = s r c @ sH e Zd ZdZedd Zeddd Zeddd Z d d
Z
d S ) DeployHookTestz.Tests for certbot._internal.hooks.deploy_hook.c O s ddl m} ||i |S )Nr ) deploy_hook)r r )r r r r r
r
r r m s zDeployHookTest._callr] c C s@ d| j _d| j _| | j dgd}|jdu s0J |jjs ddg}d}d| j _| | j ||}|jd| j jtjd d S )Nr example.netr r- deploy-hookrW )r r r= rY r rZ )r r r r< r
r
r test_success s zDeployHookTest.test_successN) r# r$ r% r& r' r r r( r r r r
r
r
r r j s
r c s^ e Zd ZdZedd Z fddZdd Ze dd d
Z
d d Z d
d Z dd Z
ZS )
RenewHookTestz,Tests for certbot._internal.hooks.renew_hookc O s ddl m} ||i |S )Nr )
renew_hook)r r )r r r r r
r
r r s zRenewHookTest._callc sB t d| j_t| jj tj | jjd| _
t | j
d S r@ ) rA rB r r r rC Zrenewal_deploy_hooks_dirr rD rE rF rG r1 rI r
r rB s
zRenewHookTest.setUpc C s4 d| j _| | j dgd}|jd| j jtjd d S )NFr r r rW )r rb r= rY r r rZ r\ r
r
r test_disabled_dir_hooks s
z%RenewHookTest.test_disabled_dir_hooksr] c C s< d| j _| | j dgd}|jdu s(J |jjdks8J d S )NTr r Fr )r r r= r4 r_ Z
call_countr` r
r
r r s
zRenewHookTest.test_dry_runc C sr d | j _t| j td"}| | j dgd}W d n1 sF0 Y |jdu s^J |j jdu snJ d S )Nr] r r F)
r r r r^ rF r r( r= r4 r_ r` r
r
r ra s
"zRenewHookTest.test_no_hooksc C s6 | j | j_| | jddgd}|jd| j tjd d S )Nr r r r rW )rF r r r= rY r rZ r\ r
r
r test_overlap s
zRenewHookTest.test_overlapc C s@ | | jdgd}|jd| jtjd |jd| jjtjd d S )Nr r r rW )r= r re rF r rZ rf r r\ r
r
r test_no_overlap s
z RenewHookTest.test_no_overlap)r# r$ r% r& r' r rB r r r( r ra r r ri r
r
rI r r s
r c @ s< e Zd ZdZedd Zdd Zdd Zdd Zd
d Z d S )
ListHooksTestz-Tests for certbot._internal.hooks.list_hooks.c O s ddl m} ||i |S )Nr )
list_hooks)r r )r r r r r
r
r r s zListHooksTest._callc C s | | jrJ d S rK )r tempdirr1 r
r
r r s zListHooksTest.test_emptyc s@ t fdddD }|D ] }t| q j|ks s z.ListHooksTest.test_multiple..r )sortedrG r r )r namesnamer
r1 r r s
zListHooksTest.test_multiplec C s2 t j| jd}t| | | j|gks.J d S )Nr- r rD rE r rG r r r r
r
r r s zListHooksTest.test_singlec C s0 t j| jd}t| | | jg ks,J d S )Nzfoo~r r r
r
r test_ignore_tilde s z ListHooksTest.test_ignore_tildeN)
r# r$ r% r& r' r r r r r r
r
r
r r s
r c C s t j| ddd dS )zmCreates an executable file at the specified path.
:param str file_path: path to create the file at
wi )modechmodN)r Z safe_openclose)Z file_pathr
r
r rG s rG __main__r )&r& sysZunittestplatformr r r. Zcertbotr r Zcertbot.compatr r Z
certbot.testsZ test_utiltypingr r r ZTestCaser ZTempDirTestCaser) ZConfigTestCaser7 r> rj r r r r r rG r# exitmainargv__file__r
r
r
r s2 NX2-"6"