File: //usr/local/lib/python3.10/test/__pycache__/test_shelve.cpython-310.opt-2.pyc
o
�i � @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
dd� ZG dd � d e �ZG d
d � d e j�ZG d d
� d
�ZG dd� de�ZG dd� de�Zd dlmZ eejd �D ]CZeejfZde� d�Zeeeddeii�e � e< eejfZe
� D ] Z e j dd� Z de� de � d�Zeeee deid��e � e< q�qie d kr�e � � dS dS ) � N)�support)� os_helper)�MutableMapping)� dbm_iteratorc C s
| � d�S �N�latin-1)�decode)�s� r
�-/usr/local/lib/python3.10/test/test_shelve.py�L1
�
r c @ sR e Zd Z dd� 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__ r
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__ � zbyteskeydict.__delitem__c C �
t | j�S r )�lenr r r
r
r �__len__ r
zbyteskeydict.__len__c c s" � | j �� D ]}|�d�V qd S r )r �keys�encode)r �kr
r
r �iterkeys" s ��zbyteskeydict.iterkeysc C s t | �� �S r )�listr" r r
r
r r ( � zbyteskeydict.keysc C r r )r r r r
r
r �copy+ r
zbyteskeydict.copyN) �__name__�
__module__� __qualname__r r r r r r" �__iter__r r% r
r
r
r r s r c @ s| e Zd ZejZej�ejd�Z dd� Z
ddd�Z dd� Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�TestCase� shelftemp.dbc C s� i }t j|ddd�}g d�|d< | �|d g d�� | �t|�d� |�� | �tt|� z|d W n
ty= Y d S w | �d� d S �N� F��protocol� writeback�� r- � � �key1r2 z"Closed shelf should not find a key)�shelve�Shelf� assertEqualr �close� assertRaises�
ValueError�fail�r �d1r r
r
r �
test_close3 s �zTestCase.test_closeNc C s^ t �| j� | �tj| j� tj| j|d�}zd|d< | � |d d� W |�
� d S |�
� w )N�r/ r1 r5 ) �os�mkdir�dirname�
addCleanupr �rmtreer6 �open�fnr8 r9 )r r/ r r
r
r �test_open_templateB s zTestCase.test_open_templatec C � | j dd� d S )Nr r@ �rH r r
r
r �test_ascii_file_shelfL r z TestCase.test_ascii_file_shelfc C rI )Nr2 r@ rJ r r
r
r �test_binary_file_shelfO r z TestCase.test_binary_file_shelfc C rI )Nr- r@ rJ r r
r
r �test_proto2_file_shelfR r z TestCase.test_proto2_file_shelfc C s� t � }tj|dd��}d|d< | �|d d� W d � n1 s!w Y t � }tj|dd��}d|d< | �|d d� W d � n1 sGw Y | �t|�d� | �t|�d� | �|�� |�� � d S )Nr r@ r1 r5 r2 )r r6 r7 r8 r �assertNotEqual�items�r r> r Zd2r
r
r �test_in_memory_shelfU s � �z TestCase.test_in_memory_shelfc C s t � }tj|ddd��)}g d�|d< | �|d g d�� |d �d� | �|d g d�� W d � n1 s7w Y t � }tj|ddd��)}g d�|d< | �|d g d�� |d �d� | �|d g d�� W d � n1 ssw Y | �t|�d � | �t|�d � d S )
Nr- Fr. r1 r5 � T)r2 r- r3 r4 rR r2 )r r6 r7 r8 �appendr rP r
r
r �test_mutable_entryc 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öpr2 �utf-8r )Z keyencoding�ascii)r6 r7 ZassertInr r: �UnicodeEncodeErrorr )r r r r r
r
r �test_keyencodingu s zTestCase.test_keyencodingc C sx i }d}|� d�}tj|dd��}dg||< || }|d �d� W d � n1 s+w Y || }| �||� d S )Nr rU T)r0 r2 r- )r r6 r7 rS rN )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�� }g d�|d< | �|d g d�� | �t|�d� W d � n1 s-w Y | �tt|� z|d W n
tyI Y d S w | �d� d S r, )r6 r7 r8 r r: r; r< r= r
r
r � test_with� s � �zTestCase.test_withc C s@ t �i ��}| �|jtj� W d � d S 1 sw Y d S r )r6 r7 r8 Z _protocol�pickle�DEFAULT_PROTOCOL)r r r
r
r �test_default_protocol� s "�z TestCase.test_default_protocolr )r& r' r( r �TESTFNrC rA �path�joinrG r? rH rK rL rM rQ rT rX rZ r[ r^ r
r
r
r r* / s
r* c @ s e Zd ZejZdd� ZdS )�TestShelveBasec C s dddd�S )NZvalue1r- )r2 r- r3 )r5 Zkey2Zkey3r
r r
r
r �
_reference� r$ zTestShelveBase._referenceN)r& r' r( r6 r7 Z type2testrc r
r
r
r rb � s rb c @ s e Zd Zdd� ZdS )�TestShelveInMemBasec C s t jt� fi | j��S r )r6 r7 r �_argsr r
r
r �_empty_mapping� s z"TestShelveInMemBase._empty_mappingN)r& r' r( rf r
r
r
r rd � s rd c @ s e Zd ZdZdd� Zdd� ZdS )�TestShelveFileBaser c C s> | j d7 _ tj| jt| j � fi | j��}| �|j� |S )Nr2 )�counterr6 rF � base_path�strre rD r9 )r �xr
r
r rf � s z!TestShelveFileBase._empty_mappingc C sL t j}t�|� | �t j|� tj�|d�| _| �t t
dt
j � | j t
_ d S )Nr+ � _defaultmod)
r r_ rA rB rD rE r` ra ri �setattr�dbmrl �dbm_mod)r rC r
r
r �setUp� s
zTestShelveFileBase.setUpN)r& r' r( rh rf rp r
r
r
r rg � s rg )�
mapping_testsr2 Z TestProtoZ MemShelvere r/ r4 ZFile_ZShelve)ro re �__main__)!Zunittestrn rA r6 �globr\ �testr Z test.supportr �collections.abcr Z
test.test_dbmr r r r* rb rd rg rq �range�HIGHEST_PROTOCOL�protoZBasicTestMappingProtocol�bases�name�type�globalsro r& �suffix�mainr
r
r
r �<module> sD r
�
�� �