JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/reverter_test.cpython-39.pyc
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 Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg