Your IP : 216.73.216.52
��Yf� � @ 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 �5/opt/alt/python35/lib64/python3.5/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 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 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/ r0 r r2 r3 r4 )r r5 r r r r � test_with� s
zTestCase.test_withc
C s0 t j i � � } | j | j d � Wd QRXd S)Nr+ )r. r/ r0 Z _protocol)r r r r r �test_default_protocol� s z TestCase.test_default_protocolN)r r r r# r& r6 r8 r9 r: r> rA rD rF rG rH 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)�_dbrI �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�counterr7 r# �strrK 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! )rK r1 rP r" r# r r$ )r Zdbr% r r r r& � s zTestShelveBase.tearDownN) r r r r# rR r
r. r/ Z type2testrO rU r& r r r r rJ � s rJ c @ s" e Z d Z d d i Z d Z d S)�TestAsciiFileShelver' r FN)r r r rQ rP r r r r rV � s rV c @ s" e Z d Z d d i Z d Z d S)�TestBinaryFileShelver' r* FN)r r r rQ rP r r r r rW � s rW c @ s" e Z d Z d d i Z d Z d S)�TestProto2FileShelver' r( FN)r r r rQ rP r r r r rX � s rX c @ s" e Z d Z d d i Z d Z d S)�TestAsciiMemShelver' r TN)r r r rQ rP r r r r rY � s rY c @ s" e Z d Z d d i Z d Z d S)�TestBinaryMemShelver' r* TN)r r r rQ rP r r r r rZ � s rZ c @ s" e Z d Z d d i Z d Z d S)�TestProto2MemShelver' r( TN)r r r rQ rP r r r r r[ � s r[ c C s7 x0 t � D]% } t j t t t t t t t � q
Wd S)N)
r r Z run_unittestrV rW rX rY rZ r[ r )�moduler r r � test_main� s r] �__main__)Zunittestr. r" Ztestr �collections.abcr Z
test.test_dbmr r r
r rI rL rJ rV rW rX rY rZ r[ r] r r r r r �<module> s&