File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/archive.cpython-36.opt-1.pyc
3
y��b% �
@ 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 lmZmZmZ dd lmZ dd
lmZ ddlm Z m Z e j � dd� �Z dd� Z dBdd�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)j*d$d%d&d'gd(d)� dd d*d+d,i g f gd-d.d/d'd0d1id0d1gdCg D � �d2d3� �Z+e,e-e��Z.e j)j*d$d4d)� d5d6ied7d8� e.D ��ed9d8� e.D ��gD ��d:d;� �Z/d<d=� Z0d>d?� Z1d@dA� Z2dS )D� N)� OrderedDict)�datetime�timezone)�StringIO)�Mock� )� BaseTestCase� )� PlaintextKey)�Archive�CacheChunkBuffer�RobustUnpacker�valid_msgpacked_dict� ITEM_KEYS�
Statistics)�
BackupOSError� backup_io�backup_io_iter)�Manifest)�msgpack)�Item� ArchiveItemc C s t � } | jdddd� | S )N� �
T)�unique)r �update)�stats� r � /usr/lib64/python3.6/archive.pyr s r c C s | j dddd� d S )Nr r F)r )r )r r r r �test_stats_basic s r �P c C s� |j dt|�� t� }| j|d� d}d|t|� }t� }| jdddd � | jtd
d �d|d � d
}d|t|� }t� }| jtd
d d �d|d � d}d|t|� }d S )NZCOLUMNS)�streamz20 B O 10 B C 10 B D 0 N � r � r F)r �foo)�path)�item�finalr! z 1.02 kB O 10 B C 10 B D 0 N foo�( zP1.02 kB O 10 B C 10 B D 0 N foofoofoofoofoofoofoofo...oofoofoofoofoofoofoofoofooi� )Zsetenv�strr �
show_progress�lenr r )r Z monkeypatch�columns�out�sZbufr r r �tests_stats_progress" s r/ c C s dj | �}d S )Nz
{0.osize_fmt})�format)r r. r r r �test_stats_format7 s
r1 c C sT d| _ t� }| jtdd�|d� tj|j� �}t� }| j|dd� tj|j� �}d S )NTr$ )r% )r& r! )r! r' )Z output_jsonr r* r �json�loads�getvalue)r r- �resultr r r �test_stats_progress_json@ s
r6 c @ s, e Zd ZG dd� d�Zdd� Zd dd�ZdS )
� MockCachec @ s e Zd Zddd�ZdS )zMockCache.MockRepoTc C s d S )Nr )�self�waitr r r �async_response_ s z!MockCache.MockRepo.async_responseN)T)�__name__�
__module__� __qualname__r: r r r r �MockRepo^ s r> c C s i | _ | j� | _d S )N)�objectsr> �
repository)r8 r r r �__init__b s zMockCache.__init__NTc C s || j |< |t|�t|�fS )N)r? r+ )r8 �id�chunkr r9 r r r � add_chunkf s
zMockCache.add_chunk)NT)r; r< r= r> rA rD r r r r r7 \ s r7 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ArchiveTimestampTestCasec C sH t � }t|�}t||�}t|||ddd�}t|d�|_| j|j|� d S )N�testT)Zcreate)Ztime)r r
r r r Zmetadata� assert_equalZts)r8 Z isoformatZexpectedr@ �keyZmanifest�ar r r �_test_timestamp_parsingm s
z0ArchiveTimestampTestCase._test_timestamp_parsingc C s$ | j dtdddddddtj�� d S )Nz1970-01-01T00:00:01.000001i� r r )rJ r r �utc)r8 r r r �test_with_microsecondsu s z/ArchiveTimestampTestCase.test_with_microsecondsc C s$ | j dtdddddddtj�� d S )Nz1970-01-01T00:00:01i� r r )rJ r r rK )r8 r r r �test_without_microsecondsz s z2ArchiveTimestampTestCase.test_without_microsecondsN)r; r< r= rJ rL rM r r r r rE k s rE c @ s e Zd Zdd� Zdd� ZdS )�ChunkBufferTestCasec C s� t dd�t dd�g}t� }td �}t||d �}x |D ]}|j|� |j� q4W |jdd� | jt|j�d� t j
� }x |jD ]}|j |j | � q|W | j|dd� |D �� d S ) NZp1)r% Zp2T)�flushr c S s g | ]}t |d ��qS ))�
internal_dict)r )�.0�dr r r �
<listcomp>� s z,ChunkBufferTestCase.test.<locals>.<listcomp>)
r r7 r
r �addrO rG r+ �chunksr �Unpacker�feedr? )r8 �data�cacherH rU rR �unpackerrB r r r rF � s
zChunkBufferTestCase.testc C s� dd }t d|d�t d|d�g}t� }td �}t||d �}x|D ]}|j|� q@W |jdd� | jt|j�d� |jd d� | jt|j�d
� t j
� }x |jD ]}|j |j | � q�W | j|d d � |D �� d S )
NZ$0123456789abcdefghijklmnopqrstuvwxyzi�a Zfull)r% �source�partialF)rO r# T� c S s g | ]}t |d ��qS ))rP )r )rQ rR r r r rS � s z4ChunkBufferTestCase.test_partial.<locals>.<listcomp>)
r r7 r
r rT rO rG r+ rU r rV rW r? ) r8 ZbigrX rY rH rU rR rZ rB r r r � test_partial� s
z ChunkBufferTestCase.test_partialN)r; r< r= rF r^ r r r r rN � s rN c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RobustUnpackerTestCasec C s dj dd� |D ��S )N� c s s | ]}t jd |i�V qdS )r% N)r �packb)rQ r& r r r � <genexpr>� s z5RobustUnpackerTestCase.make_chunks.<locals>.<genexpr>)�join)r8 �itemsr r r � make_chunks� s z"RobustUnpackerTestCase.make_chunksc C s t |t�o|jd�dkS )Ns path� foo� bar� boo� baz)rf rg rh ri )�
isinstance�dict�get)r8 �valuer r r �
_validator� s z!RobustUnpackerTestCase._validatorc C sd t | jtd�}g }xL|D ]D\}}|r,|j� x.|D ]&}|j|� x|D ]}|j|� qFW q2W qW |S )N)Z validatorZ item_keys)r
rn r ZresyncrW �append)r8 �inputrZ r5 Z should_syncrU rX r& r r r �process� s
z RobustUnpackerTestCase.processc C sn d| j ddg�gfddg| j ddg�g fg}| j|�}| j|ddiddidd d
d d dd ddiddig � d S )
NFs foos bars garbages boos bazs path�g �a �r �b �e )re rq rG )r8 rU r5 r r r �test_extra_garbage_no_sync� s
z1RobustUnpackerTestCase.test_extra_garbage_no_syncc C s0 g }x&|r*|j |d |� � ||d � }qW |S )N)ro )r8 �leftZlength�partsr r r �split� s
z RobustUnpackerTestCase.splitc C sV | j | jddddg�d�}d|fg}| j|�}| j|ddiddiddiddig� d S )Ns foos bars boos bazr Fs path)rz re rq rG )r8 rU rp r5 r r r �test_correct_stream� s
z*RobustUnpackerTestCase.test_correct_streamc C sf | j | jddddg�d�}d|d d� fd|dd � fg}| j|�}| j|d did did dig� d S )
Ns foos bars boos bazr] Fr# Ts path)rz re rq rG )r8 rU rp r5 r r r �test_missing_chunk� s
z)RobustUnpackerTestCase.test_missing_chunkc C sn | j | jddddg�d�}d|d d� fdd d
g|dd � fg}| j|�}| j|d did did dig� d S ) Ns foos bars boos bazr] Fr# Ts gars bages path)rz re rq rG )r8 rU rp r5 r r r �test_corrupt_chunk� s (
z)RobustUnpackerTestCase.test_corrupt_chunkN) r; r< r= re rn rq rw rz r{ r| r} r r r r r_ � s
r_ c C s dd� t D �S )Nc S s g | ]}t j|��qS r )r ra )rQ �namer r r rS � s z(item_keys_serialized.<locals>.<listcomp>)r r r r r �item_keys_serialized� s r �packedr` � xs foobarc C s g | ]}t j|��qS r )r ra )rQ �or r r rS � s rS g F� �* g�Q� �k7@T� foo� barc C s d S )Nr )r� r r r r �test_invalid_msgpacked_item� s r� c C s g | ]}t j|��qS r )r ra )rQ r� r r r rS � s s paths /a/b/cc c s | ]}|d fV qdS )r` Nr )rQ �kr r r rb � s rb c c s | ]}|d d fV qdS )r� i� Nr )rQ r� r r r rb � s c C s d S )Nr )r� r r r r �test_valid_msgpacked_items� s r� c C s dd } | di}t j| �g}d S )Nr� � r` )r ra )rH rX r r r r � test_key_length_msgpacked_items� s r� c C s2 t jt�� t� td��W d Q R X W d Q R X d S )N�{ )�pytest�raisesr r �OSErrorr r r r �test_backup_io s r� c C s\ G dd� d�} | t �}tjt�� xt|�D ]}q,W W d Q R X | t�}xt|�D ]}qPW d S )Nc @ s e Zd Zdd� Zdd� ZdS )z%test_backup_io_iter.<locals>.Iteratorc S s
|| _ d S )N)�exc)r8 r� r r r rA s z.test_backup_io_iter.<locals>.Iterator.__init__c S s | j � �d S )N)r� )r8 r r r �__next__ s z.test_backup_io_iter.<locals>.Iterator.__next__N)r; r< r= rA 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� )r )r� r� )3r2 � collectionsr r r �ior Z
unittest.mockr r� r� r Z
crypto.keyr
�archiver r r
r r r r r r Zhelpersr r r& r r Zfixturer r r/ r1 r6 r7 rE rN r_ r ZmarkZ parametrizer� �sorted�listZIKr� r� r� r� r r r r �<module> sH
'8, "