← Back
Editing: storage_test.cpython-38.pyc
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
Save File
Cancel