Your IP : 216.73.216.213
��Yf� � @ s� d Z d d l 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 � 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/python35/lib64/python3.5/test/test_dbm_dumb.py�
_delete_files s
r
c @ s� e Z d Z d d d d d d d d d d
d d d
d d j d � d i Z 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 e j e e d � d � d< d= � � Z d> d? � Z d@ dA � Z dB S)C�DumbDBMTestCase� 0� � as Python:� bs Programming� cs the� ds way� fs Guido� gs intended� üzutf-8� !c 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 r^ d } 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_modificationG 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_readN 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_keysT 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_containsZ 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_readj 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_containso s
z'DumbDBMTestCase.test_str_write_containsc C s� t j t � } d | d <d | d <| j � t j t d d � � } | j � } Wd QRXt j d k r} | j d d � } n | j d d � } t j t d d
� � } | j | � Wd QRXt 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_endings{ 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 s9 d d l } i } x t d � D]} t j t � } x� t d � D]� } | j d � } | j � d k r� | | k r� | | =| | =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 QRXt j t d � �& } | j t | j � � d g � Wd QRX| j t j � � | j � Wd QRXd S)Nr zcontext managerzdumbdbm context managerr1 s dumbdbm context manager)r r r r r r � assertRaises�error)r �dbr 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]E } | j
t j � � } | d � Wd QRX| j t
| j � d � qb Wxt t j d � t j d � t j d � t f D]E } | j
t j � � } | | � Wd QRX| 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 sw t j t d � �, } x" | j D] } | j | | | <q WWd QRXt j t d � � } | j | j � g � Wd QRXd S)N�n)r r r r r r )r r rJ r r r �test_create_new� s
z DumbDBMTestCase.test_create_newc C s� t t d d � � } | j d � Wd QRXt j � �M } | j t � � t j t � � } Wd QRXWd QRX| j | j � d � Wd QRXd 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 j � � } t j j | d � } t j | d � �H } | j t | j � � g � x" | j
D] } | j
| | | <q_ WWd QRXt j | d t j
� t j | d t j
� t j | t j
t j B� t j | d � �6 } | j t | j � � t | j
� � | j � Wd QRXWd QRXd S)NrX rg z.dirz.datr1 )r Ztemp_dirr �path�joinr r r r r r r$ r( �S_IRUSR�S_IXUSRrK r )r �dirZfnamer r! r r r �test_readonly_files� s %z#DumbDBMTestCase.test_readonly_filesc 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 rY rf rh rn rt ru rv r r r r r s> *
' r �__main__)�__doc__rB r\ r r( rz Zdbm.dumbZdumbr r[ r � functoolsr ZTESTFNr r
ZTestCaser rw �mainr r r r �<module> s �