JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/hook_test.cpython-39.pyc
a
}|�g�E � @ 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� dej�ZG dd� dej�ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z G dd � d e �Z G d d � d e �Z G d d � d ej�Z d!d"� Z e!d#k�rfe�"e�#ej$d$d� e%g �� dS )%z"Tests for certbot._internal.hooks.� N)�python_version_tuple)�mock)�errors)�util)�
filesystem)�os)�List)�major�minorc 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.py�pyver_lt s r c @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�ValidateHooksTestz1Tests for certbot._internal.hooks.validate_hooks.c O s ddl m} ||i |��S )Nr )�validate_hooks)�certbot._internal.hooksr )�cls�args�kwargsr r
r
r �_call s zValidateHooksTest._callz%certbot._internal.hooks.validate_hookc C sR t �� }| �|� dd� |jD �}h d�t|d d� �ks>J �d|d ksNJ �d S )Nc S s g | ]}|d d �qS )r r r
)�.0�callr
r
r �
<listcomp>) � z-ValidateHooksTest.test_it.<locals>.<listcomp>> ZdeployZpostZpre����renew)r Z MagicMockr �call_args_list�set)�selfZmock_validate_hook�config�typesr
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� Ze�d�dd� �Z e�d �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 �d��v t jddd��H t�tj�� | �dd� W d � n1 sJ0 Y W d � n1 sh0 Y W d � n1 s�0 Y d S )N�.certbot._internal.hooks.plug_util.path_surgeryz0certbot._internal.hooks.filesystem.is_executableF)� return_valueZdummy�foo)r r( �pytest�raisesr �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|_ t�d��H}t�tj�� | �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 )z�Calls 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_status�r � 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� fdd�Z� fdd�Zdd � 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 )
�super�setUpr r? r �makedirsZrenewal_pre_hooks_dirr �path�join�dir_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 )N�certonly�r �verb�_test_nonrenew_commonr1 r
r
r �
test_certonly� s zPreHookTest.test_certonlyc C s d| j _| �� d S )N�runrP r1 r
r
r �test_run� s zPreHookTest.test_runc C s. | � | j�}|jd| jjtjd� | �� d S )N�pre-hook��env)r= r �assert_called_once_withr? r �ANY�_test_no_executions_common�r r< r
r
r rR � s z!PreHookTest._test_nonrenew_commonc C st d | j _d| j _t�| j� t�d�� }| �| 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 �info�r 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� fdd�Z� fdd�Zdd � 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_hooks�r 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 )N�rO 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 dg�j}|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 )N�baz)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 _td�D ]"}| �| 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� fdd�Zdd � 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_hooks�renewed_domains�failed_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 )z�Call 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<RunSavedPostHooksTest._call_with_mock_execute_and_eventuallyc s t � �� g | _d S rK )rA rB r� r1 rI r
r rB " s
zRunSavedPostHooksTest.setUpc C s | � g g �jrJ �d S rK )r� r4 r1 r
r
r �
test_empty& s z RunSavedPostHooksTest.test_emptyc C sJ g d�| _ | �g g �}|j}t|| j �D ] \}}|d d |ks(J �q(d S )N�r- r3 r~ Zquxr r )r� r� r �zip)r r< ZcallsZ actual_callZ expected_argr
r
r �
test_multiple) s
z#RunSavedPostHooksTest.test_multiplec C s0 dg| _ | �g g �}|jd| j d tjd� d S )Nr- rr r rW )r� r� rY r rZ r\ r
r
r � test_single1 s z!RunSavedPostHooksTest.test_singlec C sL dg| _ | �dgdg�}|jjd d dks0J �|jjd d dksHJ �d S )Nr- rv z
failed.orgrX rw ZFAILED_DOMAINS)r� r� rx r r\ r
r
r �test_env6 s z RunSavedPostHooksTest.test_env)
r# r$ r% r& r' r r� rB r� r� r� r� ri r
r
rI r r� s
r� c s4 e Zd ZdZdd� Z� fdd�Z� fdd�Z� ZS )�RenewalHookTestz1Common base class for testing deploy/renew hooks.c s� d|v r|d n|d � d|v r(|d n|d �� �fdd�}t �d��&}||_| j|i |�� W d � n1 st0 Y |S ) a Calls 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. The mock execute object asserts that environment
variables were properly set.
�domainsr �lineager� c s. t jd d�� �ksJ �t jd �ks*J �dS )z�Assert environment variables are properly set.
:returns: two strings imitating no output from the hook
:rtype: `tuple` of `str`
rw � �RENEWED_LINEAGEr: )r �environrE )Z unused_argsZ
unused_kwargs�r� r� r
r �execute_side_effectM s zDRenewalHookTest._call_with_mock_execute.<locals>.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 � <setcomp>_ s
�z(RenewalHookTest.setUp.<locals>.<setcomp>)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� �Ze�d�dd� �Ze�d�dd� �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<J �d S )Nr- T� example.org�/foo/barF)r r� �dry_runr= r4 r_ r` r
r
r � test_dry_runr s
�zDeployHookTest.test_dry_runc C s< d | j _| �| j dgd�}|jdu s(J �|jjdu s8J �d S )Nr� r� F)r r� r= r4 r_ r` r
r
r � test_no_hook{ s
�zDeployHookTest.test_no_hookc C s> 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� fdd�Zdd� Ze� d�d 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� t�d��"}| �| 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 | � | j�rJ �d S rK )r �tempdirr1 r
r
r r� � s zListHooksTest.test_emptyc s@ t � fdd�dD ��}|D ] }t|� q� �� j�|ks<J �d S )Nc 3 s | ]}t j�� j|�V qd S rK )r rD rE r� )r �basenamer1 r
r � <genexpr>� s �z.ListHooksTest.test_multiple.<locals>.<genexpr>r� )�sortedrG r r� )r �names�namer
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|� | �| j�g 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� )�mode�chmodN)r Z safe_open�close)Z file_pathr
r
r rG � s rG �__main__r )&r& �sysZunittest�platformr r r. Zcertbotr r Zcertbot.compatr r Z
certbot.testsZ test_util�typingr r r ZTestCaser ZTempDirTestCaser) ZConfigTestCaser7 r> rj r� r� r� r� r� rG r# �exit�main�argv�__file__r
r
r
r �<module> s2 NX2-"6"
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg