Your IP : 216.73.216.213
3
� f�( � @ s� d Z ddlZddlZddlZddlZddlZddlZddljZ ddl
m Z ddl m
Z
e jZdd� ZG dd� dej�Zed kr�ej� dS )
zATest script for the dumbdbm module
Original by Roger E. Masse
� N)�support)�partialc C s: x4dD ],} yt jt| � W q tk
r0 Y qX qW d S )N�.dir�.dat�.bak)r r r )�os�unlink�_fname�OSError)Zext� r �7/opt/alt/python36/lib64/python3.6/test/test_dbm_dumb.py�
_delete_files s
r
c @ sB e Zd Zddddddddd d
d d d
ddjd�diZdd� Zejee d�d�ejee d�d�dd� ��Z
dd� Z d d � Z d d � Z
d d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� Zd8d9� Zd:d;� Zd<d=� Z d>d?� Z ejee d�d�d@dA� �Z eje j dB�dCdD� �Z!dEdF� Z"dGdH� Z#dIS )J�DumbDBMTestCase� 0� � as Python:� bs Programming� cs the� ds way� fs Guido� gs intended� üzutf-8� !c C sT t jtd�}| jt|j� �g � x | jD ]}| j| ||< q(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� z$t jd�}tjtdd�}|j� W d t j|� X d}t jdkrDd}dd l}t jtd �}| j|j |j
�|� t jtd �}| 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 s6 t jt�}d|d< | j|d d� |j� |j� d S )Nr r )r r r r r! )r" r# r r r �test_close_twiceA s
z DumbDBMTestCase.test_close_twicec C s` | j � tjtd�}d | jd< |d< | j|� | j|jdd�d� | j|d d� |j� d S )N�ws indentedr s xxxs foo) �init_dbr r r r r r �
setdefaultr! )r" r# r r r �test_dumbdbm_modificationH s
z)DumbDBMTestCase.test_dumbdbm_modificationc C s� | j � tjtd�}| j|� | jtd�� d|d< W d Q R X | jtd�� |d= W d Q R X | j|jd�| j d � | j|jdd�d� | j
|jd�� | j t �� |d W d Q R X |j
� d S ) N�rz'The database is opened for reading only� xr r r s xxxs foo)r2 r r r r �assertWarnsRegex�DeprecationWarningr �getr Z assertIsNone� assertRaises�KeyErrorr! )r" r# r r r �test_dumbdbm_readR s
z!DumbDBMTestCase.test_dumbdbm_readc C s( | j � tjt�}| j|�}|j� d S )N)r2 r r r � keys_helperr! )r" r# r r r r �test_dumbdbm_keysd s
z!DumbDBMTestCase.test_dumbdbm_keysc C s* t jt�}d|d< | jd|� |j� d S )Ns hello� 1)r r r �assertInr! )r" r# r r r �test_write_containsj s
z#DumbDBMTestCase.test_write_containsc C sH t jt�}d|d< d|d< |j� t jt�}| j|d d� |j� d S )Ns hellor? s hello2)r r r r! r )r" r# r r r �test_write_write_readp s
z%DumbDBMTestCase.test_write_write_readc C s4 | j � tjtd�}| j|d | jdjd� � d S )Nr5 r zutf-8)r2 r r r r r �encode)r" r# r r r �
test_str_readz s z DumbDBMTestCase.test_str_readc C sx | j � tjt�}d|d< d|d< |j� tjtd�}| jd|� | j|djd� | jdjd� � | j|d d� d S ) Nr r �a�1r5 zutf-8r? r ) r2 r r r r! r@ r rC r )r" r# r r r �test_str_write_contains s
z'DumbDBMTestCase.test_str_write_containsc C s� t jt�}d|d< d|d< |j� tjtd d��}|j� }W d Q R X tjdkr^|jdd �}n |jd d�}tjtd d
��}|j |� W d Q R X t jt�}| j
|d d� | j
|d d� d S ) Ns hellor? s hello2� 2z.dir�rb�
�
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 s4 | j |�}x$| jD ]}| j| j| || � qW d S )N)r= r r )r" r# r r$ r r r r � s
zDumbDBMTestCase.read_helperc C s6 t jtd�}x | jD ]}| j| ||< qW |j� d S )N�n)r r r r r! )r" r# �kr r r r2 � s zDumbDBMTestCase.init_dbc C s* t |j� �}t | jj� �}| j||� |S )N)�sortedr r r )r" r# r Zdkeysr r r r= � s zDumbDBMTestCase.keys_helperc C s� dd l }i }x�td�D ]�}tjt�}xptd�D ]d}|jd�}|j � dk r^||kr�||= ||= q.|jd �|jd � }|||< |||< | j|| |� q.W |j� tjt�}t d
d � |j
� D ��}t |j
� �}| j||� |j� qW d S )
Nr � �d Z
abcdefghijklmg�������?r r r i' c s s | ]\}}|j d �|fV qdS )zlatin-1N)rC )�.0rV �vr r r � <genexpr>� s z.DumbDBMTestCase.test_random.<locals>.<genexpr>)r r r ) �random�ranger r r ZchoiceZ randranger r! rW �items) r" r] �dZdummyr# rV r[ ZexpectedZgotr r r � test_random� s(
zDumbDBMTestCase.test_randomc C sr t jtd��}d|d< W d Q R X t jtd�� }| jt|j� �dg� W d Q R X | jt j�� |j� W d Q R X d S )Nr zcontext managerzdumbdbm context managerr5 s dumbdbm context manager)r r r r r r r: �error)r" �dbr r r �test_context_manager� s z$DumbDBMTestCase.test_context_managerc C s� t jtd�}|j� xhttj|�ttj|d�ttj|�ttj |�fD ]6}| j
t j ��}|d� W d Q R X | j t
|j�d� qDW xXtjd�tjd�tjd�tfD ]6}| j
t j ��}||� W d Q R X | j t
|j�d� q�W d S )Nr �b�testz"DBM object has already been closedr Ziterkeysr_ )r r r r! r �operator�delitem�setitem�getitem�containsr: rb r �strZ exception� methodcaller�len)r" r# �meth�cmr r r �test_check_closed� s$
z!DumbDBMTestCase.test_check_closedc C sb t jtd��$}x | jD ]}| j| ||< qW W d Q R X t jtd��}| j|j� g � W d Q R X d S )NrU )r r r r r r )r" r# rV r r r �test_create_new� s
z DumbDBMTestCase.test_create_newc C sx t td d��}|jd� W d Q R X tj� �B}| jt�� tj t��}W d Q R X W d Q R X | j|j � d� W d Q R X d S )Nz.dirr1 zstr(print('Hacked!')), 0
� )
r r rQ r Zcaptured_stdoutr: �
ValueErrorr r �getvalue)r" �stream�stdoutr# r r r � test_eval� s
zDumbDBMTestCase.test_evalc C sD x>dD ]6}t � | jtd�� tjt|�}W d Q R X |j� qW d S )Nr5 r1 zIThe database file is missing, the semantics of the 'c' flag will be used.)r5 r1 )r
r7 r8 r r r r! )r" �valuer# r r r �test_warn_on_ignored_flags� s
z*DumbDBMTestCase.test_warn_on_ignored_flagsc C s> x8dD ]0}| j td�� tjt|�}W d Q R X |j� qW d S )N�x�rfz)Flag must be one of 'r', 'w', 'c', or 'n')r{ r| N)r7 r8 r r r r! )r" �flagr# r r r �test_invalid_flag s
z!DumbDBMTestCase.test_invalid_flagc C s� t j� ��}tjj|d�}tj|d��8}| jt|j � �g � x | j
D ]}| j
| ||< qBW W d Q R X tj |d t j
� tj |d t j
� tj |t j
t jB � tj|d��(}| jt|j � �t| j
�� |j� W d Q R X W d Q R X d S )Nrc rU z.dirz.datr5 )r Ztemp_dirr �path�joinr r r r r r r' r+ �S_IRUSR�S_IXUSRrW r! )r" �dirZfnamer# r$ r r r �test_readonly_files s
z#DumbDBMTestCase.test_readonly_filesz*requires OS support of non-ASCII encodingsc C s� t j}x dD ]}| jt j|| � q W tj|d��}d|d< W d Q R X | jtjj |d �� | jtjj |d �� tj|d��:}| j
t |j � �dg� | jd|k� | j
|d d� W d Q R X d S ) N�.dir�.dat�.bakr s values keyr5 )r� r� r� )
r �TESTFN_NONASCIIZ
addCleanupr r r Z
assertTruer r �existsr r r )r" �filename�suffixrc r r r �test_nonascii_filename s
z&DumbDBMTestCase.test_nonascii_filenamec 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 �setUp1 s zDumbDBMTestCase.setUpN)$�__name__�
__module__� __qualname__rC r r% �unittestZ
skipUnless�hasattrr r/ r0 r4 r< r> rA rB rD rG rT r r2 r= ra rd rq rr rx rz r~ r� r r� r� r� r� r r r r r sF
r �__main__)�__doc__rM rg r r+ r� �warningsZdbm.dumbZdumbr rf r � functoolsr ZTESTFNr r
ZTestCaser r� �mainr r r r �<module> s