Your IP : 216.73.216.172
U
�]� � @ s~ d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z
G dd � d e
j�Zed
krze�� dS ) z/Tests for certbot.plugins.storage.PluginStorage� N)�errors)�os)�
filesystem)�common)�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 sP t t| ��� tj| _t�| jj � t
� d�� | j| jdd�| _ W 5 Q R X d S )N�certbot.reverter.util�
mockplugin)�config�name)
�superr �setUpr Z Installer�
plugin_clsr �mkdirr
�
config_dir�mock�patch�plugin��self�� __class__� �>/usr/lib/python3/dist-packages/certbot/plugins/storage_test.pyr
s
zPluginStorageTest.setUpc C s� t tj�| jjd�d��}|�d� W 5 Q R X t�� }t |_
tj�| jjd�| j j _
t�d|��J tjddd��0 t�d�� | �tj| j j j� W 5 Q R X W 5 Q R X W 5 Q R X d S ) N�.pluginstorage.json�wZdummyzsix.moves.builtins.openz certbot.compat.os.path.isfileT)Z return_valuer )�openr �path�joinr
r �writer � mock_open�IOError� side_effectr �storage� storagepathr � assertRaisesr �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 5 Q R X t�d��`}t�d�� | � | jd�}W 5 Q R X | �
t |j j
d� | �|j� | �d|jd d k� W 5 Q R X 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% �KeyErrorr# �fetch�
assertTrueZcalled� call_args)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 5 Q R X t�d��V}t�d�� | � | jd�}W 5 Q R X | �
t j |j
jd� | �d|jd d k� W 5 Q R X 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 Z PluginErrorr# r- r. r/ )r r( r0 Z corruptedr r r �test_load_errors_corrupted3 s �� �z,PluginStorageTest.test_load_errors_corruptedc C sf t �d��R}d| jj_d| jj_| j| jj_| �t j
| jjj � | � d|j
d d k� W 5 Q R X d S )Nr2 Tz
/tmp/whateverzCould not serializer )r r r r# � _initializedr$ r �_datar% r r& �saver. r/ )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��J}ddi| jj_d| jj_| � t
j | jjj � | �
d|jd d k� W 5 Q R X W 5 Q R X d S )Nz certbot.compat.filesystem.openr2 Zvalid�dataTzCould not writer )r r r! r" r r r# r5 r4 r% r r& r6 r. r/ )r r r0 r r r �%test_save_errors_unable_to_write_fileI s
�z7PluginStorageTest.test_save_errors_unable_to_write_filec C s6 t �d��" | �tj| �| jd�jj� W 5 Q R X d S )Nr �x) 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 5 Q R X |j�dd� | �t|jjd� | �t|jjd� | � |j�d�d� d S )Nr �first�secondZ first_keyZ first_value)
r r r r
r# �putr% r, r- � assertEqual)r Zplugin1Zplugin2r r r �test_namespace_isolationY s � �z*PluginStorageTest.test_namespace_isolationc C s� | j j�dd� | j j�� t�d�� | �| jd�}W 5 Q R X | �|j� d�d� t
t j �
| jjd�d��}|�� }W 5 Q R X t�|�}| �d|�� k� | �t|�d� | �|d d d� d S )NZtestkeyZ testvaluer r r �r� )r r# r> r6 r r r r
r? r- r r r r r �read�json�loadsr. �keys�len)r Zanotherr( ZpsdataZpsjsonr r r �test_saved_statee s ��
z"PluginStorageTest.test_saved_state)�__name__�
__module__� __qualname__�__doc__r
r) r1 r3 r7 r9 r; r@ rH �
__classcell__r r r r r s
r �__main__)rL rD Zunittestr Zcertbotr Zcertbot.compatr r Zcertbot.pluginsr Z
certbot.testsr Z test_utilZConfigTestCaser rI �mainr r r r �<module> s h