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|ddV 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|ddV 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 Ú s2