🔐 Sid Gifari File Manager Pro
v8.0.5 | 2026-06-14 15:48:19 | PHP 8.1.34
📂
/ (Root)
/
usr
/
lib
/
python3.9
/
site-packages
/
certbot
/
_internal
/
tests
/
__pycache__
📍 /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
🔄 Refresh
✏️
Editing: reverter_test.cpython-39.pyc
Read Only
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|v s�J �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)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�s<J �t| jj�dksPJ �tj� tj� | jjd��rnJ �t | jj�d� | j | j�ks�J �d S )Nr r# � �save2Z save1save2� NEW_FILESz{0} {1} )r �add_to_temp_checkpointr r r, �isdirr �temp_checkpoint_dir�get_save_notes�isfiler- � get_filepaths�formatr r 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 r6 rA rB r rC �r �config3r r r �test_checkpoint_conflictH s 2z4ReverterCheckpointLocalTest.test_checkpoint_conflictc C s` | j �| jd d� t| jd� | j �| jd d� t| jd� | j �� t| j�dks\J �d S )Nr r# zupdated-directivezsave2-updated dirz'new directive change that we won't keep�directive-dir1)r r6 r rI r �revert_temporary_config�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�r�J �t j� | j �r�J �t j� |�r�J �t j� |�r�J �d S )NrF ZConfig3� config4.txtZConfig4T) r r, r- r rI r r rH r r �recovery_routiner: �r rK Zconfig4r r r �*test_multiple_registration_fail_and_revertd 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|�dks\J �d S )NTr3 )r rH r � get_new_filesr r8 �len)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_openr@ r& r' rA rB r rC r rH 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)rA rB r rC r rH r! r r r �test_bad_registration� s z1ReverterCheckpointLocalTest.test_bad_registrationc C s\ ddgddgdgg}|D ]}| j �d|� qt| jj�}t||�D ]\}}||ksBJ �qBd S )N�a2dismod�sslZrewriteZ cleanslateT)r �register_undo_command�get_undo_commandsr r8 �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 )NrY TrZ r[ Zcommand) r r\ r@ r& r' rA rB r rC r rc r] 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�� |jdksJJ �d S )N� Zinvalid_commandra rb TrG )r ZSubprocessErrorr' r rc rN � call_count)r Zmock_runrf rg 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 rC �_recover_checkpointrA rB rR 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�ern r � config1 save) r ro r rC r rp r6 r rA rB rN �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 )Nrr rn r rs �Titler3 ) r ro r rC r rp r( r r) rA rB �rollback_checkpointsrt 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# r> r? � r r6 r r r@ r& r' rA rB r rC rN rD 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 treery )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 �� |jdks2J �d S )NTzmissing_file.txtr3 )r rH r r, r- r rN rk )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 rH r r& r' rA rB r rC rN )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�|�r�J �t j�|�r�J �t| j �dks�J �t| j �dks�J �d S )NrF FzThis is a new perm file!r z perm save1zupdated perm config1r3 z perm save2zupdated perm config2z temp save1z second update now temp config1rQ TzNew temporary file!rM �directive-dir2)r r, r- r r rH rI r( r r r r6 r rR r: rO rS 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 r@ r2 r= rE rL rP rT rX r^ r` rh ri rl rq ru rx rz 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� 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)rA rB r rC r rw r! r r r � test_rollback_improper_inputs( s **z9TestFullCheckpointsReverter.test_rollback_improper_inputsc C s | � � }tt�| jj��dks"J �| j�d� t| j �dks@J �t| j �dksRJ �t|�dksbJ �| j�d� t| j �dks�J �t| j �dks�J �tj � |�r�J �t�| jj�}t|�dks�J �dt tj �| jj|d ��v s�J �| j�d� t| j �d k�sJ �d S ) N� r3 �update config1�update config2�Final form config3r~ �First Checkpointr rM )�_setup_three_checkpointsrV r r* r r+ r rw rO r r r, r: r9 r- )r rK 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 rm z cannot moverv � r r( r r& r' rA rB r rC 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 rm rv r� )r Z mock_replacer r r �,test_finalize_checkpoint_no_rename_directoryY s zHTestFullCheckpointsReverter.test_finalize_checkpoint_no_rename_directoryzcertbot.reverter.loggerc C sL | j �d� |jjdks J �| �� d|j_| j �d� |jjdksHJ �d S )Nr3 r � )r rw Zwarningrk r� )r Z mock_loggerr r r �test_rollback_too_manyb s z2TestFullCheckpointsReverter.test_rollback_too_manyc C sL | � � }| j�d� t| j�dks&J �t| j�dks8J �tj�|�rHJ �d S )Nr� rM r~ ) r� r rw rO r r r r, r: rJ 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 saver� r� rF Fzdirective-config3r3 z second savezSecond Checkpointr� zupdate config3rG z third savez Third Checkpoint - Save bothzFinal form config1zFinal form config2r� ) r r( r r) rI r r r, r- r rH r rJ r r r r� v s z4TestFullCheckpointsReverter._setup_three_checkpoints)r� r� r� r� r r" r� r� r� r r@ r� r� 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 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�wrM Nr~ )�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$ �rO r r, r- ��direr r r r9 � s r9 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.r5 )rO r r, r- � splitlinesr� r r r rU � s rU 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 rd � s rd 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 rO � s rO 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 rI � s rI �__main__r3 ) r� r� r r �sysr� Zunittestr rA Zcertbotr Zcertbot.compatr Z certbot.testsr Z test_utilZConfigTestCaser r� r r9 r; rU rd rO rI r� �exit�main�argv�__file__r r r r �<module> s2
💾 Save Changes
❌ Cancel