Your IP : 216.73.216.147
3
��2a� � @ 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
| j d�S )Nzlatin-1)�decode)�s� r �0/opt/alt/python36/lib64/python3.6/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 j� D ]}|jd�V q W d S )Nzlatin-1)r �keys�encode)r �kr r r �iterkeys s zbyteskeydict.iterkeysc C s t | j� �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 j | jd �D ]}tj|� qW d S )N�*)�glob�fnr �unlink)r �fr r r �tearDown. s zTestCase.tearDownc C s� i }t j|ddd�}ddddg|d< | j|d ddddg� | jt|�d� |j� | jtt|� y |d W n tk
r� Y n X | jd� 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< | j|d d � W d |j� X d S )
Nr )r' r) r& r* r+ r, )r) r& 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< | j|d d� W d |j� X d S ) Nr) )r' r& r* r+ r, )r) r& 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< | j|d d� W d |j� X d S ) Nr& )r' r) r* r+ r, )r) r& 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< | j|d d � |j� t � }tj|dd�}d
|d< | j|d d � |j� | jt|�d� | jt|�d� | j|j� |j� � d S ) Nr )r' r) r& r* r+ r, )r) r& r* r+ )r) r& r* r+ )r) 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< | j|d ddddg� |d jd� | j|d ddddg� |j� t � }tj|dd d�}ddddg|d< | j|d ddddg� |d jd� | j|d dddddg� |j� | jt|�d� | jt|�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 j|�|< | j|jd�|� dgt j|dd�|< | j|jd�|� t j|dd�}| jt|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}|j d�}tj|dd�}dg||< || }|d jd� |j� || }| j||� 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< | j|d ddddg� | jt|�d� W d Q R X | jtt|� y |d W n tk
r� Y n X | jd� 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 ji ��}| j|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, Zkey2Zkey3)r) r& r* r )r r r r �
_reference� s zTestShelveBase._referencec C sT | j rtjt� f| j�}n*| jd7 _tj| jt| j� f| j�}| 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 ] }|j� qW g | _ | jsHx"tj| jd �D ]}tj|� 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 ]} tjttttttt � 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 �collections.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&