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 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 ksJ ‚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 Ú s2