|
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
�]cH � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl
m Z ddl
mZ G dd� dej�ZG d d
� d
ej�Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zedkr�e�� dS )zTest certbot.reverter.� N)�errors)�os)�
filesystem)�utilc s� e Zd ZdZ� fdd�Zdd� Ze�d�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Ze�d �d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Ze�d*�d+d,� �Ze�d-�d.d/� �Zd0d1� Z� Z S )2�ReverterCheckpointLocalTestzTest the Reverter Class.c sT t t| ��� ddlm} t�tj� || j�| _ t
� }|\| _ | _ | _
| _| _d S �Nr )�Reverter)�superr �setUp�certbot.reverterr �logging�disable�CRITICAL�config�reverter�setup_test_files�config1�config2�dir1�dir2�sets��selfr �tup�� __class__� �=/usr/lib/python3/dist-packages/certbot/tests/reverter_test.pyr
s z!ReverterCheckpointLocalTest.setUpc C s6 t �| jj� t �| j� t �| j� t�tj� d S �N� �shutilZrmtreer Zwork_dirr r r r
ZNOTSET�r r r r �tearDown s z$ReverterCheckpointLocalTest.tearDownz*certbot.reverter.Reverter._read_and_appendc C s� t d�|_z| j�| jd d� W n t k
r6 Y nX | j�d� t�| jjj �d }tj
� | jjj |d�}t |d��}|�
� }W 5 Q R X | �d|k� d S )Nz cannot evenr �save1Zblah�
CHANGES_SINCE�rz
No changes)�OSError� side_effectr �add_to_checkpointr �finalize_checkpointr �listdirr �
backup_dir�path�join�open�read�
assertTrue)r Z mock_readr, Z no_change�f�xr r r �test_no_change' s
z*ReverterCheckpointLocalTest.test_no_changec C s� | j �| jd d� | j �| jd d� | �tj�| jj�� | � t
| jj�d� | � tj� tj�
| jjd��� | � t| jj�d�| j| j�� d S )Nr r# � �save2Z
save1save2� NEW_FILESz{0}
{1}
)r �add_to_temp_checkpointr r0 r r, �isdirr �temp_checkpoint_dir� assertEqual�get_save_notes� assertFalse�isfiler- �
get_filepaths�formatr r r! r r r �!test_basic_add_to_temp_checkpoint5 s ��
�
�z=ReverterCheckpointLocalTest.test_basic_add_to_temp_checkpointc C s@ t �d��,}td�|_| �tj| jj| j d d� W 5 Q R X d S )N� certbot.reverter.shutil.copy2�bad copyr r# )
�mock�patch�IOErrorr' � assertRaisesr �
ReverterErrorr r( r �r Z
mock_copy2r r r �#test_add_to_checkpoint_copy_failureD s
�z?ReverterCheckpointLocalTest.test_add_to_checkpoint_copy_failurec C s� t j�| jd�}| j�d|� t|d� | j�| jd d� | j� | jd d� | �
t j | jj| jd d� | j�| jd d
� | �
t j | jjt
|g�d � d S )
z:Make sure that checkpoint errors are thrown appropriately.� config3.txtTzThis is a new file!� r# r r5 Zsave3r4 Zsave4z invalid saveN)r r, r- r r �register_file_creation� update_filer( r r7 rF r rG �set�r �config3r r r �test_checkpoint_conflictK s
� �z4ReverterCheckpointLocalTest.test_checkpoint_conflictc C s` | j �| jd d� t| jd� | j �| jd d� t| jd� | j �� | �t| j�d� d S )Nr r# zupdated-directivezsave2-updated dirz'new directive change that we won't keep�directive-dir1)r r7 r rM r �revert_temporary_configr: �read_inr! r r r �#test_multiple_saves_and_temp_revert^ s
z?ReverterCheckpointLocalTest.test_multiple_saves_and_temp_revertc C s� t j�| jd�}t|d� t j�| jd�}t|d� | j�d| j� | j�d| j � | j�d||� | j�
� | � t j� | j�� | � t j� | j �� | � t j� |�� | � t j� |�� d S )NrJ ZConfig3� config4.txtZConfig4T)
r r, r- r rM r r rL r r �recovery_routiner< r= �r rP Zconfig4r r r �*test_multiple_registration_fail_and_revertg s
zFReverterCheckpointLocalTest.test_multiple_registration_fail_and_revertc C s` | j �d| j� | j �d| j� | j �d| j� | j �d| j� t| jj�}| �t|�d� d S )NTr4 )r rL r �
get_new_filesr r9 r: �len)r �filesr r r �$test_multiple_registration_same_file{ s z@ReverterCheckpointLocalTest.test_multiple_registration_same_filec C sJ t �� }t jd|dd��( td�|_| �tj| jj d| j
� W 5 Q R X d S )N�certbot.reverter.openT�Zcreate�bad open) rC � mock_openrD r& r' rF r rG r rL r �r Zm_openr r r �'test_register_file_creation_write_error� s
�zCReverterCheckpointLocalTest.test_register_file_creation_write_errorc C s | � tj| jjd� d S )N�filepath)rF r rG r rL r! r r r �test_bad_registration� s
�z1ReverterCheckpointLocalTest.test_bad_registrationc C s` ddgddgdgg}|D ]}| j �d|� qt| jj�}tj�||�D ]\}}| �||� qFd S )N�a2dismod�sslZrewriteZ
cleanslateT) r �register_undo_command�get_undo_commandsr r9 �sixZmoves�zipr: )r �coms�comZact_comsZa_comr r r �test_register_undo_command� s � z6ReverterCheckpointLocalTest.test_register_undo_commandc C sJ t �� }t jd|dd��( td�|_| �tj| jj ddg� W 5 Q R X d S )Nr^ Tr_ r` Zcommand)
rC ra rD r& r' rF r rG r rh rb r r r � test_bad_register_undo_command� s
�z:ReverterCheckpointLocalTest.test_bad_register_undo_commandzcertbot.util.run_scriptc C sN dt jg|_dgddgg}|D ]}| j�d|� q | j�� | �|jd� d S )N� Zinvalid_commandrf rg TrK )r ZSubprocessErrorr' r rh rS r: �
call_count)r Zmock_runrl rm r r r �test_run_undo_commands� s �
z2ReverterCheckpointLocalTest.test_run_undo_commandsc C s< | j �| jd d� tjtjd�| j _| �tj| j j � d S )Nr � perm save�r' )
r r( r rC � MagicMockr rG �_recover_checkpointrF rW r! r r r �)test_recovery_routine_in_progress_failure� s
�
zEReverterCheckpointLocalTest.test_recovery_routine_in_progress_failurec C sD t jt�d�d�}|| j_| j�| jd d� | �tj| jj � d S )N�ert r � config1 save)
rC ru r rG r rv r7 r rF rS �r Z mock_recoverr r r �,test_recover_checkpoint_revert_temp_failures� s � �zHReverterCheckpointLocalTest.test_recover_checkpoint_revert_temp_failuresc C sR t jt�d�d�}|| j_| j�| jd d� | j�d� | � tj| jj
d� d S )Nrx rt r ry �Titler4 ) rC ru r rG r rv r( r r) rF �rollback_checkpointsrz r r r �(test_recover_checkpoint_rollback_failure� s � �zDReverterCheckpointLocalTest.test_recover_checkpoint_rollback_failurec C sJ | j �| jd d� t�d��"}td�|_| �tj | j j
� W 5 Q R X d S )Nr r# rA rB � r r7 r rC rD r& r' rF r rG rS rH r r r �$test_recover_checkpoint_copy_failure� s
�z@ReverterCheckpointLocalTest.test_recover_checkpoint_copy_failurec C sJ | j �| jd d� t�d��"}td�|_| �tj | j j
� W 5 Q R X d S )Nr z temp savez certbot.reverter.shutil.rmtreezCannot remove treer )r Z mock_rmtreer r r �"test_recover_checkpoint_rm_failure� s
�z>ReverterCheckpointLocalTest.test_recover_checkpoint_rm_failurez certbot.reverter.logger.warningc C s6 | j �dtj�| jd�� | j �� | �|jd� d S )NTzmissing_file.txtr4 ) r rL r r, r- r rS r: rq )r Z mock_warnr r r �)test_recover_checkpoint_missing_new_files� s �
zEReverterCheckpointLocalTest.test_recover_checkpoint_missing_new_fileszcertbot.reverter.os.removec C s0 | j �d| j� td�|_| �tj| j j� d S )NTz Can't remove) r rL r r& r' rF r rG rS )r Z mock_remover r r �&test_recover_checkpoint_remove_failure� s
�zBReverterCheckpointLocalTest.test_recover_checkpoint_remove_failurec C s t j�| jd�}| j�d|� t|d� | j�| jd d� t| j d� | j�| jd d� t| j
d � | j� | jd d
� t| j d � t j�| j d �}| j�d
|� t|d� | j�
� | �t j�|�� | �t j�|�� | �t| j �d� | �t| j
�d� d S )NrJ FzThis is a new perm file!r z
perm save1zupdated perm config1r4 z
perm save2zupdated perm config2z
temp save1z second update now temp config1rV TzNew temporary file!rR �directive-dir2)r r, r- r r rL rM r( r r r r7 r rW r< r= r: rT rX r r r �#test_recovery_routine_temp_and_perm� s"
z?ReverterCheckpointLocalTest.test_recovery_routine_temp_and_perm) �__name__�
__module__� __qualname__�__doc__r
r" rC rD r3 r@ rI rQ rU rY r] rc re rn ro rr rw r{ r~ r� r� r� r� r� �
__classcell__r r r r r s6
r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Ze �
d �d
d� �Z e �
d�dd� �Z e �
d�dd� �Z
dd� Ze�� dd� �Ze �
d�dd� �Zdd � Zd d � Z� ZS ) �TestFullCheckpointsReverterz5Tests functions having to deal with full checkpoints.c sT t t| ��� ddlm} t�tj� || j�| _ t
� }|\| _ | _ | _
| _| _d S r )r r� r
r r r r
r r r r r r r r r r r r r r
s z!TestFullCheckpointsReverter.setUpc C s6 t �| jj� t �| j� t �| j� t�tj� d S r r r! r r r r" & s z$TestFullCheckpointsReverter.tearDownc C s@ | � tj| jjd� | � tj| jjd� | � tj| jjd� d S )Nz-1i���Zone)rF r rG r r} r! r r r � test_rollback_improper_inputs- s � � �z9TestFullCheckpointsReverter.test_rollback_improper_inputsc C s | � � }| �tt�| jj��d� | j�d� | �t | j
�d� | �t | j �d� | �t |�d� | j�d� | �t | j
�d� | �t | j �d� | � tj
�|�� t�| jj�}| �t|�d� | �dttj
�| jj|d ��k� | j�d� | �t | j
�d � d S )
N� r4 �update config1�update config2�Final form config3r� �First Checkpointr rR )�_setup_three_checkpointsr: r[ r r* r r+ r r} rT r r r< r, r= r0 r; r- )r rP Zall_dirsr r r �.test_rollback_finalize_checkpoint_valid_inputs5 s& �� zJTestFullCheckpointsReverter.test_rollback_finalize_checkpoint_valid_inputsc C s | j �d� d S )NzNo checkpoint...)r r) r! r r r �'test_finalize_checkpoint_no_in_progressS s zCTestFullCheckpointsReverter.test_finalize_checkpoint_no_in_progressz certbot.reverter.shutil.movec C s6 | j �| jd d� td�|_| �tj| j jd� d S )Nr rs z cannot mover| � r r( r r& r' rF r rG r) )r Z mock_mover r r �%test_finalize_checkpoint_cannot_titleW s
�zATestFullCheckpointsReverter.test_finalize_checkpoint_cannot_titlez#certbot.reverter.filesystem.replacec C s2 | j �| jd d� t|_| �tj| j jd� d S )Nr rs r| r� )r Z mock_replacer r r �,test_finalize_checkpoint_no_rename_directory_ s �zHTestFullCheckpointsReverter.test_finalize_checkpoint_no_rename_directoryzcertbot.reverter.loggerc C sL | j �d� | �|jjd� | �� d|j_| j �d� | �|jjd� d S )Nr4 r � )r r} r: Zwarningrq r� �r Z mock_loggerr r r �test_rollback_too_manyh s z2TestFullCheckpointsReverter.test_rollback_too_manyc C sN | � � }| j�d� | �t| j�d� | �t| j�d� | �tj �
|�� d S )Nr� rR r� ) r� r r} r: rT r r r< r r, r= rO r r r �test_multi_rollbackt s
z/TestFullCheckpointsReverter.test_multi_rollbackc C s( | � � | j�� | �|� jjd� dS )z0This is not strict as this is subject to change.r4 N)r� r �view_config_changesr: Z notificationrq )r Z mock_outputr r r �test_view_config_changes| s
z4TestFullCheckpointsReverter.test_view_config_changesc C s | j �� | �|jjdk� d S )Nr )r r� r0 �inforq r� r r r �#test_view_config_changes_no_backups� s
z?TestFullCheckpointsReverter.test_view_config_changes_no_backupsc C s. t �tj�| jjd�� | �tj | j
j � d S )NZ in_progress) r �makedirsr r, r- r r+ rF r rG r r� r! r r r �(test_view_config_changes_bad_backups_dir� s
�zDTestFullCheckpointsReverter.test_view_config_changes_bad_backups_dirc C s� | j �| jd d� | j �d� t| jd� tj�| j d�}| j �
d|� t|d� | j �| jd d � | j �d
� t| j d � t|d � | j �| jd
d� | j �d� t| jd� t| j d� t|d� |S )z$Generate some finalized checkpoints.r z
first saver� r� rJ Fzdirective-config3r4 z second savezSecond Checkpointr� zupdate config3rK z
third savez Third Checkpoint - Save bothzFinal form config1zFinal form config2r� ) r r( r r) rM r r r, r- r rL r rO r r r r� � s
z4TestFullCheckpointsReverter._setup_three_checkpoints)r� r� r� r� r
r" r� r� r� rC rD r� r� r� r� � test_utilZpatch_get_utilityr� r� r� r� r� r r r r r� s&
r� c C s� t �d�} t �d�}tj�| d�}tj�|d�}t|d��}|�d� W 5 Q R X t|d��}|�d� W 5 Q R X t|g�t|g�t||g�g}||| ||fS )z!Setup sample configuration files.r r z
config.txt�wrR r� )�tempfileZmkdtempr r, r- r. �writerN )r r r r �file_fdr r r r r � s
�r c C s t tj�| d��S )zRead save notesr$ �rT r r, r- ��direr r r r; � s r; c C s t tj�| d��S )z
Get FilepathsZ FILEPATHSr� r� r r r r> � s r> c C s t tj�| d���� S )�Get new files.r6 )rT r r, r- �
splitlinesr� r r r rZ � s rZ c
C s8 t tj�| d��� }tt�|��W 5 Q R � S Q R X dS )r� ZCOMMANDSN)r. r r, r- �list�csv�reader)r� Zcsvfiler r r ri � s ri c
C s* t | d��}|�� W 5 Q R � S Q R X dS )z Read in a file, return the strr% N)r. r/ )r, r� r r r rT � s rT c C s$ t | d��}|�|� W 5 Q R X dS )z Update a file with a new value.r� N)r. r� )�filename�stringr� r r r rM � s rM �__main__) r� r� r r r� ZunittestrC rj Zcertbotr Zcertbot.compatr r Z
certbot.testsr r� ZConfigTestCaser r� r r; r> rZ ri rT rM r� �mainr r r r �<module> s2