Your IP : 216.73.216.85
U
i�ft* � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl
m Z ddl m
Z
e jZdd� ZG dd� dej�Zed kr�e�� dS )
zATest script for the dumbdbm module
Original by Roger E. Masse
� N)�support)�partialc C s6 dD ],} zt �t| � W q tk
r. Y qX qd S )N��.dir�.dat�.bak)�os�unlink�_fname�OSError)Zext� r �7/opt/alt/python38/lib64/python3.8/test/test_dbm_dumb.py�
_delete_files s
r c @ s& e Zd Zddddddddd d
d d d
dd�d�diZdd� Ze�ee d�d�dd� �Z
dd� Z dd� Z d d � Z
d d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� Zd8d9� Zd:d;� Z d<d=� Z d>d?� Z d@dA� Z e�e j!dB�dCdD� �Z"dEdF� Z#dGdH� Z$dIS )J�DumbDBMTestCase� 0� � as Python:� bs Programming� cs the� ds way� fs Guido� gs intended� ü�utf-8� !c C sZ t �t�td���>}| �t|�� �g � | jD ]}| j| ||< q.| � |� W 5 Q R X d S )N�c)
�
contextlib�closing�dumbdbm�openr
� assertEqual�list�keys�_dict� read_helper)�self�f�keyr r r
�test_dumbdbm_creation# s
z%DumbDBMTestCase.test_dumbdbm_creation�umaskztest needs os.umask()c C s� z$t �d�}t�tdd�}|�� W 5 t �|� X d}t jdkrDd}dd l}t �td �}| �|� |j
�|� t �td �}| �|� |j
�|� d S )
N� r i� i� �posixi� r r r ) r r) r r r
�close�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 �t�}d|d< | �|d d� |�� |�� d S )Nr r )r r r
r r, �r% r& r r r
�test_close_twice? s
z DumbDBMTestCase.test_close_twicec C sj | � � t�t�td���F}d | jd< |d< | �|� | �|� dd�d� | �|d d� W 5 Q R X d S )N�ws indentedr � xxx� foo)
�init_dbr r r r r
r# r$ r �
setdefaultr3 r r r
�test_dumbdbm_modificationF s
z)DumbDBMTestCase.test_dumbdbm_modificationc
C s� | � � t�t�td����}| �|� | �tjd�� d|d< W 5 Q R X | �tjd�� |d= W 5 Q R X | � |�
d�| j d � | � |�
dd�d� | � |�
d�� | �
t�� |d W 5 Q R X W 5 Q R X d S )N�rz'The database is opened for reading only� xr r r6 r7 )r8 r r r r r
r$ �assertRaisesRegex�errorr �getr# Z assertIsNone� assertRaises�KeyErrorr3 r r r
�test_dumbdbm_readO s
�� z!DumbDBMTestCase.test_dumbdbm_readc C s2 | � � t�t�t���}| �|�}W 5 Q R X d S �N)r8 r r r r r
� keys_helper)r% r& r" r r r
�test_dumbdbm_keys` s z!DumbDBMTestCase.test_dumbdbm_keysc C s4 t �t�t���}d|d< | �d|� W 5 Q R X d S )N� hello� 1)r r r r r
�assertInr3 r r r
�test_write_containse s z#DumbDBMTestCase.test_write_containsc C s\ t �t�t���}d|d< d|d< W 5 Q R X t �t�t���}| �|d d� W 5 Q R X d S )NrF rG � hello2)r r r r r
r r3 r r r
�test_write_write_readj s
z%DumbDBMTestCase.test_write_write_readc C sF | � � t�t�td���"}| �|d | jd�d� � W 5 Q R X d S )Nr; r r ) r8 r r r r r
r r# �encoder3 r r r
�
test_str_readr s z DumbDBMTestCase.test_str_readc C s� | � � t�t�t���}d|d< d|d< W 5 Q R X t�t�td���D}| �d|� | �|d�d� | j d�d� � | �|d d� W 5 Q R X d S ) Nr r �a�1r; r rG r )
r8 r r r r r
rH r rL r# r3 r r r
�test_str_write_containsw s �z'DumbDBMTestCase.test_str_write_containsc C s� t �t�t���}d|d< d|d< W 5 Q R X t�td d��}|�� }W 5 Q R X tjdkrh|� dd �}n |� d d�}t�td d
��}|�
|� W 5 Q R X t�t�}| � |d d� | � |d d� d S ) NrF rG rJ � 2r �rb�
�
s
�wb) r 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 s0 | � |�}| jD ]}| �| j| || � qd S rC )rD r# r )r% r& r" r' r r r
r$ � s
zDumbDBMTestCase.read_helperc C s< t �t�td��� }| jD ]}| j| ||< qW 5 Q R X d S �N�n)r r r r r
r# �r% r& �kr r r
r8 � s
zDumbDBMTestCase.init_dbc C s* t |�� �}t | j�� �}| �||� |S rC )�sortedr" r# r )r% r& r" Zdkeysr r r
rD � s zDumbDBMTestCase.keys_helperc
C s� dd l }i }td�D ]�}t�t�t���t}td�D ]d}|�d�}|� � dk rb||kr�||= ||= q2|�d�|�d� }|||< |||< | � || |� q2W 5 Q R X t�t�t���4}t
dd � |� � D ��}t
|� � �}| � ||� W 5 Q R X qd S )
Nr � �d Z
abcdefghijklmg�������?)r r r i' c s s | ]\}}|� d �|fV qdS )zlatin-1N)rL )�.0ra �vr r r
� <genexpr>� s z.DumbDBMTestCase.test_random.<locals>.<genexpr>) �random�ranger r r r r
ZchoiceZ randranger rb �items) r% rh �dZdummyr& ra rf ZexpectedZgotr r r
� test_random� s$
zDumbDBMTestCase.test_randomc C sr t �td��}d|d< W 5 Q R X t �td�� }| �t|�� �dg� W 5 Q R X | �t j�� |�� W 5 Q R X d S )Nr zcontext managerzdumbdbm context managerr; s dumbdbm context manager)r r r
r r! r" r@ r> )r% �dbr r r
�test_context_manager� s z$DumbDBMTestCase.test_context_managerc
C s� t �td�}|�� ttj|�ttj|d�ttj|�ttj |�fD ]6}| �
t j ��}|d� W 5 Q R X | � t
|j�d� qBt�d�t�d�t�d�tfD ]6}| �
t j ��}||� W 5 Q R X | � t
|j�d� q�d S )Nr �b�testz"DBM object has already been closedr" Ziterkeysrj )r r r
r, r �operator�delitem�setitem�getitem�containsr@ r> r �strZ exception� methodcaller�len)r% r& Zmeth�cmr r r
�test_check_closed� s,
� �� �z!DumbDBMTestCase.test_check_closedc C s^ t �td�� }| jD ]}| j| ||< qW 5 Q R X t �td��}| �|�� g � W 5 Q R X d S r^ )r r r
r# r r" r` r r r
�test_create_new� s
z DumbDBMTestCase.test_create_newc C sx t td d��}|�d� W 5 Q R X t�� �B}| �t�� t� t��}W 5 Q R X W 5 Q R X | �|� � d� W 5 Q R X d S )Nr r5 zstr(print('Hacked!')), 0
� )
r r
rZ r Zcaptured_stdoutr@ �
ValueErrorr r �getvalue)r% �stream�stdoutr& r r r
� test_eval� s
zDumbDBMTestCase.test_evalc
C sb dD ]X}t � | �t�� t�t|� W 5 Q R X | �tj� td �� | �tj� td �� qd S )N�r; r5 r r )
r r@ �FileNotFoundErrorr r r
� assertFalser �path�exists)r% �valuer r r
�test_missing_data� s z!DumbDBMTestCase.test_missing_datac
C s� t �td��}W 5 Q R X t�td � dD ]R}| �t�� t �t|� W 5 Q R X | �tj� td �� | �tj� td �� q*d S )Nr_ r r� r )
r r r
r r r@ r� r� r� r� )r% r& r� r r r
�test_missing_index� s
z"DumbDBMTestCase.test_missing_indexc
C s2 dD ](}| � td�� t�t|� W 5 Q R X qd S )N)�xZrfNz)Flag must be one of 'r', 'w', 'c', or 'n')r= r} r r r
)r% �flagr r r
�test_invalid_flag s
�z!DumbDBMTestCase.test_invalid_flagc
C s� t �� ��}tj�|d�}t�|d��4}| �t|� � �g � | j
D ]}| j
| ||< q@W 5 Q R X t� |d t j
� t� |d t j
� t� |t j
t jB � t�|d��(}| �t|� � �t| j
�� |�� W 5 Q R X W 5 Q R X d S )Nrm r_ r r r; )r Ztemp_dirr r� �joinr r r r! r" r# �chmodr. �S_IRUSR�S_IXUSRrb 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}dD ]}| �t j|| � q
t�|d��}d|d< W 5 Q R X | �tj� |d �� | �tj� |d �� t�|d��:}| �
t |� � �dg� | �d|k� | �
|d d� W 5 Q R X d S )Nr r s values keyr r r; )
r �TESTFN_NONASCIIZ
addCleanupr r r Z
assertTruer r� r� r r! r" )r% �filename�suffixrm r r r
�test_nonascii_filename s z&DumbDBMTestCase.test_nonascii_filenamec C s
t � d S rC �r �r% r r r
�tearDown' s zDumbDBMTestCase.tearDownc C s
t � d S rC r� r� r r r
�setUp* s zDumbDBMTestCase.setUpN)%�__name__�
__module__� __qualname__rL r# r( �unittestZ
skipUnless�hasattrr r2 r4 r: rB rE rI rK rM rP r] r$ r8 rD rl rn rz r{ r� r� r� r� r� r r� r� r� r� r r r r
r s\ �
�
r �__main__)�__doc__r rV rq r r. r� Zdbm.dumbZdumbr rp r � functoolsr ZTESTFNr
r ZTestCaser r� �mainr r r r
�<module> s