Your IP : 216.73.216.213
�
g fL � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l j Z d d l m Z d d l
m Z e j Z
d d � Z Gd d � d e j � Z e d k r� e j � n d S)
zATest script for the dumbdbm module
Original by Roger E. Masse
� N)�support)�partialc C sH xA d d d g D]0 } y t j t | � Wq t k
r? Yq Xq Wd S)Nz.dirz.datz.bak)�os�unlink�_fname�OSError)Zext� r �7/opt/alt/python34/lib64/python3.4/test/test_dbm_dumb.py�
_delete_files s
r
c @ s} e Z d Z i d d 6d d 6d d 6d d 6d d
6d d 6d
d 6d d j d � 6Z d d � Z e j e e d � d � e j e e d � d � 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. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z d> S)?�DumbDBMTestCase� � 0s Python:� as Programming� bs the� cs way� ds Guido� fs intended� g� !� üzutf-8c C sn t j t d � } | j t | j � � g � x" | j D] } | j | | | <q8 W| j | � | j � d S)N�c) �dumbdbm�openr � assertEqual�list�keys�_dict� read_helper�close)�self�f�keyr r r �test_dumbdbm_creation! s
z%DumbDBMTestCase.test_dumbdbm_creation�umaskztest needs os.umask()�chmodztest needs os.chmod()c C s� z2 t j d � } t j t d d � } | j � Wd t j | � Xd } t j d k ra d } n d d l } t j t d � } | j | j | j
� | � t j t d � } | j | j | j
� | � d S)
N� r i� i� �posixi� r z.datz.dir) r r# r r r r �name�statr �S_IMODE�st_mode)r Z old_umaskr Z
expected_moder( �str r r �test_dumbdbm_creation_mode) s z*DumbDBMTestCase.test_dumbdbm_creation_modec C sE t j t � } d | d <| j | d d � | j � | j � d S)Nr r )r r r r r )r r r r r �test_close_twice? s
z DumbDBMTestCase.test_close_twicec C sL | j � t j t d � } d | j d <| d <| j | � | j � d S)N�ws indentedr )�init_dbr r r r r r )r r r r r �test_dumbdbm_modificationF s
z)DumbDBMTestCase.test_dumbdbm_modificationc C s7 | j � t j t d � } | j | � | j � d S)N�r)r/ r r r r r )r r r r r �test_dumbdbm_readM s
z!DumbDBMTestCase.test_dumbdbm_readc C s6 | j � t j t � } | j | � } | j � d S)N)r/ r r r � keys_helperr )r r r r r r �test_dumbdbm_keysS s
z!DumbDBMTestCase.test_dumbdbm_keysc C s7 t j t � } d | d <| j d | � | j � d S)Ns hello� 1)r r r �assertInr )r r r r r �test_write_containsY s
z#DumbDBMTestCase.test_write_containsc C s^ t j t � } d | d <d | d <| j � t j t � } | j | d d � | j � d S)Ns hellor5 s hello2)r r r r r )r r r r r �test_write_write_read_ s
z%DumbDBMTestCase.test_write_write_readc C sD | j � t j t d � } | j | d | j d j d � � d S)Nr1 r zutf-8)r/ r r r r r �encode)r r r r r �
test_str_readi s
z DumbDBMTestCase.test_str_readc C s� | j � t j t � } d | d <d | d <| j � t j t d � } | j d | � | j | d j d � | j d j d � � | j | d d � d S) Nr r �a�1r1 zutf-8r5 r ) r/ r r r r r6 r r9 r )r r r r r �test_str_write_containsn s
z'DumbDBMTestCase.test_str_write_containsc C s� t j t � } d | d <d | d <| j � t j t d d � � } | j � } Wd QXt j d k r| | j d d � } n | j d d � } t j t d d
� � } | j | � Wd QXt j t � } | j
| d d � | j
| d d � d S) Ns hellor5 s hello2� 2z.dir�rb�
s
s
�wb) r r r r �io�readr �linesep�replace�writer )r r �file�datar r r �test_line_endingsz s
z!DumbDBMTestCase.test_line_endingsc C sB | j | � } x, | j D]! } | j | j | | | � q Wd S)N)r3 r r )r r r r! r r r r � s zDumbDBMTestCase.read_helperc C sE t j t d � } x" | j D] } | j | | | <q W| j � d S)Nr. )r r r r r )r r �kr r r r/ � s zDumbDBMTestCase.init_dbc C s; t | j � � } t | j j � � } | j | | � | S)N)�sortedr r r )r r r Zdkeysr r r r3 � s zDumbDBMTestCase.keys_helperc C s< d d l } i } x#t d � D]} t j t � } x� t d � D]� } | j d � } | j � d k r� | | k r� | | =| | =q� qA | j d � | j d � } | | | <| | | <| j | | | � qA W| j � t j t � } t d
d � | j
� D� � } t | j
� � } | j | | � | j � q Wd S)
Nr � �d Z
abcdefghijklmg�������?r r r i' c s s* | ] \ } } | j d � | f Vq d S)zlatin-1N)r9 )�.0rJ �vr r r � <genexpr>� s z.DumbDBMTestCase.test_random.<locals>.<genexpr>)r r r ) �random�ranger r r ZchoiceZ randranger r rK �items) r rQ �dZdummyr rJ rO ZexpectedZgotr r r � test_random� s(
zDumbDBMTestCase.test_randomc C s� t j t d � � } d | d <Wd QXt j t d � �& } | j t | j � � d g � Wd QX| j t j � � | j � Wd QXd S)Nr zcontext managerzdumbdbm context managerr1 s dumbdbm context manager)r r r r r r � assertRaises�error)r Zdbr r r �test_context_manager� s %z$DumbDBMTestCase.test_context_managerc
C s$ t j t d � } | j � x� t t j | � t t j | d � t t j | � t t j | � f D]D } | j
t j � � } | d � Wd QX| j t
| j � d � qb Wxs t j d � t j d � t j d � t f D]D } | j
t j � � } | | � Wd QX| j t
| j � d � q� Wd S)Nr �b�testz"DBM object has already been closedr ZiterkeysrS )r r r r r �operator�delitem�setitem�getitem�containsrV rW r �strZ exception� methodcaller�len)r r �meth�cmr r r �test_check_closed� s$
z!DumbDBMTestCase.test_check_closedc C s� t t d d � � } | j d � Wd QXt j � �K } | j t � � t j t � � } Wd QXWd QX| j | j � d � Wd QXd S)Nz.dirr. zstr(print('Hacked!')), 0
� )
r r rF r Zcaptured_stdoutrV �
ValueErrorr r �getvalue)r �stream�stdoutr r r r � test_eval� s zDumbDBMTestCase.test_evalc C s t � d S)N)r
)r r r r �tearDown� s zDumbDBMTestCase.tearDownc C s t � d S)N)r
)r r r r �setUp� s zDumbDBMTestCase.setUpN) �__name__�
__module__� __qualname__r9 r r" �unittestZ
skipUnless�hasattrr r, r- r0 r2 r4 r7 r8 r: r= rI r r/ r3 rU rX re rk rl rm r r r r r s:
*
r �__main__)�__doc__rB r[ r rq Zdbm.dumbZdumbr rZ r � functoolsr ZTESTFNr r
ZTestCaser rn �mainr r r r �<module> s �