Your IP : 216.73.216.52
���]� � @ s� 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) � 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)�ext� r �2/opt/alt/python35/lib64/python3.5/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)r0 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)r0 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 hellor6 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)Nr2 r zutf-8)r0 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�1r2 zutf-8r6 r ) r0 r r r r r7 r r: 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 hellor6 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)r4 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
r0 � s zDumbDBMTestCase.init_dbc C s; t | j � � } t | j j � � } | j | | � | S)N)�sortedr r r )r r! r Zdkeysr r r
r4 � 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)r: )�.0rK �vr r r
� <genexpr>� s z.DumbDBMTestCase.test_random.<locals>.<genexpr>)r r r ) �random�ranger r r �choice� randranger r rL �items) r rR �d�dummyr! rK rP 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 managerr2 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 ZiterkeysrV )r r r r r �operator�delitem�setitem�getitem�containsrZ r[ r �str� 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! rK 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 rG r Zcaptured_stdoutrZ �
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)Nr\ rl z.dirz.datr2 )r Ztemp_dirr �path�joinr r r r r r r% r) �S_IRUSR�S_IXUSRrL 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__r: r r# �unittestZ
skipUnless�hasattrr r- r. r1 r3 r5 r8 r9 r; r> rJ r r0 r4 rY r] rk rm rs ry rz r{ r r r r
r s> *
' r �__main__)rC r` r r) r Zdbm.dumbZdumbr r_ r � functoolsr ZTESTFNr r ZTestCaser r| �mainr r r r
�<module> s �