File: //opt/cppython/lib/python3.8/test/__pycache__/test_shelve.cpython-38.opt-2.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 @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � ZeZ d
d� Z
dd� Z dS )� byteskeydictc 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__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 �r1 �r3 r/ r4 r5 r6 �r7 �openr* r: r9 �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 )Nr3 rA rB r6 rC rE 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 )Nr/ rA rB r6 rC rE 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 rA rB r6 r3 )r r7 r8 r9 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 )
Nr/ Fr0 r3 r4 r5 r6 � T)r r7 r8 r9 �appendr rK 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öpr3 �utf-8r )Z keyencoding�ascii)r7 r8 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 rP T)r2 r3 r/ )r r7 r8 rN rI )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. )r7 r8 r9 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 )Nr4 )r7 r8 r9 Z _protocolrE r r r
�test_default_protocol� s z TestCase.test_default_protocolN)r! r" r# r* r- r@ rF rG rH rL rO rS rU rV rW 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
)�_dbrX �BasicTestMappingProtocolr )r �args�kwr r r
r � s zTestShelveBase.__init__c C s dddd�S )NZvalue1r/ )r3 r/ r4 )r6 Zkey2Zkey3r r r r r
�
_reference� s zTestShelveBase._referencec C sT | j rtjt� f| j�}n*| jd7 _tj| jt| j� f| j�}| j �
|� |S )Nr3 ) �_in_memr7 r8 r �_args�counterrD r* �strrZ rN )r �xr r r
�_empty_mapping� s z TestShelveBase._empty_mappingc C sD | j D ] }|�� qg | _ | js@t�| jd �D ]}t�|� q0d S r' )rZ r: r_ r) r* r r+ )r Zdbr, r r r
r- � s
zTestShelveBase.tearDownN) r! r" r# r* ra r r7 r8 Z type2testr^ rd r- r r r r
rY � s rY c @ s e Zd ZddiZdZdS )�TestAsciiFileShelver1 r FN�r! r" r# r` r_ r r r r
re � s re c @ s e Zd ZddiZdZdS )�TestBinaryFileShelver1 r3 FNrf r r r r
rg � s rg c @ s e Zd ZddiZdZdS )�TestProto2FileShelver1 r/ FNrf r r r r
rh � s rh c @ s e Zd ZddiZdZdS )�TestAsciiMemShelver1 r TNrf r r r r
ri � s ri c @ s e Zd ZddiZdZdS )�TestBinaryMemShelver1 r3 TNrf r r r r
rj � s rj c @ s e Zd ZddiZdZdS )�TestProto2MemShelver1 r/ TNrf r r r r
rk � s rk c
C s& t � D ]} t�ttttttt � qd S r
)
r r Z run_unittestre rg rh ri rj rk r% )�moduler r r
� test_main� s
�rm �__main__)Zunittestr7 r) �testr �collections.abcr Z
test.test_dbmr r r r% rX r[ rY re rg rh ri rj rk rm r! r r r r
�<module> s& z