a
}|gE @ s d Z ddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddl m Z
G dd d e
jZed
kreeejd d eg dS ) z/Tests for certbot.plugins.storage.PluginStorage N)mock)errors)
filesystem)os)utilc s` e Zd ZdZ fddZdd 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 )Ncertbot.reverter.util
mockplugin)configname) supersetUp test_utilZDummyInstaller
plugin_clsr mkdirr
config_dirr patchpluginself __class__ P/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/storage_test.pyr
s
zPluginStorageTest.setUpc C s* t tj| jjdd}|d W d n1 s80 Y t }t |_
tj| jjd| j j _
td| tjdddr tdH ttj | j j W d n1 s0 Y W d n1 s0 Y W d n1 s0 Y W d n1 s 0 Y d S ) N.pluginstorage.jsonwZdummyz
builtins.openz certbot.compat.os.path.isfileT)Z return_valuer )openr pathjoinr
r writer mock_openIOError side_effectr storage _storagepathr pytestraisesr PluginStorageError_load)r fhr r r r test_load_errors_cant_read s" (
z,PluginStorageTest.test_load_errors_cant_readc C s t tj| jjdd}|d W d n1 s80 Y td}td | | jd}W d n1 s|0 Y t
t |j
d W d n1 s0 Y |jsJ d|jd d v sJ W d n1 s0 Y d S )
Nr r z$certbot.plugins.storage.logger.debugr r valuezno values loadedr )r r r r r
r r r r r r% r& KeyErrorr# fetchZcalled call_args)r r) mock_logZ nocontentr r r test_load_errors_empty' s ( , *
z(PluginStorageTest.test_load_errors_emptyc C s t tj| jjdd}|d W d n1 s80 Y td}td | | jd}W d n1 s|0 Y t
t j
|jd W d n1 s0 Y d|jd d v sJ W d n1 s0 Y d S )
Nr r z invalid json$certbot.plugins.storage.logger.errorr r r, z is corruptedr )r r r r r
r r r r r r% r& r Z PluginErrorr# r. r/ )r r) r0 Z corruptedr r r test_load_errors_corrupted3 s ( ,*z,PluginStorageTest.test_load_errors_corruptedc C s t d~}d| jj_d| jj_| j| jj_t t
j | jj W d n1 sZ0 Y d|j
d d v szJ W d n1 s0 Y d S )Nr2 T
/tmp/whateverzCould not serializer )r r r r# _initializedr$ r _datar% r& r r' saver/ )r r0 r r r test_save_errors_cant_serialize> s
*z1PluginStorageTest.test_save_errors_cant_serializec
C s t }t|_t d| t d}ddi| jj_d| jj_d| jj_ t
t j
| jj W d n1 sx0 Y d|jd d v sJ W d n1 s0 Y W d n1 s0 Y d S ) Nz certbot.compat.filesystem.openr2 ZvaliddataTr4 zCould not writer )r r r! r" r r r# r6 r5 r$ r% r& r r' r7 r/ )r r r0 r r r %test_save_errors_unable_to_write_fileH s
*z7PluginStorageTest.test_save_errors_unable_to_write_filec C sn t dP ttj$ | | jdj W d n1 sB0 Y W d n1 s`0 Y d S )Nr x)
r r r% r& r r' r r
r# r7 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 |jdd tt |j d W d n1 s0 Y tt |j d W d n1 s0 Y |j ddksJ d S )Nr firstsecondZ 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 jdd | j j td | | jd}W d n1 sJ0 Y |jddkshJ t t
j | jj
dd}| }W d n1 s0 Y t|}d| v sJ t|dksJ |d d dksJ d S )NZtestkeyZ testvaluer r r r )r r# r? r7 r r r r
r. r r r r r readjsonloadskeyslen)r Zanotherr) ZpsdataZpsjsonr r r test_saved_stated s ,&
z"PluginStorageTest.test_saved_state)__name__
__module__ __qualname____doc__r
r* r1 r3 r8 r: r< r@ rH
__classcell__r r r r r s
r __main__rB )rL rD sysZunittestr r% Zcertbotr Zcertbot.compatr r Z
certbot.testsr r ZConfigTestCaser rI exitmainargv__file__r r r r s g