|
Server : Apache/2.4.41 (Ubuntu) System : Linux vmi1525618.contaboserver.net 5.4.0-105-generic #119-Ubuntu SMP Mon Mar 7 18:49:24 UTC 2022 x86_64 User : www-data ( 33) PHP Version : 8.2.12 Disable Function : NONE Directory : /lib/python3/dist-packages/certbot/tests/__pycache__/ |
Upload File : |
U
�]�E � @ s4 d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddl mZ G dd � d ej
�ZG d
d � d e j�ZG d d
� d
e j�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� dej
�ZG d d � d e j�Zd d � Ze d k�r0e� � dS )!zTests for certbot.hooks.� N)�List)�errors)�util)�os)�
filesystemc @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�ValidateHooksTestz'Tests for certbot.hooks.validate_hooks.c O s ddl m} |||�S )Nr )�validate_hooks)�
certbot.hooksr )�cls�args�kwargsr � r
�9/usr/lib/python3/dist-packages/certbot/tests/hook_test.py�_call s zValidateHooksTest._callzcertbot.hooks.validate_hookc C sR t �� }| �|� dd� |jD �}| �td�t|d d� �� | �d|d � d S )Nc S s g | ]}|d d �qS )r � r
)�.0Zcallr
r
r �
<listcomp> s z-ValidateHooksTest.test_it.<locals>.<listcomp>)ZpreZpostZdeploy����renew)�mockZ MagicMockr �call_args_list� assertEqual�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 )
�ValidateHookTestz&Tests for certbot.hooks.validate_hook.c O s ddl m} |||�S )Nr )�
validate_hook)r r$ )r
r r r$ r
r
r r $ s zValidateHookTest._callc
C sH t �d��4 t jddd�� | �tj| jdd� W 5 Q R X W 5 Q R X d S )N�$certbot.hooks.plug_util.path_surgeryz&certbot.hooks.filesystem.is_executableF)� return_valueZdummy�foo)r r" � assertRaisesr �HookCommandNotFoundr �r r
r
r �test_hook_not_executable) s z)ValidateHookTest.test_hook_not_executablez certbot.hooks.util.exe_existsc C s@ d|_ t�d��}| �tj| jdd� W 5 Q R X | �|j� d S )NFr% r' �bar) r& r r" r( r r) r �
assertTrue�called)r Zmock_exe_existsZmock_psr
r
r �test_not_found2 s z ValidateHookTest.test_not_foundzcertbot.hooks._progc C s | � d d� | �|j� d S �Nr' )r � assertFalser. )r Z mock_progr
r
r �
test_unset9 s zValidateHookTest.test_unsetN) r r r r r! r r+ r r" r/ r2 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 B s zHookTest._callc O s, t �d��}d|_| j||� W 5 Q R X |S )z�Calls self._call after mocking out certbot.hooks.execute.
The mock execute object is returned rather than the return value
of self._call.
�certbot.hooks.execute�� r7 )r r" r& r )r
r r � mock_executer
r
r �_call_with_mock_executeG s z HookTest._call_with_mock_executeN)r r r r r! r r9 r
r
r
r r3 ? s
r3 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.hooks.pre_hook.c O s ddl m} |||�S )Nr )�pre_hook)r r; )r
r r r; r
r
r r X s zPreHookTest._callc sN t t| ��� d| j_t�| jj� tj �
| jjd�| _ t | j � | �
� d S �Nr' r, )�superr: �setUpr r; r �makedirsZrenewal_pre_hooks_dirr �path�join�dir_hook� create_hook�_reset_pre_hook_alreadyr* �� __class__r
r r> ] s
zPreHookTest.setUpc s | � � tt| ��� d S �N)rD r= r: �tearDownr* rE r
r rH h s zPreHookTest.tearDownc C s ddl m} |�� d S )Nr )�executed_pre_hooks)r rI �clear)r rI r
r
r rD m s z#PreHookTest._reset_pre_hook_alreadyc C s d| j _| �� d S )N�certonly�r �verb�_test_nonrenew_commonr* r
r
r �
test_certonlyq s zPreHookTest.test_certonlyc C s d| j _| �� d S )N�runrL r* r
r
r �test_runu s zPreHookTest.test_runc C s( | � | j�}|�d| jj� | �� d S )N�pre-hook)r9 r �assert_called_once_withr; �_test_no_executions_common�r r8 r
r
r rN y s z!PreHookTest._test_nonrenew_commonc C s\ d | j _d| j _t�| j� t�d��}| �| j �}W 5 Q R X | � |j
� | � |j j
� d S )Nr �certbot.hooks.logger) r r; rM r �removerB r r" r9 r1 r. �info�r � mock_loggerr8 r
r
r �
test_no_hooks~ s zPreHookTest.test_no_hooksc C s0 d| j _| �| j �}|�d| j j� | �� d S )NFrR )r �directory_hooksr9 rS r; rT rU r
r
r � test_renew_disabled_dir_hooks� s z)PreHookTest.test_renew_disabled_dir_hooksc C s> d| j _| �| j �}|�d| j� |�d| j j� | �� d S �Nr rR )r rM r9 �assert_any_callrB �assert_called_withr; rT rU r
r
r �test_renew_no_overlap� s
z!PreHookTest.test_renew_no_overlapc C s8 | j | j_d| j_| �| j�}|�d| j � | �� d S r^ )rB r r; rM r9 rS rT rU r
r
r �test_renew_with_overlap� s
z#PreHookTest.test_renew_with_overlapc C s@ t �d��}| �| j�}W 5 Q R X | �|j� | �|jj� d S )NrV )r r" r9 r r1 r. r- rX rY r
r
r rT � s z&PreHookTest._test_no_executions_common)r r r r r! r r> rH rD rO rQ rN r[ r] ra rb rT �
__classcell__r
r
rE r r: U 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 � Z� ZS ) � PostHookTestz"Tests for certbot.hooks.post_hook.c O s ddl m} |||�S )Nr )� post_hook)r re )r
r r re r
r
r r � s zPostHookTest._callc sN t t| ��� d| j_t�| jj� tj �
| jjd�| _ t | j � | �
� d S )Nr, r' )r= rd r> r re r r? Zrenewal_post_hooks_dirr r@ rA rB rC �_reset_post_hook_eventuallyr* rE r
r r> � s
zPostHookTest.setUpc s | � � tt| ��� d S rG )rf r= rd rH r* rE r
r rH � s zPostHookTest.tearDownc C s ddl m} |d d �= d S �Nr )�
post_hooks�r rh �r rh r
r
r rf � s z(PostHookTest._reset_post_hook_eventuallyc C s@ dD ]6}|| j _| �| j �}|�d| j j� | �| �� � qd S )N�rK rP � post-hook)r rM r9 rS re r1 �_get_eventually)r rM r8 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 �j� | �| �� � q d S )Nrk )r re rM r1 r9 r. rm )r rM r
r
r �#test_cert_only_and_run_without_hook� s
z0PostHookTest.test_cert_only_and_run_without_hookc C s d| j _| �| j jg� d S )NF)r r\ �_test_renew_commonre r* r
r
r r] � s z*PostHookTest.test_renew_disabled_dir_hooksc C s d | j _| �| jg� d S rG )r re rp rB r* r
r
r �test_renew_no_config_hook� s z&PostHookTest.test_renew_no_config_hookc C s t �| j� | �| jjg� d S rG )r rW rB rp r re r* 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 rG )r re r rW rB rp r* 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)rB r re rp �append)r �expectedr
r
r ra � s
z"PostHookTest.test_renew_no_overlapc C s | j | j_| �| j g� d S rG )rB r re rp r* r
r
r rb � s
z$PostHookTest.test_renew_with_overlapc C s6 d| j _td�D ] }| �| j � | �| �� |� qd S )Nr � )r rM �ranger r rm )r rv �_r
r
r rp � s z PostHookTest._test_renew_commonc C s ddl m} |S rg ri rj r
r
r rm � s z PostHookTest._get_eventually)r r r r r! r r> rH rf rn ro r] rq rr rs ra rb rp rm rc r
r
rE r rd � s
rd c sL e Zd ZdZedd� �Zdd� Z� fdd�Zdd � Zd
d � Z d d
� Z
� Z S )�RunSavedPostHooksTestz-Tests for certbot.hooks.run_saved_post_hooks.c O s ddl m} |� S )Nr )�run_saved_post_hooks)r r{ )r
r r r{ r
r
r r � s zRunSavedPostHooksTest._callc
O s8 d}t j|| jd�� | j||�W 5 Q R � S Q R X dS )z�Call run_saved_post_hooks but mock out execute and eventually
certbot.hooks.post_hooks is replaced with
self.eventually. The mock execute object is returned rather than
the return value of run_saved_post_hooks.
zcertbot.hooks.post_hooks)�newN)r r" �
eventuallyr9 )r r r Zeventually_pathr
r
r �&_call_with_mock_execute_and_eventually� s z<RunSavedPostHooksTest._call_with_mock_execute_and_eventuallyc s t t| ��� g | _d S rG )r= rz r> r} r* rE r
r r>
s zRunSavedPostHooksTest.setUpc C s | � | �� j� d S rG )r1 r~ r. r* r
r
r �
test_empty s z RunSavedPostHooksTest.test_emptyc C sJ ddddg| _ | �� }|j}t|| j �D ] \}}| �|d d |� q(d S )Nr' r, rt �quxr r )r} r~ r �zipr )r r8 ZcallsZ actual_callZ expected_argr
r
r �
test_multiple s
z#RunSavedPostHooksTest.test_multiplec C s& dg| _ | �� }|�d| j d � d S )Nr' rl r )r} r~ rS rU r
r
r � test_single s z!RunSavedPostHooksTest.test_single) r r r r r! r r~ r> r r� r� rc r
r
rE r rz � s
rz c s4 e Zd ZdZdd� Z� fdd�Z� fdd�Z� ZS )�RenewalHookTestz1Common base class for testing deploy/renew hooks.c sl d|kr|d n|d � d|kr(|d n|d �� ��fdd�}t �d��}||_�j||� W 5 Q R X |S )z�Calls self._call after mocking out certbot.hooks.execute.
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 �lineagerw c s. �� tjd d�� �� �� tjd �� dS )z�Assert environment variables are properly set.
:returns: two strings imitating no output from the hook
:rtype: `tuple` of `str`
�RENEWED_DOMAINS� �RENEWED_LINEAGEr6 )r r �environrA )Z unused_argsZ
unused_kwargs�r� r� r r
r �execute_side_effect/ s zDRenewalHookTest._call_with_mock_execute.<locals>.execute_side_effectr5 )r r" Z side_effectr )r r r r� r8 r
r� r r9 $ s z'RenewalHookTest._call_with_mock_executec s& t t| ��� tdd� dD ��| _d S )Nc s s | ]}|t jkr|V qd S rG )r r� )r �varr
r
r � <genexpr>A s
�z(RenewalHookTest.setUp.<locals>.<genexpr>)r� r� )r= r� r> r �
vars_to_clearr* rE r
r r> ? s �zRenewalHookTest.setUpc s, | j D ]}tj�|d � qtt| ��� d S rG )r� r r� �popr= r� rH )r r� rE r
r rH E s
zRenewalHookTest.tearDown)r r r r r9 r> rH rc r
r
rE 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.hooks.deploy_hook.c O s ddl m} |||�S )Nr )� deploy_hook)r r� )r
r r r� r
r
r r N s zDeployHookTest._callrV c C s@ d| j _d| j _| �| j dgd�}| �|j� | �|jj� d S )Nr' T� example.org�/foo/bar)r r� �dry_runr9 r1 r. r- �warningrY r
r
r � test_dry_runS s � zDeployHookTest.test_dry_runc C s8 d | j _| �| j dgd�}| �|j� | �|jj� d S )Nr� r� )r r� r9 r1 r. rX rY r
r
r � test_no_hook\ s � zDeployHookTest.test_no_hookc C s8 ddg}d}d| j _| �| j ||�}|�d| j j� d S )Nr� � example.netr� r' � deploy-hook)r r� r9 rS )r r� r� r8 r
r
r � test_successd s �zDeployHookTest.test_successN) r r r r r! r r r" r� r� r� r
r
r
r r� K 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.hooks.renew_hookc O s ddl m} |||�S )Nr )�
renew_hook)r r� )r
r r r� r
r
r r p s zRenewHookTest._callc sF t t| ��� d| j_t�| jj� tj �
| jjd�| _ t | j � d S r< )
r= r� r> r r� r r? Zrenewal_deploy_hooks_dirr r@ rA rB rC r* rE r
r r> u s �zRenewHookTest.setUpc C s. d| j _| �| j dgd�}|�d| j j� d S )NFr� r� r� )r r\ r9 rS r� rU r
r
r �test_disabled_dir_hooks~ s �z%RenewHookTest.test_disabled_dir_hooksrV c C s: d| j _| �| j dgd�}| �|j� | �|jjd� d S )NTr� r� rw )r r� r9 r1 r. r r� Z
call_countrY r
r
r r� � s � zRenewHookTest.test_dry_runc C sZ d | j _t�| j� t�d��}| �| j dgd�}W 5 Q R X | �|j � | �|j
j � d S )NrV r� r� ) r r� r rW rB r r" r9 r1 r. rX rY r
r
r r[ � s � zRenewHookTest.test_no_hooksc C s0 | j | j_| �| jddgd�}|�d| j � d S )Nr� r� r� r� )rB r r� r9 rS rU r
r
r � test_overlap� s
�zRenewHookTest.test_overlapc C s4 | � | jdgd�}|�d| j� |�d| jj� d S )Nr� r� r� )r9 r r_ rB r` r� rU r
r
r �test_no_overlap� s �z RenewHookTest.test_no_overlap)r r r r r! r r> r� r r" r� r[ r� r� rc r
r
rE r r� m s
r� c @ s, e Zd ZdZedd� �Zdd� Zdd� ZdS ) � ExecuteTestz Tests for certbot.hooks.execute.c O s ddl m} |||�S )Nr )�execute)r r� )r
r r r� r
r
r r � s zExecuteTest._callc C s6 t dd�D ]&}dD ] }dD ]}| �|||� qqq
d S )Nr rw )r7 z Hello World!)r7 zGoodbye Cruel World!)rx � _test_common)r �
returncode�stdout�stderrr
r
r r � s zExecuteTest.test_itc
C s� d}d}t �d��J}||f|jj_||j_t �d�� }| �| �||�||f� W 5 Q R X W 5 Q R X |jd �d|jd d �}| �||� |j �
d||� |r�|j �
t j t j t j |� |s�|r�| � |j
j� d S ) Nr' zfoo-hookzcertbot.hooks.PopenrV r r r zRunning %s command: %s)r r" r& Z communicater� r r Z call_args�getrX r_ �ANYr- �errorr. ) r r� r� r� Z
given_commandZ
given_nameZ
mock_popenrZ Zexecuted_commandr
r
r r� � s. ,
� � �zExecuteTest._test_commonN)r r r r r! r r r� r
r
r
r r� � s
r� c @ s< e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d S )
�
ListHooksTestz#Tests for certbot.hooks.list_hooks.c O s ddl m} |||�S )Nr )�
list_hooks)r r� )r
r r r� r
r
r r � s zListHooksTest._callc C s | � | �| j�� d S rG )r1 r �tempdirr* r
r
r r � s zListHooksTest.test_emptyc s@ t � fdd�dD ��}|D ] }t|� q� �� �� j�|� d S )Nc 3 s | ]}t j�� j|�V qd S rG )r r@ rA r� )r �basenamer* r
r r� � s �z.ListHooksTest.test_multiple.<locals>.<genexpr>)r' r, rt r� )�sortedrC r r r� )r �names�namer
r* r r� � s �
zListHooksTest.test_multiplec C s2 t j�| jd�}t|� | �| �| j�|g� d S r0 �r r@ rA r� rC r r �r r� r
r
r r� � s zListHooksTest.test_singlec C s0 t j�| jd�}t|� | �| �| j�g � 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 rC � s rC �__main__) r Zunittestr Zacme.magic_typingr Zcertbotr r Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser ZTempDirTestCaser# ZConfigTestCaser3 r: rd rz r� r� r� r� r� rC r �mainr
r
r
r �<module> s, NS),"6$"