File: //lib/python3/dist-packages/twisted/test/__pycache__/test_dirdbm.cpython-38.pyc
U
W[' � @ sv d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z G dd � d ej �Z
G d
d � d e
�Ze
egZdS ) z
Test cases for dirdbm module.
� N)� b64decode)�unittest)�dirdbm)�rebuild)�FilePathc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )� DirDbmTestsc C s( t | �� �| _t�| jj�| _d| _d S )N)�s abcs foo�s /lalals �s
s baz)r �mktemp�pathr �open�dbm�items��self� r �:/usr/lib/python3/dist-packages/twisted/test/test_dirdbm.py�setUp s zDirDbmTests.setUpc C s2 t d�}d| j|< d| j|< | �| j| d� d S )Nz//==� a)r r � assertEqual)r �kr r r �test_all s
zDirDbmTests.test_allc C s t �d�}d|d< t�t � d S )Nzdirdbm.rebuild.test� values key)r �Shelfr )r �sr r r �test_rebuildInteraction! s
z#DirDbmTests.test_rebuildInteractionc C s� | j }g }t� }| jD ]$\}}|||< |�|� |�|� q|�� | jD ]$\}}| �||� | �|| |� qJz |d W n tk
r� Y nX ds�t d��|�
� }t|� � �}t|�� �}|�� t| j�} | �||dt |�t |�f � | �||dt |�t |�f � | �| |dt | �t |�f � | �
� }
|�|
�} |�
� } t|� � �}
t|�� �}| �� | �|| dt |�t | �f � | �||
dt |�t |
�f � | �||d t |�t |�f � | �� | �t| �
� �t| � � � k�ot| �� � k�ot| � k�odkn d
� | �t|�t| �� t�|
� | jD ] \}}||= | �||d � �q@| �t|�
� �dd � | �t|� � �dd
� | �t|�� �dd� | �t|�dd� d S )Ns XXXr z)didn't raise KeyError on non-existent keyz%.keys() output didn't match: %s != %sz'.values() output didn't match: %s != %sz items() didn't match: %s != %sz'.copyTo().keys() didn't match: %s != %sz).copyTo().values() didn't match: %s != %sz(.copyTo().items() didn't match: %s != %sz.clear() failedz3key is still in database, even though we deleted itzdatabase has keyszdatabase has valueszdatabase has items)r �setr �append�add�sortZassertInr �KeyError�AssertionError�keys�values�reprr ZcopyTo�clear�
assertTrue�lenZassertNotEqual�shutilZrmtreeZ assertNotIn)r �dr# r$ r �vZdbkeysZdbvaluesZdbitemsr ZcopyPathZd2ZcopykeysZ
copyvaluesZ copyitemsr r r �test_dbm' s�
������
������"
�
�
��
zDirDbmTests.test_dbmc C sJ dd l }d| jd< | �t|� � | j�d� �dk� | �t| jjd� d S )Nr � v� k� s nokey)�timer r' �abs�getModificationTime� assertRaisesr! )r r0 r r r �test_modificationTimer s
"z!DirDbmTests.test_modificationTimec C s | j �d�}| j�|d �jdd��}|�d� W 5 Q R X | j �d�}| j�|�jdd��}|�d� W 5 Q R X | j�|d �jdd��}|�d� W 5 Q R X | j�d �jdd��}|�d
� W 5 Q R X t�| jj�}| �|d d� | �|d d� | � | j�
d �� | � | j�
d �� d
S )zJ
DirDBM: test recovery from directory after a faked crash
s key1s .rpl�wb)�moder s key2s corrects wrongzaa.news deletedz*.newz*.rplN) r Z_encoder Zchildr
�writer ZDirDBMr Z assertFalseZ globChildren)r r �fZk2r r r r �
test_recovery� s zDirDbmTests.test_recoveryc C s� | � t| jjdd� z| � t| jjdd� W n& tjk
rR | �| jtj� Y nX | � t| jj d� | � t| jj
d� | � t| jj d� | � t| jj d� | � t| jj
d� dS )zx
L{dirdbm.DirDBM} operations only support string keys: other types
should raise a L{TypeError}.
� �3�2r/ N)r3 � TypeErrorr � __setitem__r ZFailTestZassertIsInstancer r � __getitem__� __delitem__Zhas_key� __contains__r2 r r r r �test_nonStringKeys� s z DirDbmTests.test_nonStringKeysc C s: dd� }d| j d< | �| j d|� | �t| j jdd� dS )z4
Failure path when setting an item.
c S s | � |� td��d S )Nz
fail to write)Z
setContent�IOError)r �datar r r �
_writeFail� s
z,DirDbmTests.test_failSet.<locals>._writeFails tests failkeyZ
_writeFiles test2N)r Zpatchr3 rC r> )r rE r r r � test_failSet� s
zDirDbmTests.test_failSetN) �__name__�
__module__� __qualname__r r r r, r4 r9 rB rF r r r r r s Kr c @ s e Zd Zdd� ZdS )�
ShelfTestsc C s( t | �� �| _t�| jj�| _d| _d S )N)r r r
)s int� )s floatg (@)s tuple)NrK )r r r r r r r r r r r r � s zShelfTests.setUpN)rG rH rI r r r r r rJ � s rJ )�__doc__r) �base64r Z
twisted.trialr Ztwisted.persistedr Ztwisted.pythonr Ztwisted.python.filepathr ZTestCaser rJ Z testCasesr r r r �<module> s (