File: //usr/local/lib/python3.7/test/__pycache__/test_dbm_dumb.cpython-37.opt-1.pyc
B
��g+ � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z e
j
Zdd� ZG dd� dej�Zed kr�e�� dS )
zATest script for the dumbdbm module
Original by Roger E. Masse
� N)�support)�partialc C s: x4dD ],} yt �t| � W q tk
r0 Y qX qW d S )N)z.dirz.datz.bak)�os�unlink�_fname�OSError)Zext� r �./usr/local/lib/python3.7/test/test_dbm_dumb.py�
_delete_files s
r
c @ sJ e Zd Zddddddddd d
d d d
dd�d�diZdd� Ze�ee d�d�e�ee 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 d@dA� Z e�ee d�d�dBdC� �Z e�e j!dD�dEdF� �Z"dGdH� Z#dIdJ� Z$dKS )L�DumbDBMTestCase� 0� � as Python:� bs Programming� cs the� ds way� fs Guido� gs intended� üzutf-8� !c C sT t �td�}| �t|�� �g � x | jD ]}| j| ||< q(W | �|� |�� 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 �d�}t�tdd�}|�� W d t �|� X d}t jdkrDd}dd l}t �td �}| �|� |j
�|� t �td �}| �|� |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 �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 s` | � � t�td�}d | jd< |d< | �|� | �|�dd�d� | �|d d� |�� d S )N�ws indentedr s xxxs foo) �init_dbr r r r r r �
setdefaultr )r r r r r �test_dumbdbm_modificationG s
z)DumbDBMTestCase.test_dumbdbm_modificationc C s� | � � t�td�}| �|� | �td�� d|d< W d Q R X | �td�� |d= W d Q R X | �|�d�| j d � | �|�dd�d� | �
|�d�� | � t �� |d W d Q R X |�
� d S ) N�rz'The database is opened for reading only� xr r r s xxxs foo)r/ r r r r �assertWarnsRegex�DeprecationWarningr �getr Z assertIsNone� assertRaises�KeyErrorr )r r r r r �test_dumbdbm_readQ s
z!DumbDBMTestCase.test_dumbdbm_readc C s( | � � t�t�}| �|�}|�� d S )N)r/ r r r � keys_helperr )r r r r r r �test_dumbdbm_keysc s
z!DumbDBMTestCase.test_dumbdbm_keysc C s* t �t�}d|d< | �d|� |�� d S )Ns hello� 1)r r r �assertInr )r r r r r �test_write_containsi s
z#DumbDBMTestCase.test_write_containsc C sH t �t�}d|d< d|d< |�� t �t�}| �|d d� |�� d S )Ns hellor< s hello2)r r r r r )r r r r r �test_write_write_reado s
z%DumbDBMTestCase.test_write_write_readc C s4 | � � t�td�}| �|d | jd�d� � d S )Nr2 r zutf-8)r/ r r r r r �encode)r r r r r �
test_str_ready s z DumbDBMTestCase.test_str_readc C sx | � � t�t�}d|d< d|d< |�� t�td�}| �d|� | �|d�d� | jd�d� � | �|d d� d S ) Nr r �a�1r2 zutf-8r< r ) r/ r r r r r= r r@ r )r r r r r �test_str_write_contains~ s
z'DumbDBMTestCase.test_str_write_containsc C s� t �t�}d|d< d|d< |�� t�td d��}|�� }W d Q R X tjdkr^|�dd �}n |�d d�}t�td d
��}|� |� W d Q R X t �t�}| �
|d d� | �
|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 | � |�}x$| jD ]}| �| j| || � qW d S )N)r: r r )r r r r! r r r r � s
zDumbDBMTestCase.read_helperc C s6 t �td�}x | jD ]}| j| ||< qW |�� d S )N�n)r r r r r )r r �kr r r r/ � s zDumbDBMTestCase.init_dbc C s* t |�� �}t | 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 ]�}t�t�}xptd�D ]d}|�d�}|� � dk r^||kr�||= ||= q.|�d�|�d� }|||< |||< | �|| |� q.W |�� t�t�}t dd � |�
� D ��}t |�
� �}| �||� |�� qW d S )
Nr � �d Z
abcdefghijklmg�������?)r r r i' c s s | ]\}}|� d �|fV qdS )zlatin-1N)r@ )�.0rS �vr r r � <genexpr>� s z.DumbDBMTestCase.test_random.<locals>.<genexpr>) �random�ranger r r �choiceZ randranger r rT �items) r rZ �dZdummyr rS rX ZexpectedZgotr r r � test_random� s(
zDumbDBMTestCase.test_randomc C sr t �td��}d|d< W d Q R X t �td�� }| �t|�� �dg� W d Q R X | �t j�� |�� W d Q R X d S )Nr zcontext managerzdumbdbm context managerr2 s dumbdbm context manager)r r r r r r r7 �error)r �dbr r r �test_context_manager� s z$DumbDBMTestCase.test_context_managerc
C s� t �td�}|�� xhttj|�ttj|d�ttj|�ttj |�fD ]6}| �
t j ��}|d� W d Q R X | � t
|j�d� qDW xXt�d�t�d�t�d�tfD ]6}| �
t j ��}||� W d Q R X | � 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�containsr7 r` r �strZ exception� methodcaller�len)r r Zmeth�cmr r r �test_check_closed� s$
z!DumbDBMTestCase.test_check_closedc C sb t �td��$}x | jD ]}| j| ||< qW W d Q R X t �td��}| �|�� g � W d Q R X d S )NrR )r r r r r r )r r rS r r r �test_create_new� s
z DumbDBMTestCase.test_create_newc C sx t td d��}|�d� W d Q R X t�� �B}| �t�� t� t��}W d Q R X W d Q R X | �|� � d� W d Q R X d S )Nz.dirr. zstr(print('Hacked!')), 0
� )
r r rN r Zcaptured_stdoutr7 �
ValueErrorr r �getvalue)r �stream�stdoutr r r r � test_eval� s
zDumbDBMTestCase.test_evalc
C sD x>dD ]6}t � | �td�� t�t|�}W d Q R X |�� qW d S )N)r2 r. zIThe database file is missing, the semantics of the 'c' flag will be used.)r
r4 r5 r r r r )r �valuer r r r �test_warn_on_ignored_flags� s
z*DumbDBMTestCase.test_warn_on_ignored_flagsc
C s� t �td��}W d Q R X t�td � xjdD ]b}| �td�� t �t|�}W d Q R X |�� | �tj �
td �|dk� | � tj �
td �� q,W d S )NrR z.dir)r2 r. zFThe index file is missing, the semantics of the 'c' flag will be used.r. z.bak) r r r r r r4 r5 r r �path�existsZ assertFalse)r r rv r r r �test_missing_index s
z"DumbDBMTestCase.test_missing_indexc
C s> x8dD ]0}| � td�� t�t|�}W d Q R X |�� qW d S )N)�xZrfNz)Flag must be one of 'r', 'w', 'c', or 'n')r4 r5 r r r r )r �flagr r r r �test_invalid_flag s
z!DumbDBMTestCase.test_invalid_flagc
C s� t �� ��}tj�|d�}t�|d��8}| �t|� � �g � x | j
D ]}| j
| ||< qBW W d Q R X t� |d t j
� t� |d t j
� t� |t j
t jB � t�|d��(}| �t|� � �t| j
�� |�� W d Q R X W d Q R X d S )Nra rR z.dirz.datr2 )r Ztemp_dirr rx �joinr r r r r r r$ r( �S_IRUSR�S_IXUSRrT 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 ]}| �t j|| � q W t�|d��}d|d< W d Q R X | �tj� |d �� | �tj� |d �� t�|d��:}| �
t |� � �dg� | �d|k� | �
|d d� W d Q R X d S )N)z.dirz.datz.bakr s values keyz.datz.dirr2 )
r �TESTFN_NONASCIIZ
addCleanupr r r Z
assertTruer rx ry r r r )r �filename�suffixra 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 �setUp> s zDumbDBMTestCase.setUpN)%�__name__�
__module__� __qualname__r@ r r" �unittestZ
skipUnless�hasattrr r, r- r1 r9 r; r> r? rA rD rQ r r/ r: r_ rb rn ro ru rw rz r} r� r r� r� r� r� r r r r r sH
r �__main__)�__doc__rJ re r r( r� Zdbm.dumbZdumbr rd r � functoolsr ZTESTFNr r
ZTestCaser r� �mainr r r r �<module> s -