File: //usr/local/lib/python3.7/test/__pycache__/test_shelve.cpython-37.opt-2.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ dd� Z G dd� de�Z
G d d
� d
e j �Z d d lm Z G d d
� d
e j
�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zdd� Zed kr�e� dS ) � N)�support)�MutableMapping)� dbm_iteratorc C s
| � d�S )Nzlatin-1)�decode)�s� r �,/usr/local/lib/python3.7/test/test_shelve.py�L1 s r c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � ZeZ d
d� Z
dd� Z dS )� byteskeydictc C s
i | _ d S )N)�d)�selfr r r �__init__ s zbyteskeydict.__init__c C s | j t|� S )N)r r )r �keyr r r � __getitem__ s zbyteskeydict.__getitem__c C s || j t|�< d S )N)r r )r r �valuer r r � __setitem__ s zbyteskeydict.__setitem__c C s | j t|�= d S )N)r r )r r r r r � __delitem__ s zbyteskeydict.__delitem__c C s
t | j�S )N)�lenr )r r r r �__len__ s zbyteskeydict.__len__c c s$ x | j �� D ]}|�d�V q W d S )Nzlatin-1)r �keys�encode)r �kr r r �iterkeys s zbyteskeydict.iterkeysc C s t | �� �S )N)�listr )r r r r r # s zbyteskeydict.keysc C s
t | j�S )N)r
r )r r r r �copy& s zbyteskeydict.copyN) �__name__�
__module__� __qualname__r
r r r r r �__iter__r r r r r r r
s r
c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�TestCasez shelftemp.dbc C s( x"t � | jd �D ]}t�|� qW d S )N�*)�glob�fnr �unlink)r �fr r r �tearDown. s zTestCase.tearDownc C s� i }t j|ddd�}ddddg|d< | �|d ddddg� | �t|�d� |�� | �tt|� y |d W n tk
r� Y n X | �d� d S ) N� F)�protocol� writeback� � � �key1z"Closed shelf should not find a key)�shelve�Shelf� assertEqualr �close� assertRaises�
ValueError�fail)r �d1r r r r �
test_close2 s zTestCase.test_closec C s< t j| jdd�}z d|d< | �|d d� W d |�� X d S )Nr )r' )r) r& r* r+ r, )r- �openr" r/ r0 )r r r r r �test_ascii_file_shelfA s
z TestCase.test_ascii_file_shelfc C s< t j| jdd�}z d|d< | �|d d� W d |�� X d S )Nr) )r' )r) r& r* r+ r, )r- r6 r" r/ r0 )r r r r r �test_binary_file_shelfI s
z TestCase.test_binary_file_shelfc C s< t j| jdd�}z d|d< | �|d d� W d |�� X d S )Nr& )r' )r) r& r* r+ r, )r- r6 r" r/ r0 )r r r r r �test_proto2_file_shelfQ s
z TestCase.test_proto2_file_shelfc C s� t � }tj|dd�}d|d< | �|d d� |�� t � }tj|dd�}d|d< | �|d d� |�� | �t|�d� | �t|�d� | �|�� |�� � d S )Nr )r' )r) r& r* r+ r, r) )r
r- r. r/ r0 r �assertNotEqual�items)r r4 r �d2r r r �test_in_memory_shelfY s z TestCase.test_in_memory_shelfc C s� t � }tj|ddd�}ddddg|d< | �|d ddddg� |d �d� | �|d ddddg� |�� t � }tj|dd d�}ddddg|d< | �|d ddddg� |d �d� | �|d dddddg� |�� | �t|�d� | �t|�d� d S )
Nr& F)r' r( r) r* r+ r, � T)r
r- r. r/ �appendr0 r )r r4 r r< r r r �test_mutable_entryi s zTestCase.test_mutable_entryc C sv i }d}dgt �|�|< | �|�d�|� dgt j|dd�|< | �|�d�|� t j|dd�}| �t|j|dg� d S )Nu Pöpr) zutf-8zlatin-1)Z keyencoding�ascii)r- r. ZassertInr r1 �UnicodeEncodeErrorr )r r r r r r r �test_keyencoding} s zTestCase.test_keyencodingc C s` i }d}|� d�}tj|dd�}dg||< || }|d �d� |�� || }| �||� d S )Nr zutf-8T)r( r) r& )r r- r. r? r0 r: )r r r �
encodedkeyr Zp1Zp2r r r �&test_writeback_also_writes_immediately� s
z/TestCase.test_writeback_also_writes_immediatelyc C s� i }t j|ddd��>}ddddg|d< | �|d ddddg� | �t|�d� W d Q R X | �tt|� y |d W n tk
r� Y n X | �d� d S ) Nr& F)r' r( r) r* r+ r, z"Closed shelf should not find a key)r- r. r/ r r1 r2 r3 )r r4 r r r r � test_with� s zTestCase.test_withc C s( t �i ��}| �|jd� W d Q R X d S )Nr* )r- r. r/ Z _protocol)r r r r r �test_default_protocol� s z TestCase.test_default_protocolN)r r r r" r% r5 r7 r8 r9 r= r@ rC rE rF rG r r r r r * s
r )�
mapping_testsc @ s: e Zd ZdZdZdd� ZejZdd� Z dd� Z
d d
� Z d S ) �TestShelveBasez shelftemp.dbr c O s g | _ tjj| f|�|� d S )N)�_dbrH �BasicTestMappingProtocolr
)r �args�kwr r r r
� s zTestShelveBase.__init__c C s dddd�S )NZvalue1r& )r) r& r* )r, Zkey2Zkey3r )r r r r �
_reference� s zTestShelveBase._referencec C sT | j rtjt� f| j�}n*| jd7 _tj| jt| j� f| j�}| j �
|� |S )Nr) ) �_in_memr- r. r
�_args�counterr6 r" �strrJ r? )r �xr r r �_empty_mapping� s z TestShelveBase._empty_mappingc C sL x| j D ] }|�� qW g | _ | jsHx"t�| jd �D ]}t�|� q6W d S )Nr )rJ r0 rO r! r" r r# )r Zdbr$ r r r r% � s zTestShelveBase.tearDownN) r r r r" rQ r
r- r. Z type2testrN rT r% r r r r rI � s rI c @ s e Zd ZddiZdZdS )�TestAsciiFileShelver' r FN)r r r rP rO r r r r rU � s rU c @ s e Zd ZddiZdZdS )�TestBinaryFileShelver' r) FN)r r r rP rO r r r r rV � s rV c @ s e Zd ZddiZdZdS )�TestProto2FileShelver' r& FN)r r r rP rO r r r r rW � s rW c @ s e Zd ZddiZdZdS )�TestAsciiMemShelver' r TN)r r r rP rO r r r r rX � s rX c @ s e Zd ZddiZdZdS )�TestBinaryMemShelver' r) TN)r r r rP rO r r r r rY � s rY c @ s e Zd ZddiZdZdS )�TestProto2MemShelver' r& TN)r r r rP rO r r r r rZ � s rZ c
C s* x$t � D ]} t�ttttttt � qW d S )N)
r r Z run_unittestrU rV rW rX rY rZ r )�moduler r r � test_main� s r\ �__main__)Zunittestr- r! Ztestr Zcollections.abcr Z
test.test_dbmr r r
r rH rK rI rU rV rW rX rY rZ r\ r r r r r �<module> s&