Your IP : 216.73.216.213
�
f f~ � @ sb 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 � n d S) � N)�support)�MutableMapping)� dbm_iteratorc C s
| j d � S)Nzlatin-1)�decode)�s� r �5/opt/alt/python34/lib64/python3.4/test/test_shelve.py�L1 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 e Z
d d � Z d d � Z d S)� byteskeydictz Mapping that supports bytes keysc 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__�__doc__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 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# r0 r1 )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. r7 r# r0 r1 )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. r7 r# r0 r1 )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/ r0 r1 r �assertNotEqual�items)r r5 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/ r0 �appendr1 r )r r5 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 r2 �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@ r1 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 QX| 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/ r0 r r2 r3 r4 )r r5 r r r r � test_with� s
zTestCase.test_withN)r r r r# r& r6 r8 r9 r: r> rA rD 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 i d d 6d d 6d d 6S) NZvalue1r- r( Zkey2r* r+ Zkey3)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�counterr7 r# �strrJ r@ )r �xr r r �_empty_mapping� s %z TestShelveBase._empty_mappingc C se x | j D] } | j � q
Wg | _ | j sa x. t j | j d � D] } t j | � qG Wn d S)Nr! )rJ r1 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 i d d 6Z 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 i d d 6Z 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 i d d 6Z 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 i d d 6Z 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 i d d 6Z 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 i d d 6Z 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 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& {