File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/remote.cpython-39.pyc
a
HZ�h � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ d d
lmZ d d lmZ G d d
� d
�ZG dd� d�ZdS )� N)�patch� )�SleepingBandwidthLimiter�RepositoryCache�cache_if_remote)�
Repository)� PlaintextKey)�CompressionSpec)�IntegrityError� )�H)�TestKeyc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestSleepingBandwidthLimiterc C s || _ || _d S �N)� expected_fd�
expected_data��self�fd�data� r �;/usr/lib64/python3.9/site-packages/borg/testsuite/remote.py� expect_write s z)TestSleepingBandwidthLimiter.expect_writec C s$ || j ksJ �|| jks J �t|�S r )r r �lenr r r r � check_write s z(TestSleepingBandwidthLimiter.check_writec C s4 |� td| j� td�}| �dd� |�dd� d S )N�writer � � test)�setattr�osr r r r �r � monkeypatch�itr r r �test_write_unlimited s z1TestSleepingBandwidthLimiter.test_write_unlimitedc s� |� td| j� |� td� fdd�� |� tddd� � d� td�}| �dd � |�dd � | �dd
� |�dd � | �dd
� |�dd
� � d 7 � | �dd
� |�dd
� � d 7 � | �dd
� |�dd
� � d 7 � | �dd
� |�dd
� d S )Nr � monotonicc s � S r r r �Znowr r �<lambda>% � z9TestSleepingBandwidthLimiter.test_write.<locals>.<lambda>�sleepc S s d S r r )�xr r r r& & r' �d r r s 123456s
1234567890�
� 1)r r r �timer r r r r r% r �
test_write# s( z'TestSleepingBandwidthLimiter.test_writeN)�__name__�
__module__� __qualname__r r r# r. r r r r r s r c @ s� e Zd Zejdd� �Zejdd� �Zed�dd�Zed�dd �Z ed�d
d �Z
ejd d
� �Z dd� Z ejdd� �Z
ejdd� �Zejdd� �Zejdd� �Zed�dd�ZdS ) �TestRepositoryCachec c s� t j�t|�d�| _t| jddd��J}|�td�d� |�td�d� |�td�td �� |V W d � n1 st0 Y d S )
N�
repositoryT)Z exclusive�creater � 1234r � 5678� r* ) r �path�join�strZrepository_locationr �putr �bytes)r Ztmpdirr3 r r r r3 I s z TestRepositoryCache.repositoryc C s t |�S r )r )r r3 r r r �cacheR s zTestRepositoryCache.cache�r= c C s� |� td��dksJ �|jdks$J �|jdks2J �t|�td�g��dgksPJ �|jdks^J �|jdkslJ �t|�td�g��dgks�J �|jdks�J �|jdks�J �|� td��dks�J �|jdks�J �|jdks�J �d S )Nr r5 r r )�getr �misses�hits�list�get_many)r r= r r r � test_simpleV s z TestRepositoryCache.test_simplec s � fdd�}t � �td�td�g��ddgks2J �� jdks@J �� jdksNJ �� �td�td�td�g�}t|�dkszJ �� j}|� _� �� |� _� jdks�J �td�� jvs�J �td�� jvs�J �t|�t d �ks�J �� j
dks�J �t|�dks�J �� j
dk�sJ �d S )
Nc s
d� _ d S )Nr )Z
size_limitr r> r r �query_size_limiti s z:TestRepositoryCache.test_backoff.<locals>.query_size_limitr r r5 r6 r r7 r* ) rB rC r r@ � evictions�nextrE Zbackoffr= r< Z slow_misses)r r= rE �iteratorZqslr r> r � test_backoffh s" & z TestRepositoryCache.test_backoffc C s� G dd� d�}|� td�td�td�g�}t|�dks:J �td|��< t|�dksVJ �|jdksdJ �|jd ksrJ �W d � n1 s�0 Y t|�td
�ks�J �d S ) Nc @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd S )
z4TestRepositoryCache.test_enospc.<locals>.enospc_openc W s d S r r )r �argsr r r �__init__� s z=TestRepositoryCache.test_enospc.<locals>.enospc_open.__init__c S s | S r r )r r r r � __enter__� s z>TestRepositoryCache.test_enospc.<locals>.enospc_open.__enter__c S s d S r r )r �exc_typeZexc_valZexc_tbr r r �__exit__� s z=TestRepositoryCache.test_enospc.<locals>.enospc_open.__exit__c S s t tjd��d S )NZfoo)�OSError�errno�ENOSPC)r r r r r r � s z:TestRepositoryCache.test_enospc.<locals>.enospc_open.writeNc S s d S r r )r �nr r r �truncate� s z=TestRepositoryCache.test_enospc.<locals>.enospc_open.truncate)N)r/ r0 r1 rK rL rN r rS r r r r � enospc_open� s
rT r r r7 r5 z
builtins.openr6 r r* )rC r rG r ZenospcrF r< )r r= rT rH r r r � test_enospc� s ,z TestRepositoryCache.test_enospcc C s, |� dd� t�|t�� �}td�j|_|S )NZBORG_PASSPHRASE�testZnone)Zsetenvr r4 r
ZMockArgsr Z
compressor)r r3 r! �keyr r r rW � s zTestRepositoryCache.keyc C s |� |�}|�||�|�� |S r )Zid_hashr; Zencrypt)r rW r3 r Zid_r r r �_put_encrypted_object� s
z)TestRepositoryCache._put_encrypted_objectc C s | � ||d�S )Nr5 �rX �r rW r3 r r r �H1� s zTestRepositoryCache.H1c C s | � ||d�S )Nr6 rY rZ r r r �H2� s zTestRepositoryCache.H2c C s | � ||td��S )Nr* )rX r< rZ r r r �H3� s zTestRepositoryCache.H3c C s t ||dd�S )NT)�decrypted_cacheZ force_cache)r rZ r r r r^ � s z#TestRepositoryCache.decrypted_cache)r^ c C s� t |�|||g�� |�|||g�}t|�dks4J �t|�|�d��Z}|�dtj� t� |�
� d�dA � dd�}|�dtj� |� |� |�
� W d � n1 s�0 Y t�t�� t|�dks�J �W d � n1 s�0 Y d S )N)� r5 za+b����littler r )r_ r6 )rB rC rG �openZ key_filename�seek�io�SEEK_END�int�
from_bytes�read�to_bytesr rS �pytestZraisesr
)r r^ r[ r\ r] rH r Z corruptedr r r �test_cache_corruption� s
& z)TestRepositoryCache.test_cache_corruptionN)r/ r0 r1 rj Zfixturer3 r= r rD rI rU rW rX r[ r\ r] r^ rk r r r r r2 H s&
r2 )rP r rd r- Z
unittest.mockr rj Zremoter r r r3 r Z
crypto.keyr �compressr Zhelpersr
Z hashindexr rW r
r r2 r r r r �<module> s 6