File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/archive.cpython-39.opt-1.pyc
a
HZ�h/ �
@ s d dl Z d dlmZ d dlmZmZ d dlmZ d dlmZ d dl Z ddl
m Z dd l m
Z
dd
lmZmZmZmZmZmZ dd lmZmZmZmZ dd lmZ dd
lmZ ddl m Z m Z ddl m Z m!Z! e �"� dd� �Z#dd� Z$dHdd�Z%dd� Z&dd� Z'G dd � d �Z(G d d � d e �Z)G d d � d e �Z*G d!d"� d"e �Z+e j"d#d$� �Z,e j-�.d%g d&�d'd(� dd d)d*d+i g d,gd-d.d/d0d1d2id1d2gd3g D � �d4d5� �Z/e0e1e��Z2e j-j.d%d6d(� d7d8ied9d:� e2D ��ed;d:� e2D ��fD �g d<�d=�d>d?� �Z3d@dA� Z4dBdC� Z5dDdE� Z6dFdG� Z7dS )I� N)� OrderedDict)�datetime�timezone)�StringIO)�Mock� )� BaseTestCase� )� PlaintextKey)�Archive�CacheChunkBuffer�RobustUnpacker�valid_msgpacked_dict� ITEM_KEYS�
Statistics)�
BackupOSError� backup_io�backup_io_iter�get_item_uid_gid)�Manifest)�msgpack)�Item� ArchiveItem)�uid2user� gid2groupc C s t � } | jdddd� | S )N� �
T��unique)r �update��stats� r" �</usr/lib64/python3.9/site-packages/borg/testsuite/archive.pyr! s r! c C s | j dddd� d S )Nr r Fr )r r r" r" r# �test_stats_basic s r$ �P c C s� |� dt|�� t� }| j|d� d}d|t|� }t� }| jdddd� | jtd d
�d|d � d }d|t|� }t� }| jtd
d
�d|d � d}d|t|� }d S )N�COLUMNS)�streamz20 B O 10 B C 10 B D 0 N � i� r Fr �foo��path)�item�finalr' z 1.02 kB O 10 B C 10 B D 0 N fooZxfoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoozP1.02 kB O 10 B C 10 B D 0 N foofoofoofoofoofoofoofo...oofoofoofoofoofoofoofoofoo)Zsetenv�strr �
show_progress�lenr r )r! Z monkeypatch�columns�out�s�bufr" r" r# �tests_stats_progress# s r5 c C s | j � }d S �N)Z osize_fmt)r! r3 r" r" r# �test_stats_format8 s r7 c C sT d| _ t� }| jtdd�|d� t�|�� �}t� }| j|dd� t�|�� �}d S )NTr) r* )r, r' )r' r- )Z output_jsonr r/ r �json�loads�getvalue)r! r2 �resultr" r" r# �test_stats_progress_jsonA s
r< c @ s, e Zd ZG dd� d�Zdd� Zd dd�ZdS )
� MockCachec @ s e Zd Zddd�ZdS )zMockCache.MockRepoTc C s d S r6 r" )�self�waitr" r" r# �async_response` s z!MockCache.MockRepo.async_responseN)T)�__name__�
__module__� __qualname__r@ r" r" r" r# �MockRepo_ s rD c C s i | _ | �� | _d S r6 )�objectsrD �
repository�r> r" r" r# �__init__c s zMockCache.__init__NTc C s || j |< |t|�t|�fS r6 )rE r0 )r> �id�chunkr! r? r" r" r# � add_chunkg s
zMockCache.add_chunk)NT)rA rB rC rD rH rK r" r" r" r# r= ] s r= c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ArchiveTimestampTestCasec C sH t � }t|�}t||�}t|||ddd�}t|d�|_| �|j|� d S )N�testT)Zcreate)�time)r r
r r r �metadata� assert_equal�ts)r> Z isoformatZexpectedrF �keyZmanifest�ar" r" r# �_test_timestamp_parsingn s
z0ArchiveTimestampTestCase._test_timestamp_parsingc C s$ | � dtdddddddtj�� d S )Nz1970-01-01T00:00:01.000001� r r �rT r r ZutcrG r" r" r# �test_with_microsecondsv s �z/ArchiveTimestampTestCase.test_with_microsecondsc C s$ | � dtdddddddtj�� d S )Nz1970-01-01T00:00:01rU r r rV rG r" r" r# �test_without_microseconds{ s �z2ArchiveTimestampTestCase.test_without_microsecondsN)rA rB rC rT rW rX r" r" r" r# rL l s rL c @ s e Zd Zdd� Zdd� ZdS )�ChunkBufferTestCasec C s� t dd�t dd�g}t� }td �}t||d �}|D ]}|�|� |�� q2|jdd� | �t|j�d� t �
� }|jD ]}|� |j | � qv| �|dd� |D �� d S ) NZp1r* Zp2T��flushr c S s g | ]}t |d ��qS �)Z
internal_dict�r ��.0�dr" r" r# �
<listcomp>� � z,ChunkBufferTestCase.test.<locals>.<listcomp>�
r r= r
r �addr[ rP r0 �chunksr ZUnpacker�feedrE )r> �data�cacherR re r` �unpackerrI r" r" r# rM � s
zChunkBufferTestCase.testc C s� dd }t d|d�t d|d�g}t� }td �}t||d �}|D ]}|�|� q>|jdd� | �t|j�d� |jd d� | �t|j�d
� t �
� }|jD ]}|� |j | � q�| �|d d � |D �� d S )
NZ$0123456789abcdefghijklmnopqrstuvwxyzi�a Zfull)r+ �source�partialFrZ � T� c S s g | ]}t |d ��qS r\ r] r^ r" r" r# ra � rb z4ChunkBufferTestCase.test_partial.<locals>.<listcomp>rc ) r> �bigrg rh rR re r` ri rI r" r" r# � test_partial� s
z ChunkBufferTestCase.test_partialN)rA rB rC rM ro r" r" r" r# rY � s rY c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RobustUnpackerTestCasec C s d� dd� |D ��S )Nrb c s s | ]}t �d |i�V qdS )r+ N�r Zpackb)r_ r, r" r" r# � <genexpr>� rb z5RobustUnpackerTestCase.make_chunks.<locals>.<genexpr>)�join)r> �itemsr" r" r# � make_chunks� s z"RobustUnpackerTestCase.make_chunksc C s t |t�o|�d�dv S )N� path�� foo� bar� boo� baz)�
isinstance�dict�get)r> �valuer" r" r# �
_validator� s z!RobustUnpackerTestCase._validatorc C sX t | jtd�}g }|D ]<\}}|r*|�� |D ]"}|�|� |D ]}|�|� q@q.q|S )N)Z validatorZ item_keys)r
r� r Zresyncrf �append)r> �inputri r; Z should_syncre rg r, r" r" r# �process� s
z RobustUnpackerTestCase.processc C sn d| � ddg�gfddg| � ddg�g fg}| �|�}| �|ddiddidd d
d d dd ddiddig � d S )
NFrx ry s garbagerz r{ rv �g �a �r �b �e )ru r� rP )r> re r; r" r" r# �test_extra_garbage_no_sync� s �
�z1RobustUnpackerTestCase.test_extra_garbage_no_syncc C s, g }|r(|� |d |� � ||d � }q|S r6 )r� )r> �left�length�partsr" r" r# �split� s
z RobustUnpackerTestCase.splitc C sR | � | �g d��d�}d|fg}| �|�}| �|ddiddiddiddig� d S ) Nrw r Frv rx ry rz r{ �r� ru r� rP �r> re r� r; r" r" r# �test_correct_stream� s
z*RobustUnpackerTestCase.test_correct_streamc C sb | � | �g d��d�}d|d d� fd|dd � fg}| �|�}| �|ddiddidd ig� d S )
Nrw rm Frl Trv rx rz r{ r� r� r" r" r# �test_missing_chunk� s
z)RobustUnpackerTestCase.test_missing_chunkc C sj | � | �g d��d�}d|d d� fdddg|dd � fg}| �|�}| �|dd idd
idd ig� d S ) Nrw rm Frl Ts gars bagerv rx rz r{ r� r� r" r" r# �test_corrupt_chunk� s (
z)RobustUnpackerTestCase.test_corrupt_chunkN) rA rB rC ru r� r� r� r� r� r� r� r" r" r" r# rp � s
rp c C s dd� t D �S )Nc S s g | ]}t �|��qS r" rq )r_ �namer" r" r# ra � rb z(item_keys_serialized.<locals>.<listcomp>)r r" r" r" r# �item_keys_serialized� s r� �packed)rb � x� foobarc C s g | ]}t �|��qS r" rq �r_ �or" r" r# ra � rb ra g F� r" �* g�Q� �k7@Tr� rx ry )rx ry c C s d S r6 r" �r� r� r" r" r# �test_invalid_msgpacked_item� s r� c C s g | ]}t �|��qS r" rq r� r" r" r# ra � rb rv s /a/b/cc c s | ]}|d fV qdS )rb Nr" �r_ �kr" r" r# rr � rb rr c c s | ]}|d fV qdS )s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxNr" r� r" r" r# rr � rb )Zminimalz empty-valuesz long-values)Zidsc C s d S r6 r" r� r" r" r# �test_valid_msgpacked_items� s r� c C s d} | di}t �| �g}d S )Ns xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxrb rq )rR rg r� r" r" r# � test_key_length_msgpacked_items� s r� c C sZ t �t��< t� td��W d � n1 s.0 Y W d � n1 sL0 Y d S )N�{ )�pytest�raisesr r �OSErrorr" r" r" r# �test_backup_io s r� c C sh G dd� d�} | t �}t�t�� t|�D ]}q*W d � n1 sD0 Y | t�}t|�D ]}q^d S )Nc @ s e Zd Zdd� Zdd� ZdS )z%test_backup_io_iter.<locals>.Iteratorc S s
|| _ d S r6 ��exc)r> r� r" r" r# rH
s z.test_backup_io_iter.<locals>.Iterator.__init__c S s | � � �d S r6 r� rG r" r" r# �__next__ s z.test_backup_io_iter.<locals>.Iterator.__next__N)rA rB rC rH r� r" r" r" r# �Iterator s r� )r� r� r� r r �
StopIteration)r� Zoserror_iterator�_Znormal_iteratorr" r" r# �test_backup_io_iter s r� c C s� t d�td� } }tddd| |d�}t|dd�\}}t|dd�\}}t|dd d
d �\}}tdd d
| |d�}t|dd�\}}t|dddd�\}}tdddddd�}t|dd�\}}t|dddd�\}}tdddddd�}t|dd�\}}t|dddd�\}}d S )Nr �filenamer r )r+ �uid�gid�user�groupF)�numericTrl rm )r� Z
uid_forcedZ
gid_forced������� � )r� Z uid_defaultZ gid_default������Z
udoesnotexistZ
gdoesnotexist� � � r � � )r r r r )Zuser0Zgroup0r, r� r� r" r" r# �test_get_item_uid_gid s r� )r% )8r8 � collectionsr r r �ior Z
unittest.mockr r� r� r Z
crypto.keyr
�archiver r r
r r r r r r r Zhelpersr r r, r r �platformr r Zfixturer! r$ r5 r7 r< r= rL rY rp r� ZmarkZ parametrizer� �sorted�listZIKr� r� r� r� r� r" r" r" r# �<module> sb
'8
����
��