File: //usr/local/lib/python3.7/test/__pycache__/test_shelve.cpython-37.opt-1.pyc
B
��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 )Nzlatin-1)�decode)�s� r �,/usr/local/lib/python3.7/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 )N)r r )r �keyr r r � __getitem__ s zbyteskeydict.__getitem__c C s || j t|�< d S )N)r r )r r �valuer r r � __setitem__ s zbyteskeydict.__setitem__c C s | j t|�= d S )N)r r )r r r r r � __delitem__ s zbyteskeydict.__delitem__c C s
t | j�S )N)�lenr )r r r r �__len__ s zbyteskeydict.__len__c c s$ x | j �� D ]}|�d�V q W d S )Nzlatin-1)r �keys�encode)r �kr r r �iterkeys s zbyteskeydict.iterkeysc C s t | �� �S )N)�listr )r r r r r # s zbyteskeydict.keysc C s
t | j�S )N)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 )�TestCasez shelftemp.dbc C s( x"t � | jd �D ]}t�|� qW d 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|� y |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 d |�� X d S )Nr )r( )r* r' r+ r, r- )r. �openr# r0 r1 )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 d |�� X d S )Nr* )r( )r* r' r+ r, r- )r. r7 r# r0 r1 )r r 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 d |�� X d S )Nr' )r( )r* r' r+ r, r- )r. r7 r# r0 r1 )r r r r r �test_proto2_file_shelfQ s
z TestCase.test_proto2_file_shelfc C s� t � }tj|dd�}d|d< | �|d d� |�� t � }tj|dd�}d|d< | �|d d� |�� | �t|�d� | �t|�d� | �|�� |�� � d S )Nr )r( )r* r' r+ r, r- r* )r
r. r/ r0 r1 r �assertNotEqual�items)r r5 r �d2r r r �test_in_memory_shelfY s z TestCase.test_in_memory_shelfc C s� t � }tj|ddd�}ddddg|d< | �|d ddddg� |d �d� | �|d ddddg� |�� t � }tj|dd d�}ddddg|d< | �|d ddddg� |d �d� | �|d dddddg� |�� | �t|�d� | �t|�d� d S )
Nr' F)r( r) r* r+ r, r- � T)r
r. r/ r0 �appendr1 r )r r5 r r= r r r �test_mutable_entryi 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öpr* zutf-8zlatin-1)Z keyencoding�ascii)r. r/ ZassertInr r2 �UnicodeEncodeErrorr )r r r r r r r �test_keyencoding} s zTestCase.test_keyencodingc C s` i }d}|� d�}tj|dd�}dg||< || }|d �d� |�� || }| �||� d S )Nr zutf-8T)r) r* r' )r r. r/ r@ r1 r; )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 d Q R X | �tt|� y |d W n tk
r� Y n X | �d� d S ) Nr' F)r( r) r* r+ r, r- z"Closed shelf should not find a key)r. r/ r0 r r2 r3 r4 )r r5 r r r r � test_with� s zTestCase.test_withc C s( t �i ��}| �|jd� W d Q R X d S )Nr+ )r. r/ r0 Z _protocol)r r r r r �test_default_protocol� s z TestCase.test_default_protocolN)r r r r# r& r6 r8 r9 r: r> rA rD rF rG rH r r r r r * s
r )�
mapping_testsc @ s: e Zd ZdZdZdd� ZejZdd� Z dd� Z
d d
� Z d S ) �TestShelveBasez shelftemp.dbr c O s g | _ tjj| f|�|� d S )N)�_dbrI �BasicTestMappingProtocolr
)r �args�kwr r r r
� s zTestShelveBase.__init__c C s dddd�S )NZvalue1r' )r* r' r+ )r- Zkey2Zkey3r )r r r r �
_reference� s zTestShelveBase._referencec C sT | j rtjt� f| j�}n*| jd7 _tj| jt| j� f| j�}| j �
|� |S )Nr* ) �_in_memr. r/ r
�_args�counterr7 r# �strrK r@ )r �xr r r �_empty_mapping� s z TestShelveBase._empty_mappingc C sL x| j D ] }|�� qW g | _ | jsHx"t�| jd �D ]}t�|� q6W d S )Nr! )rK r1 rP r" r# r r$ )r Zdbr% r r r r& � s zTestShelveBase.tearDownN) r r r r# rR r
r. r/ Z type2testrO rU r& r r r r rJ � s rJ c @ s e Zd ZddiZdZdS )�TestAsciiFileShelver( r FN)r r r rQ rP r r r r rV � s rV c @ s e Zd ZddiZdZdS )�TestBinaryFileShelver( r* FN)r r r rQ rP r r r r rW � s rW c @ s e Zd ZddiZdZdS )�TestProto2FileShelver( r' FN)r r r rQ rP r r r r rX � s rX c @ s e Zd ZddiZdZdS )�TestAsciiMemShelver( r TN)r r r rQ rP r r r r rY � s rY c @ s e Zd ZddiZdZdS )�TestBinaryMemShelver( r* TN)r r r rQ rP r r r r rZ � s rZ c @ s e Zd ZddiZdZdS )�TestProto2MemShelver( r' TN)r r r rQ rP r r r r r[ � s r[ c
C s* x$t � D ]} t�ttttttt � qW d S )N)
r r Z run_unittestrV rW rX rY rZ r[ r )�moduler r r � test_main� s r] �__main__)Zunittestr. r" Ztestr Zcollections.abcr Z
test.test_dbmr r r
r rI rL rJ rV rW rX rY rZ r[ r] r r r r r �<module> s&