File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/cache.cpython-36.opt-1.pyc
3
y��b�# � @ s� d dl Z d dlZd dlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
dd lmZ dd
lmZmZ dd lmZ dd lmZ dd
lmZ ejZG dd� d�ZG dd� d�ZdS )� N� )�H)�TestKey� )�
Statistics)�
AdHocCache)�CompressionSpec)�RepoKey)�
ChunkIndex�CacheSynchronizer)�Manifest)�msgpack)�
Repositoryc
@ s2 e Zd Zejdd� �Zejdd� �Zdd� Zdd� Zd d
� Z ej
j d d d
idfe d�ddgfd2d3d4d5d6f�ej
j dd d � d d � d d � f�d d!� ��Z
ej
j d"d#e d$�d
d%d&fgid'fd#e d$�d
fgid'fd#d7gidfd#d e d$�d
fgidfd#e d$�dd
fgidff�d(d)� �Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdS )8�TestCacheSynchronizerc C s t � S )N)r
)�self� r � /usr/lib64/python3.6/cache.py�index s zTestCacheSynchronizer.indexc C s t |�S )N)r )r r r r r �sync s zTestCacheSynchronizer.syncc C s" t ddddg d��}|j|� d S )N�bari� i. �chunks)�foo�bazr �userr )�packb�feed)r r r �datar r r �test_no_chunks s
z$TestCacheSynchronizer.test_no_chunksc C s: t ddddtd�ddftd�ddfgd��}|j|� d S ) Nr i� i. r r r � )r r r r r )r r r )r r r r r r r � test_simple( s
z!TestCacheSynchronizer.test_simplec C s� t ddddtd�ddftd�ddfgd��}|t dd d
�dgd ��7 }|t dd d
�td�ddftd�ddfgdgd ��7 }|t dtd�ddfgi�7 }|t dtd�ddfgi�7 }|d d
� }|d
d� }|dd � }|j|� |j|� |j|� d S )Nr i� i. r r r r )r r r r r Z123456)z security.foor )�xattrs�stuff)r r r! �F �x )r r r )r r r )r r r )r r r r Zpart1Zpart2Zpart3r r r �
test_multiple8 s<
z#TestCacheSynchronizer.test_multiplez
elem,errorr r zUnexpected object: map�� z$Unexpected bytes in chunks structurezIncorrect key length�Unexpected object: integer� �?�Unexpected object: doubleT�Unexpected object: trueF�Unexpected object: falseN�Unexpected object: nil� structurec C s d| iS )Nr r )�elemr r r �<lambda>{ s z TestCacheSynchronizer.<lambda>c C s
d| giS )Nr r )r- r r r r. | s c C s d| ddfgiS )Nr r r r )r- r r r r. } s c
C sN t ||��}tjt��}|j|� W d Q R X t|t�r<|g}dd� |D �}d S )Nc S s g | ] }d | �qS )zcache_sync_feed failed: r )�.0�errorr r r �
<listcomp>� s z8TestCacheSynchronizer.test_corrupted.<locals>.<listcomp>)r �pytest�raises�
ValueErrorr �
isinstance�str)r r r, r- r0 �packed�excinfoZpossible_errorsr r r �test_corruptedo s
z$TestCacheSynchronizer.test_corruptedz
data,errorr � r � z Invalid chunk list entry lengthc C s, t |�}tjt��}|j|� W d Q R X d S )N)r r2 r3 r4 r )r r r r r0 r7 r8 r r r �test_corrupted_ancillary� s
z.TestCacheSynchronizer.test_corrupted_ancillaryc C s� t j� }|jd� |jdjdd�� |jdjdd�� |jdjdd�� |jd
jdd�� |jtd�� |j|jdd�� |jdjdd�� |jd jdd�� |jd� tj|�}|S ) Ns BORG_IDXr r; �littler: r r i� i. � )�io�BytesIO�write�to_bytesr �seekr
�read)r ZrefcountZ
index_datar r r r �make_index_with_refcount� s
z.TestCacheSynchronizer.make_index_with_refcountc C sT | j tjd �}t|�}tdtd�ddfgi�}tjt��}|j |� W d Q R X d S )Nr r r r )
rE r
� MAX_VALUEr r r r2 r3 r4 r )r r r r r8 r r r �test_corrupted_refcount� s z-TestCacheSynchronizer.test_corrupted_refcountc C s: | j tj�}t|�}tdtd�ddfgi�}|j|� d S )Nr r r r )rE r
rF r r r r )r r r r r r r �test_refcount_max_value� s
z-TestCacheSynchronizer.test_refcount_max_valuec C sH | j tjd �}t|�}tdtd�ddfgi�}|j|� |j|� d S )Nr r r r )rE r
rF r r r r )r r r r r r r �!test_refcount_one_below_max_value� s
z7TestCacheSynchronizer.test_refcount_one_below_max_value)r r& )r' r( )Tr) )Fr* )Nr+ )r r r )�__name__�
__module__� __qualname__r2 �fixturer r r r r$ ZmarkZ parametrize�bytesr9 r< rE rG rH rI r r r r r s: 7
$ r c @ s� e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS ) �TestAdHocCachec
c sX t jjt|�d�| _t| jddd��*}|jtd�d� |jtj d� |V W d Q R X d S )N�
repositoryT)Z exclusive�creater s 1234s 5678)
�os�path�joinr6 Zrepository_locationr Zputr r Z MANIFEST_ID)r ZtmpdirrP r r r rP � s
zTestAdHocCache.repositoryc C s, |j dd� tj|tj� �}td�j|_|S )NZBORG_PASSPHRASEZtestZnone)Zsetenvr rQ r ZMockArgsr Z
compressor)r rP Z monkeypatch�keyr r r rU � s zTestAdHocCache.keyc C s$ t ||�j� t j||t jd�d S )N)rU Z
operationsr )r rA �loadZNO_OPERATION_CHECK)r rP rU r r r �manifest� s zTestAdHocCache.manifestc C s t |||�S )N)r )r rP rU rW r r r �cache� s zTestAdHocCache.cachec C s d S )Nr )r rX r r r � test_does_not_contain_manifest� s z-TestAdHocCache.test_does_not_contain_manifestc C s |j td�t� � d S )Nr )� chunk_decrefr r )r rP rX r r r �$test_does_not_delete_existing_chunks� s z3TestAdHocCache.test_does_not_delete_existing_chunksc
C s2 t jt�� |jtd�dt� dd� W d Q R X d S )Nr s 5678T)Z overwrite)r2 r3 �AssertionError� add_chunkr r )r rX r r r �test_does_not_overwrite� s z&TestAdHocCache.test_does_not_overwritec C s d S )Nr )r rX r r r � test_seen_chunk_add_chunk_size� s z-TestAdHocCache.test_seen_chunk_add_chunk_sizec
C sP |j td�dt� � |jtd�t� � tjtj�� |jtd�� W dQ R X dS )zE.g. checkpoint archives� s 1010N) r] r r rZ r2 r3 r ZObjectNotFound�get)r rX rP r r r �#test_deletes_chunks_during_lifetime� s z2TestAdHocCache.test_deletes_chunks_during_lifetimec C s d S )Nr )r rX r r r �test_files_cache s z TestAdHocCache.test_files_cachec C s |j td�� |j� d S )Nr` )Z
seen_chunkr Zrollback)r rX r r r �test_txn s zTestAdHocCache.test_txnc C s d S )Nr )r rX r r r �test_incref_after_add_chunk s z*TestAdHocCache.test_incref_after_add_chunkc C s dS )z9This case occurs with part files, see Archive.chunk_file.Nr )r rX r r r �$test_existing_incref_after_add_chunk s z3TestAdHocCache.test_existing_incref_after_add_chunkN)rJ rK rL r2 rM rP rU rW rX rY r[ r^ r_ rb rc rd re rf r r r r rO � s rO )r? Zos.pathrR r2 Z hashindexr rU r �archiver rX r �compressr Z
crypto.keyr r
r Zhelpersr r
rP r r r rO r r r r �<module> s ?