File: //opt/cppython/lib/python3.8/test/__pycache__/test_shelve.cpython-38.pyc
U
.D�g� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ dd� Z G dd� de�Z
G d d
� d
e j �Z d d lm Z G d d
� d
e j
�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zdd� Zed kr�e� dS ) � N)�support)�MutableMapping)� dbm_iteratorc C s
| � d�S �N�latin-1)�decode)�s� r �//opt/cppython/lib/python3.8/test/test_shelve.py�L1 s r c @ sT e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e Z
dd� Z dd� Z dS )� byteskeydictz Mapping that supports bytes keysc C s
i | _ d S �N)�d��selfr r r
�__init__ s zbyteskeydict.__init__c C s | j t|� S r
�r r �r �keyr r r
� __getitem__ s zbyteskeydict.__getitem__c C s || j t|�< d S r
r )r r �valuer r r
� __setitem__ s zbyteskeydict.__setitem__c C s | j t|�= d S r
r r r r r
� __delitem__ s zbyteskeydict.__delitem__c C s
t | j�S r
)�lenr r r r r
�__len__ s zbyteskeydict.__len__c c s | j �� D ]}|�d�V q
d S r )r �keys�encode)r �kr r r
�iterkeys s zbyteskeydict.iterkeysc C s t | �� �S r
)�listr r r r r
r # s zbyteskeydict.keysc C s
t | j�S r
)r r r r r r
�copy& s zbyteskeydict.copyN)
�__name__�
__module__� __qualname__�__doc__r r r r r r �__iter__r r r r r r
r s r c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�TestCase� shelftemp.dbc C s$ t � | jd �D ]}t�|� qd S �N�*)�glob�fnr �unlink)r �fr r r
�tearDown. s zTestCase.tearDownc C s� i }t j|ddd�}ddddg|d< | �|d ddddg� | �t|�d� |�� | �tt|� z |d W n tk
r� Y n X | �d� d S � N� F��protocol� writeback� � � �key1z"Closed shelf should not find a key)�shelve�Shelf� assertEqualr �close� assertRaises�
ValueError�fail�r �d1r r r r
�
test_close2 s zTestCase.test_closec C s< t j| jdd�}z d|d< | �|d d� W 5 |�� X d S )Nr �r2 �r4 r0 r5 r6 r7 �r8 �openr+ r; r: �r r r r r
�test_ascii_file_shelfA s
z TestCase.test_ascii_file_shelfc C s< t j| jdd�}z d|d< | �|d d� W 5 |�� X d S )Nr4 rB rC r7 rD rF r r r
�test_binary_file_shelfI s
z TestCase.test_binary_file_shelfc C s< t j| jdd�}z d|d< | �|d d� W 5 |�� X d S )Nr0 rB rC r7 rD rF r r r
�test_proto2_file_shelfQ s
z TestCase.test_proto2_file_shelfc C s� t � }tj|dd�� }d|d< | �|d d� W 5 Q R X t � }tj|dd�� }d|d< | �|d d� W 5 Q R X | �t|�d� | �t|�d� | �|�� |�� � d S )Nr rB rC r7 r4 )r r8 r9 r: r �assertNotEqual�items�r r@ r Zd2r r r
�test_in_memory_shelfY s z TestCase.test_in_memory_shelfc C s t � }tj|ddd��T}ddddg|d< | �|d ddddg� |d �d� | �|d ddddg� W 5 Q R X t � }tj|dd d��V}ddddg|d< | �|d ddddg� |d �d� | �|d dddddg� W 5 Q R X | �t|�d� | �t|�d� d S )
Nr0 Fr1 r4 r5 r6 r7 � T)r r8 r9 r: �appendr rL r r r
�test_mutable_entryg s "$zTestCase.test_mutable_entryc C sv i }d}dgt �|�|< | �|�d�|� dgt j|dd�|< | �|�d�|� t j|dd�}| �t|j|dg� d S )Nu Pöpr4 �utf-8r )Z keyencoding�ascii)r8 r9 ZassertInr r< �UnicodeEncodeErrorr )r r r r r r r
�test_keyencodingy s zTestCase.test_keyencodingc C sd i }d}|� d�}tj|dd��&}dg||< || }|d �d� W 5 Q R X || }| �||� d S )Nr rQ T)r3 r4 r0 )r r8 r9 rO rJ )r r r �
encodedkeyr Zp1Zp2r r r
�&test_writeback_also_writes_immediately� s
z/TestCase.test_writeback_also_writes_immediatelyc C s� i }t j|ddd��>}ddddg|d< | �|d ddddg� | �t|�d� W 5 Q R X | �tt|� z |d W n tk
r� Y n X | �d� d S r/ )r8 r9 r: r r< r= r> r? r r r
� test_with� s zTestCase.test_withc C s( t �i ��}| �|jd� W 5 Q R X d S )Nr5 )r8 r9 r: Z _protocolrF r r r
�test_default_protocol� s z TestCase.test_default_protocolN)r! r" r# r+ r. rA rG rH rI rM rP rT rV rW rX r r r r
r& * s
r& )�
mapping_testsc @ s: e Zd ZdZdZdd� ZejZdd� Z dd� Z
d d
� Z d S ) �TestShelveBaser' r c O s g | _ tjj| f|�|� d S r
)�_dbrY �BasicTestMappingProtocolr )r �args�kwr r r
r � s zTestShelveBase.__init__c C s dddd�S )NZvalue1r0 )r4 r0 r5 )r7 Zkey2Zkey3r r r r r
�
_reference� s zTestShelveBase._referencec C sT | j rtjt� f| j�}n*| jd7 _tj| jt| j� f| j�}| j �
|� |S )Nr4 ) �_in_memr8 r9 r �_args�counterrE r+ �strr[ rO )r �xr r r
�_empty_mapping� s z TestShelveBase._empty_mappingc C sD | j D ] }|�� qg | _ | js@t�| jd �D ]}t�|� q0d S r( )r[ r; r` r* r+ r r, )r Zdbr- r r r
r. � s
zTestShelveBase.tearDownN) r! r" r# r+ rb r r8 r9 Z type2testr_ re r. r r r r
rZ � s rZ c @ s e Zd ZddiZdZdS )�TestAsciiFileShelver2 r FN�r! r" r# ra r` r r r r
rf � s rf c @ s e Zd ZddiZdZdS )�TestBinaryFileShelver2 r4 FNrg r r r r
rh � s rh c @ s e Zd ZddiZdZdS )�TestProto2FileShelver2 r0 FNrg r r r r
ri � s ri c @ s e Zd ZddiZdZdS )�TestAsciiMemShelver2 r TNrg r r r r
rj � s rj c @ s e Zd ZddiZdZdS )�TestBinaryMemShelver2 r4 TNrg r r r r
rk � s rk c @ s e Zd ZddiZdZdS )�TestProto2MemShelver2 r0 TNrg r r r r
rl � s rl c
C s& t � D ]} t�ttttttt � qd S r
)
r r Z run_unittestrf rh ri rj rk rl r& )�moduler r r
� test_main� s
�rn �__main__)Zunittestr8 r* �testr �collections.abcr Z
test.test_dbmr r r r& rY r\ rZ rf rh ri rj rk rl rn r! r r r r
�<module> s& z