File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/cache.cpython-39.pyc
a
HZ�h.$ � @ s� d dl Z d dlZddlmZ d dlZddlmZ ddlm Z ddl
m Z ddl m
Z
dd lmZ dd
lmZ dd lmZmZ dd lmZ dd
lmZ G dd� d�ZG dd� d�ZdS )� N� )�packb� )�H)�TestKey)�
Statistics��
AdHocCache)�CompressionSpec)�RepoKey)�
ChunkIndex�CacheSynchronizer)�Manifest)�
Repositoryc @ s: e Zd Zejdd� �Zejdd� �Zdd� Zdd� Zd d
� Z ej
j d d d
idfe d�ddgfdddddfg d�d�ej
� ddd� d d� d d� f�d d � ��Z
ej
� d d!e d"�d
d#d$fgid%fd!e d"�d
fgid%fd!d&gid'fd!d e d"�d
fgid'fd!e d"�d(d
fgid)ff�d*d+� �Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4S )5�TestCacheSynchronizerc C s t � S �N)r )�self� r �:/usr/lib64/python3.9/site-packages/borg/testsuite/cache.py�index s zTestCacheSynchronizer.indexc C s t |�S r )r
)r r r r r �sync s zTestCacheSynchronizer.syncc C s. t ddddg d��}|�|� t|�r*J �d S )N�bar�� �. �chunks�ZfooZbazr �userr )r �feed�len�r r r �datar r r �test_no_chunks s �
z$TestCacheSynchronizer.test_no_chunksc C sr t ddddtd�ddftd�ddfgd��}|�|� t|�dksFJ �|td� d ksZJ �|td� d
ksnJ �d S ) Nr r r 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 sL t ddddtd�ddftd�ddfgd��}|t dd d
�d gd ��7 }|t dd d
�td�ddftd�ddfgd gd
��7 }|t dtd�ddfgi�7 }|t dtd�ddfgi�7 }|d d� }|dd� }|dd � }|�|� |�|� |�|� t|�dk�sJ �|td� dk�s J �|td� dk�s2J �|td� dk�sHJ �d S )Nr r r r r r r" r Z123456)z security.foor r$ )�xattrs�stuff)r' r r( �F �x )r" r r )r r r" r# r% )r r r r Zpart1Zpart2Zpart3r r r �
test_multiple7 s\ ��
���
� ���
�� ��
z#TestCacheSynchronizer.test_multiplez
elem,errorr r zUnexpected object: map�� z$Unexpected bytes in chunks structurezIncorrect key length)r �Unexpected object: integer)� �?�Unexpected object: double)TzUnexpected object: true)FzUnexpected object: false)NzUnexpected object: nil)�map�bytes�intZdouble�trueZfalse�none)Zids� structurec C s d| iS �Nr r ��elemr r r �<lambda>z � z TestCacheSynchronizer.<lambda>c C s
d| giS r6 r r7 r r r r9 { r: c C s d| ddfgiS )Nr r r r r7 r r r r9 | r: c C st t ||��}t�t��}|�|� W d � n1 s60 Y t|t�rP|g}dd� |D �}t|j�|v spJ �d S )Nc S s g | ] }d | �qS )�cache_sync_feed failed: r )�.0�errorr r r �
<listcomp>� r: z8TestCacheSynchronizer.test_corrupted.<locals>.<listcomp>)r �pytest�raises�
ValueErrorr �
isinstance�str�value)r r r5 r8 r= �packed�excinfoZpossible_errorsr r r �test_corruptedn s (
z$TestCacheSynchronizer.test_corruptedz
data,errorr � r" � z Invalid chunk list entry lengthr$ r- r. r/ c C sV t |�}t�t��}|�|� W d � n1 s20 Y t|j�d| ksRJ �d S )Nr; )r r? r@ rA r rC rD )r r r r r= rE rF r r r �test_corrupted_ancillary� s
(z.TestCacheSynchronizer.test_corrupted_ancillaryc C s� t �� }|�d� |�d�dd�� |�d�dd�� |�d�dd�� |�d�dd�� |�td�� |�|�dd�� |�d�dd�� |�d �dd�� |�d� t�|�}|S )
Ns BORG_IDXr rI �littlerH � r r r )�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 sz | � tjd �}t|�}tdtd�ddfgi�}t�t��}|� |� W d � n1 sZ0 Y t
|j �dksvJ �d S )Nr r r r z/cache_sync_feed failed: invalid reference count) rS r � MAX_VALUEr
r r r? r@ rA r rC rD )r r r r rF r r r �test_corrupted_refcount� s �� (z-TestCacheSynchronizer.test_corrupted_refcountc C sV | � tj�}t|�}tdtd�ddfgi�}|�|� |td� tjddfksRJ �d S )Nr r r r r r �rS r rT r
r r r r r r r �test_refcount_max_value� s ��
z-TestCacheSynchronizer.test_refcount_max_valuec C s� | � tjd �}t|�}tdtd�ddfgi�}|�|� |td� tjddfksVJ �|�|� |td� tjddfks|J �d S )Nr r r r r r rV r r r r �!test_refcount_one_below_max_value� s ��
z7TestCacheSynchronizer.test_refcount_one_below_max_valueN)�__name__�
__module__� __qualname__r? �fixturer r r! r& r+ ZmarkZ parametrizer1 rG rJ rS rU rW rX r r r r r sN
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 sl t j�t|�d�| _t| jddd��4}|�td�d� |�tj d� |V W d � n1 s^0 Y d S )N�
repositoryT)Z exclusive�creater � 1234� 5678)
�os�path�joinrC Zrepository_locationr Zputr r � MANIFEST_ID)r Ztmpdirr^ r r r r^ � s
zTestAdHocCache.repositoryc C s, |� dd� t�|t�� �}td�j|_|S )NZBORG_PASSPHRASE�testr4 )Zsetenvr r_ r ZMockArgsr
Z
compressor)r r^ Z monkeypatch�keyr r r rg � s zTestAdHocCache.keyc C s$ t ||��� t j||t jd�d S )N)rg Z
operationsr )r rO �loadZNO_OPERATION_CHECK)r r^ rg r r r �manifest� s zTestAdHocCache.manifestc C s t |||�S r r )r r^ rg ri r r r �cache� s zTestAdHocCache.cachec C s |� tj�rJ �d S r )�
seen_chunkr re �r rj r r r � test_does_not_contain_manifest� s z-TestAdHocCache.test_does_not_contain_manifestc C sD |� td��tjksJ �|�td�t� � |�td��dks@J �d S )Nr r` )rk r r rT � chunk_decrefr �get)r r^ rj r r r �$test_does_not_delete_existing_chunks� s z3TestAdHocCache.test_does_not_delete_existing_chunksc C sF t �t��( |jtd�dt� dd� W d � n1 s80 Y d S )Nr ra T)Z overwrite)r? r@ �AssertionError� add_chunkr r rl r r r �test_does_not_overwrite� s z&TestAdHocCache.test_does_not_overwritec C s* |� td�dt� �td�ddfks&J �d S )Nr ra rI r )rr r r rl r r r � test_seen_chunk_add_chunk_size� s z-TestAdHocCache.test_seen_chunk_add_chunk_sizec C s� |� td�dt� � |�td��dks*J �|�td�t� � |�td��rNJ �t�tj�� |� td�� W d � n1 s~0 Y dS )zE.g. checkpoint archives� s 1010r N)
rr r r rk rn r? r@ r ZObjectNotFoundro )r rj r^ r r r �#test_deletes_chunks_during_lifetime� s z2TestAdHocCache.test_deletes_chunks_during_lifetimec C s: |� dtd�d �dksJ �|jdks(J �|jd u s6J �d S )Ns foorH )FN�d)Zfile_known_and_unchangedr1 Z
cache_mode�filesrl r r r �test_files_cache s z TestAdHocCache.test_files_cachec C sP |j r
J �|�td�� |j s"J �|js,J �|�� |j r>J �t|d�rLJ �d S )Nru r )Z _txn_activerk r r Zrollback�hasattrrl r r r �test_txn s
zTestAdHocCache.test_txnc C sN |� td�dt� �td�ddfks&J �|�td�t� �td�ddfksJJ �d S )Nr" ra rI �/ �rr r r Z chunk_increfrl r r r �test_incref_after_add_chunk s &z*TestAdHocCache.test_incref_after_add_chunkc C sN |� td�dt� �td�ddfks&J �|�td�t� �td�ddfksJJ �dS )z9This case occurs with part files, see Archive.chunk_file.r ra rI r Nr} rl r r r �$test_existing_incref_after_add_chunk s &z3TestAdHocCache.test_existing_incref_after_add_chunkN)rY rZ r[ r? r\ r^ rg ri rj rm rp rs rt rv ry r{ r~ r r r r r r] � s"
r] )rM Zos.pathrb Zhelpers.msgpackr r? Z hashindexr rg r �archiver rj r �compressr
Z
crypto.keyr r r
Zhelpersr r^ r r r] r r r r �<module> s ?