/usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
a }|�giB � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlZddl 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�e� ej dd� e g �� dS )zTest certbot.reverter.� N)�mock)�errors)�os)�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 sP t � �� ddlm} t�tj� || j�| _t � }|\| _ | _ | _ | _ | _d S �Nr )�Reverter��super�setUpZcertbot.reverterr �logging�disableZCRITICAL�config�reverter�setup_test_files�config1�config2�dir1�dir2�sets��selfr �tup�� __class__� �I/usr/lib/python3.9/site-packages/certbot/_internal/tests/reverter_test.pyr s z!ReverterCheckpointLocalTest.setUpc C s6 t �| jj� t �| j� t �| j� t�tj� d S �N� �shutil�rmtreer 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 y4 Y n0 | j�d� t�| jjj �d }tj � | jjj |d�}t |d��}|� � }W d � n1 s�0 Y d S )Nz cannot evenr �save1Zblah� CHANGES_SINCE�r)�OSError� side_effectr �add_to_checkpointr �finalize_checkpointr �listdirr � backup_dir�path�join�open�read)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� d S )Nr r# � �save2)r �add_to_temp_checkpointr r! r r r �!test_basic_add_to_temp_checkpoint4 s z=ReverterCheckpointLocalTest.test_basic_add_to_temp_checkpointc C sx t �d��Z}td�|_t�tj��$ | j� | j d d� W d � n1 sL0 Y W d � n1 sj0 Y d S )N� certbot.reverter.shutil.copy2�bad copyr r# ) r �patchr& r' �pytest�raisesr � ReverterErrorr r( r �r Z mock_copy2r r r �#test_add_to_checkpoint_copy_failureB 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 � t j ��$ | j�| jd d� W d � n1 s�0 Y | j�| jd d � t � t j �� | j�|hd � W d � n1 s�0 Y d S ) z:Make sure that checkpoint errors are thrown appropriately.� config3.txtTzThis is a new file!� r# r r4 Zsave3Nr3 Zsave4z invalid save)r r, r- r r �register_file_creation� update_filer( r r5 r: r; r r< �r �config3r r r �test_checkpoint_conflictH s 2z4ReverterCheckpointLocalTest.test_checkpoint_conflictc C sN | j �| jd d� t| jd� | j �| jd d� t| jd� | j �� d S )Nr r# zupdated-directivezsave2-updated dirz'new directive change that we won't keep)r r5 r rB r �revert_temporary_configr! r r r �#test_multiple_saves_and_temp_revert[ s z?ReverterCheckpointLocalTest.test_multiple_saves_and_temp_revertc C sr t j�| jd�}t|d� t j�| jd�}t|d� | j�d| j� | j�d| j � | j�d||� | j� � d S )Nr? ZConfig3� config4.txtZConfig4T) r r, r- r rB r r rA r r �recovery_routine�r rD Zconfig4r r r �*test_multiple_registration_fail_and_revertd s zFReverterCheckpointLocalTest.test_multiple_registration_fail_and_revertc C sP | j �d| j� | j �d| j� | j �d| j� | j �d| j� t| jj�}d S )NT)r rA r � get_new_filesr �temp_checkpoint_dir)r �filesr r r �$test_multiple_registration_same_filex s z@ReverterCheckpointLocalTest.test_multiple_registration_same_filec C s� t �� }t jd|dd��V td�|_t�tj�� | j � d| j � W d � n1 sV0 Y W d � n1 st0 Y d S )N�certbot.reverter.openT�Zcreate�bad open) r � mock_openr9 r& r' r: r; r r< r rA r �r Zm_openr r r �'test_register_file_creation_write_error� s zCReverterCheckpointLocalTest.test_register_file_creation_write_errorc C s< t �tj�� | j�d� W d � n1 s.0 Y d S )N�filepath)r: r; r r< r rA r! r r r �test_bad_registration� s z1ReverterCheckpointLocalTest.test_bad_registrationc C sP ddgddgdgg}|D ]}| j �d|� qt| jj�}t||�D ]\}}qBd S )N�a2dismod�sslZrewriteZ cleanslateT)r �register_undo_command�get_undo_commandsr rM �zip)r �coms�comZact_comsZa_comr r r �test_register_undo_command� s � z6ReverterCheckpointLocalTest.test_register_undo_commandc C s� t �� }t jd|dd��V td�|_t�tj�� | j � ddg� W d � n1 sV0 Y W d � n1 st0 Y d S )NrP TrQ rR Zcommand) r rS r9 r& r' r: r; r r< r rZ rT r r r � test_bad_register_undo_command� s z:ReverterCheckpointLocalTest.test_bad_register_undo_commandzcertbot.util.run_scriptc C s@ dt jg|_dgddgg}|D ]}| j�d|� q | j�� d S )N� Zinvalid_commandrX rY T)r ZSubprocessErrorr' r rZ rF )r Zmock_runr] r^ r r r �test_run_undo_commands� s � z2ReverterCheckpointLocalTest.test_run_undo_commandsc C s` | j �| jd d� tjtjd�| j _t� tj�� | j � � W d � n1 sR0 Y d S )Nr � perm save�r' ) r r( r r � MagicMockr r< �_recover_checkpointr: r; rI r! r r r �)test_recovery_routine_in_progress_failure� s � zEReverterCheckpointLocalTest.test_recovery_routine_in_progress_failurec C sh t jt�d�d�}|| j_| j�| jd d� t� tj�� | j� � W d � n1 sZ0 Y d S )N�erd r � config1 save) r re r r< r rf r5 r r: r; rF �r Z mock_recoverr r r �,test_recover_checkpoint_revert_temp_failures� s �zHReverterCheckpointLocalTest.test_recover_checkpoint_revert_temp_failuresc C sv t jt�d�d�}|| j_| j�| jd d� | j�d� t � tj�� | j� d� W d � n1 sh0 Y d S )Nrh rd r ri �Titler3 ) r re r r< r rf r( r r) r: r; �rollback_checkpointsrj r r r �(test_recover_checkpoint_rollback_failure� s � zDReverterCheckpointLocalTest.test_recover_checkpoint_rollback_failurec C s� | j �| jd d� t�d��P}td�|_t�t j �� | j � � W d � n1 sV0 Y W d � n1 st0 Y d S )Nr r# r7 r8 � r r5 r r r9 r& r' r: r; r r< rF r= r r r �$test_recover_checkpoint_copy_failure� s z@ReverterCheckpointLocalTest.test_recover_checkpoint_copy_failurec C s� | j �| jd d� t�d��P}td�|_t�t j �� | j � � W d � n1 sV0 Y W d � n1 st0 Y d S )Nr z temp savez certbot.reverter.shutil.rmtreezCannot remove treero )r Z mock_rmtreer r r �"test_recover_checkpoint_rm_failure� s z>ReverterCheckpointLocalTest.test_recover_checkpoint_rm_failurez certbot.reverter.logger.warningc C s( | j �dtj�| jd�� | j �� d S )NTzmissing_file.txt)r rA r r, r- r rF )r Z mock_warnr r r �)test_recover_checkpoint_missing_new_files� s � zEReverterCheckpointLocalTest.test_recover_checkpoint_missing_new_fileszcertbot.reverter.os.removec C sT | j �d| j� td�|_t�tj�� | j � � W d � n1 sF0 Y d S )NTz Can't remove) r rA r r& r' r: r; r r< rF )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� � d S )Nr? FzThis is a new perm file!r z perm save1zupdated perm config1r3 z perm save2zupdated perm config2z temp save1z second update now temp config1rH TzNew temporary file!)r r, r- r r rA rB r( r r r r5 r rI rJ r r r �#test_recovery_routine_temp_and_perm� s z?ReverterCheckpointLocalTest.test_recovery_routine_temp_and_perm) �__name__� __module__� __qualname__�__doc__r r" r r9 r2 r6 r> rE rG rK rO rU rW r_ r` rb rg rk rn rp rq rr rs rt � __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� Zdd� Z� ZS )�TestFullCheckpointsReverterz5Tests functions having to deal with full checkpoints.c sP t � �� ddlm} t�tj� || j�| _t � }|\| _ | _ | _ | _ | _d S 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� t �tj�� | j�d� W d � n1 s.0 Y t �tj�� | j�d� W d � n1 sf0 Y t �tj�� | j�d� W d � n1 s�0 Y d S )Nz-1i���Zone)r: r; r r< r rm r! r r r � test_rollback_improper_inputs( s **z9TestFullCheckpointsReverter.test_rollback_improper_inputsc C s> | � � }| j�d� | j�d� t�| jj�}| j�d� d S )Nr3 )�_setup_three_checkpointsr rm r r* r r+ )r rD Zall_dirsr r r �.test_rollback_finalize_checkpoint_valid_inputs0 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_progressM s zCTestFullCheckpointsReverter.test_finalize_checkpoint_no_in_progressz certbot.reverter.shutil.movec C sZ | j �| jd d� td�|_t�tj�� | j � d� W d � n1 sL0 Y d S )Nr rc z cannot moverl � r r( r r& r' r: r; r r< r) )r Z mock_mover r r �%test_finalize_checkpoint_cannot_titleQ s zATestFullCheckpointsReverter.test_finalize_checkpoint_cannot_titlez#certbot.reverter.filesystem.replacec C sV | j �| jd d� t|_t�tj�� | j � d� W d � n1 sH0 Y d S )Nr rc rl r )r Z mock_replacer r r �,test_finalize_checkpoint_no_rename_directoryY s zHTestFullCheckpointsReverter.test_finalize_checkpoint_no_rename_directoryzcertbot.reverter.loggerc C s, | j �d� | �� d|j_| j �d� d S )Nr3 r � )r rm r| ZwarningZ call_count)r Z mock_loggerr r r �test_rollback_too_manyb s z2TestFullCheckpointsReverter.test_rollback_too_manyc C s | � � }| j�d� d S )N� )r| r rm rC r r r �test_multi_rollbackn s z/TestFullCheckpointsReverter.test_multi_rollbackc 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 savezFirst Checkpointzupdate config1r? Fzdirective-config3r3 z second savezSecond Checkpointzupdate config2zupdate config3r@ z third savez Third Checkpoint - Save bothzFinal form config1zFinal form config2zFinal form config3) r r( r r) rB r r r, r- r rA r rC r r r r| v s z4TestFullCheckpointsReverter._setup_three_checkpoints)ru rv rw rx r r" r{ r} r~ r r9 r� r� r� r� r| ry r r r r rz s rz c C s� t �d�} t �d�}tj�| d�}tj�|d�}t|d��}|�d� W d � n1 sZ0 Y t|d��}|�d� W d � n1 s�0 Y |h|h||hg}||| ||fS )z!Setup sample configuration files.r r z config.txt�wzdirective-dir1Nzdirective-dir2)�tempfileZmkdtempr r, r- r. �write)r r r r �file_fdr r r r r � s ( (�r c C s t tj�| d��S )zRead save notesr$ ��read_inr r, r- ��direr r r �get_save_notes� s r� c C s t tj�| d��S )z Get FilepathsZ FILEPATHSr� r� r r r � get_filepaths� s r� c C s t tj�| d���� S )�Get new files.Z NEW_FILES)r� r r, r- � splitlinesr� r r r rL � s rL c C sD t tj�| d��� }tt�|��W d � S 1 s60 Y dS )r� ZCOMMANDSN)r. r r, r- �list�csv�reader)r� Zcsvfiler r r r[ � s r[ c C s6 t | d��}|�� W d � S 1 s(0 Y dS )z Read in a file, return the strr% N)r. r/ )r, r� r r r r� � s r� c C s8 t | d��}|�|� W d � n1 s*0 Y dS )z Update a file with a new value.r� N)r. r� )�filename�stringr� r r r rB � s rB �__main__r3 ) rx r� r r �sysr� Zunittestr r: Zcertbotr Zcertbot.compatr Z certbot.testsr Z test_utilZConfigTestCaser rz r r� r� rL r[ r� rB ru �exit�main�argv�__file__r r r r �<module> s2
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
account_test.cpython-39.opt-1.pyc
Edit
account_test.cpython-39.pyc
Edit
auth_handler_test.cpython-39.opt-1.pyc
Edit
auth_handler_test.cpython-39.pyc
Edit
cert_manager_test.cpython-39.opt-1.pyc
Edit
cert_manager_test.cpython-39.pyc
Edit
cli_test.cpython-39.opt-1.pyc
Edit
cli_test.cpython-39.pyc
Edit
client_test.cpython-39.opt-1.pyc
Edit
client_test.cpython-39.pyc
Edit
configuration_test.cpython-39.opt-1.pyc
Edit
configuration_test.cpython-39.pyc
Edit
crypto_util_test.cpython-39.opt-1.pyc
Edit
crypto_util_test.cpython-39.pyc
Edit
eff_test.cpython-39.opt-1.pyc
Edit
eff_test.cpython-39.pyc
Edit
error_handler_test.cpython-39.opt-1.pyc
Edit
error_handler_test.cpython-39.pyc
Edit
errors_test.cpython-39.opt-1.pyc
Edit
errors_test.cpython-39.pyc
Edit
helpful_test.cpython-39.opt-1.pyc
Edit
helpful_test.cpython-39.pyc
Edit
hook_test.cpython-39.opt-1.pyc
Edit
hook_test.cpython-39.pyc
Edit
lock_test.cpython-39.opt-1.pyc
Edit
lock_test.cpython-39.pyc
Edit
log_test.cpython-39.opt-1.pyc
Edit
log_test.cpython-39.pyc
Edit
main_test.cpython-39.opt-1.pyc
Edit
main_test.cpython-39.pyc
Edit
ocsp_test.cpython-39.opt-1.pyc
Edit
ocsp_test.cpython-39.pyc
Edit
renewal_test.cpython-39.opt-1.pyc
Edit
renewal_test.cpython-39.pyc
Edit
renewupdater_test.cpython-39.opt-1.pyc
Edit
renewupdater_test.cpython-39.pyc
Edit
reverter_test.cpython-39.opt-1.pyc
Edit
reverter_test.cpython-39.pyc
Edit
storage_test.cpython-39.opt-1.pyc
Edit
storage_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit