← Back
Editing: reverter_test.cpython-38.pyc
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
Save File
Cancel