Your IP : 216.73.216.213
���]� � @ s_ d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d � Z Gd d � d e � Z
Gd d
� d
e j � Z d d l m Z Gd d
� d
e j
� Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d d � Z e d k r[e � d S) � N)�support)�MutableMapping)� dbm_iteratorc C s
| j d � S)Nzlatin-1)�decode)�s� r �0/opt/alt/python35/lib64/python3.5/test_shelve.py�L1 s r c @ sv e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e Z d
d � Z
d d � Z d S)� 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] } | j d � Vq Wd 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 @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d S)�TestCasez shelftemp.dbc C s2 x+ t j | j d � D] } t j | � q Wd S)N�*)�glob�fnr �unlink)r �fr r r �tearDown. s zTestCase.tearDownc C s� i } t j | d d d d �} d d d d g | d <| j | d d d d d g � | j t | � d � | j � | j t t | � y | d Wn t k
r� Yn X| j d � d S)
N�protocol� � writebackF� � � �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 sL t j | j d d �} z" d | d <| j | d d � Wd | j � Xd 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 sL t j | j d d �} z" d | d <| j | d d � Wd | j � Xd 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 sL t j | j d d �} z" d | d <| j | d d � Wd | j � Xd 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 � } t j | d d �} d | d <| j | d d � | j � t � } t j | d d �} d
| d <| j | d d � | j � | j t | � d � | j t | � 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 � } t j | d d d d �} d d d d g | d <| j | d d d d d g � | d j d � | j | d d d d d g � | j � t � } t j | d d d d
�} d d d d g | d <| j | d d d d d g � | d j d � | j | d d d d d d g � | j � | j t | � d � | j t | � d � d S) Nr& r' r( Fr) 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 s� i } d } d g t j | � | <| j | j d � | � d g t j | d d �| <| j | j d � | � t j | d d �} | j t | j | d g � d S)Nu Pöpr) zutf-8Z keyencodingzlatin-1�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 � } t j | d d �} d g | | <| | } | d j d � | j � | | } | j | | � d S)Nr zutf-8r( Tr) 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 | d d d d ��S } d d d d g | d <| j | d d d d d g � | j t | � d � Wd QRX| j t t | � y | d Wn t k
r� Yn X| j d � d S)
Nr& r' r( Fr) 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 s0 t j i � � } | j | j d � Wd QRXd 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 @ sU e Z d Z d Z d Z d d � Z e j Z d d � Z d d � Z
d d
� Z d S) �TestShelveBasez shelftemp.dbr c O s# g | _ t j j | | | � d S)N)�_dbrH �BasicTestMappingProtocolr
)r �args�kwr r r r
� s zTestShelveBase.__init__c C s d d d d d d i S) Nr, Zvalue1Zkey2r' Zkey3r) r* )r) r' r* r )r r r r �
_reference� s zTestShelveBase._referencec C sl | j r$ t j t � | j � } n4 | j d 7_ t j | j t | j � | 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 sb x | j D] } | j � q
Wg | _ | j s^ x+ t j | j d � D] } t j | � qG Wd 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 Z d Z d d i Z d Z d S)�TestAsciiFileShelver& r FN)r r r rP rO r r r r rU � s rU c @ s" e Z d Z d d i Z d Z d S)�TestBinaryFileShelver& r) FN)r r r rP rO r r r r rV � s rV c @ s" e Z d Z d d i Z d Z d S)�TestProto2FileShelver& r' FN)r r r rP rO r r r r rW � s rW c @ s" e Z d Z d d i Z d Z d S)�TestAsciiMemShelver& r TN)r r r rP rO r r r r rX � s rX c @ s" e Z d Z d d i Z d Z d S)�TestBinaryMemShelver& r) TN)r r r rP rO r r r r rY � s rY c @ s" e Z d Z d d i Z d Z d S)�TestProto2MemShelver& r' TN)r r r rP rO r r r r rZ � s rZ c C s7 x0 t � D]% } t j t t t t t t t � q
Wd 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&