a
}|ägE ã @ s– d Z ddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddl m Z
G dd „ d e
jƒZed
kr’e e ejd d… eg ¡¡ dS ) z/Tests for certbot.plugins.storage.PluginStorageé N)Úmock)Úerrors)Ú
filesystem)Úos)Úutilc s` e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z ‡ Z
S )ÚPluginStorageTestz.Test for certbot.plugins.storage.PluginStoragec s` t ƒ ¡ tj| _t | jj¡ t
d¡" | j| jdd| _ W d ƒ n1 sR0 Y d S )Núcertbot.reverter.utilÚ
mockplugin)ÚconfigÚname) ÚsuperÚsetUpÚ test_utilZDummyInstallerÚ
plugin_clsr Úmkdirr
Ú
config_dirr ÚpatchÚplugin©Úself©Ú __class__© úP/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/storage_test.pyr
s
zPluginStorageTest.setUpc C s* t tj | jjd¡dƒ}| d¡ W d ƒ n1 s80 Y t ¡ }t |_
tj | jjd¡| j j _
t d|¡ tjdddr t d¡H t tj¡ | j j ¡ W d ƒ n1 sÀ0 Y W d ƒ n1 sÞ0 Y W d ƒ n1 sü0 Y W d ƒ n1 s 0 Y d S ) Nú.pluginstorage.jsonÚwZdummyz
builtins.openz certbot.compat.os.path.isfileT)Z return_valuer )Úopenr ÚpathÚjoinr
r Úwriter Ú mock_openÚIOErrorÚ side_effectr ÚstorageÚ _storagepathr ÚpytestÚraisesr ÚPluginStorageErrorÚ_load)r Úfhr r r r Útest_load_errors_cant_read s" ÿÿ( ÿ
z,PluginStorageTest.test_load_errors_cant_readc C sÞ t tj | jjd¡dƒ}| d¡ W d ƒ n1 s80 Y t d¡~}t d¡ | | jd¡}W d ƒ n1 s|0 Y t
t ¡ |j
d¡ W d ƒ n1 s²0 Y W d ƒ n1 sÐ0 Y d S )Nr r Ú z$certbot.plugins.storage.logger.debugr r Úvalue)r r r r r
r r r r r r% r& ÚKeyErrorr# Úfetch)r r) Úmock_logZ nocontentr r r Útest_load_errors_empty' s ( , *z(PluginStorageTest.test_load_errors_emptyc C sà t tj | jjd¡dƒ}| d¡ W d ƒ n1 s80 Y t d¡€}t d¡ | | jd¡}W d ƒ n1 s|0 Y t
t j
¡ |j d¡ W d ƒ n1 s´0 Y W d ƒ n1 sÒ0 Y d S )Nr r z invalid jsonú$certbot.plugins.storage.logger.errorr r r, )r r r r r
r r r r r r% r& r Z PluginErrorr# r. )r r) r/ Z corruptedr r r Útest_load_errors_corrupted3 s ÿÿ( ,*z,PluginStorageTest.test_load_errors_corruptedc C s† t d¡h}d| jj_d| jj_| j| jj_t t
j ¡ | jj ¡ W d ƒ n1 sZ0 Y W d ƒ n1 sx0 Y d S )Nr1 Tú
/tmp/whatever)
r r r r# Ú _initializedr$ r Ú_datar% r& r r' Úsave)r r/ r r r Ú test_save_errors_cant_serialize> s
*z1PluginStorageTest.test_save_errors_cant_serializec
C s t ¡ }t|_t d|¡” t d¡j}ddi| jj_d| jj_d| jj_ t
t j
¡ | jj ¡ W d ƒ n1 sx0 Y W d ƒ n1 s–0 Y W d ƒ n1 s´0 Y d S )Nz certbot.compat.filesystem.openr1 ZvalidÚdataTr3 )r r r! r" r r r# r5 r4 r$ r% r& r r' r6 )r r r/ r r r Ú%test_save_errors_unable_to_write_fileH s
*z7PluginStorageTest.test_save_errors_unable_to_write_filec C sn t d¡P t tj¡$ | | jd¡j ¡ W d ƒ n1 sB0 Y W d ƒ n1 s`0 Y d S )Nr Úx)
r r r% r& r r' r r
r# r6 r r r r Útest_save_uninitializedT s z)PluginStorageTest.test_save_uninitializedc C sÄ t d¡, | | jd¡}| | jd¡}W d ƒ n1 s<0 Y |j dd¡ t t¡ |j d¡ W d ƒ n1 s€0 Y t t¡ |j d¡ W d ƒ n1 s¶0 Y d S )Nr ÚfirstÚsecondZ first_keyZ first_value)
r r r r
r# Úputr% r& r- r. )r Zplugin1Zplugin2r r r Útest_namespace_isolationY s , * *z*PluginStorageTest.test_namespace_isolationc C s¢ | j j dd¡ | j j ¡ t d¡ | | jd¡}W d ƒ n1 sJ0 Y tt j
| jj d¡dƒ}|
¡ }W d ƒ n1 sŠ0 Y t |¡}d S )NZtestkeyZ testvaluer r r Úr)r r# r> r6 r r r r
r r r r r ÚreadÚjsonÚloads)r Zanotherr) ZpsdataZpsjsonr r r Útest_saved_stated s ,ÿÿ&
z"PluginStorageTest.test_saved_state)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r
r* r0 r2 r7 r9 r; r? rD Ú
__classcell__r r r r r s
r Ú__main__é )rH rB ÚsysZunittestr r% Zcertbotr Zcertbot.compatr r Z
certbot.testsr r ZConfigTestCaser rE ÚexitÚmainÚargvÚ__file__r r r r Ú s g