File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/remote.cpython-39.opt-1.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)Z expected_fdZ
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 t |�S 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 Ztmpdirr1 r r r r1 I s z TestRepositoryCache.repositoryc C s t |�S r )r )r r1 r r r �cacheR s zTestRepositoryCache.cache�r; c C s d S r r )r r; r r r � test_simpleV s z TestRepositoryCache.test_simplec sF � fdd�}� � td�td�td�g�}� j}|� _� �� |� _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 r5 r )�get_manyr r> Zbackoff)r r; r> �iteratorZqslr r< r � test_backoffh s
z TestRepositoryCache.test_backoffc C sX G dd� d�}|� td�td�td�g�}td|�� W d � n1 sJ0 Y 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- r. r/ rC rD rF r rK r r r r � enospc_open� s
rL r r r5 z
builtins.open)r? r r )r r; rL r@ r r r � test_enospc� s
z TestRepositoryCache.test_enospcc C s, |� dd� t�|t�� �}td�j|_|S )NZBORG_PASSPHRASE�testZnone)Zsetenvr r2 r
ZMockArgsr Z
compressor)r r1 r �keyr r r rO � s zTestRepositoryCache.keyc C s |� |�}|�||�|�� |S r )Zid_hashr9 Zencrypt)r rO r1 r Zid_r r r �_put_encrypted_object� s
z)TestRepositoryCache._put_encrypted_objectc C s | � ||d�S )Nr3 �rP �r rO r1 r r r �H1� s zTestRepositoryCache.H1c C s | � ||d�S )Nr4 rQ rR r r r �H2� s zTestRepositoryCache.H2c C s | � ||td��S )Nr( )rP r: rR r r r �H3� s zTestRepositoryCache.H3c C s t ||dd�S )NT)�decrypted_cacheZ force_cache)r rR r r r rV � s z#TestRepositoryCache.decrypted_cache)rV c C s� t |�|||g�� |�|||g�}t|�|�d��Z}|�dtj� t�|� � d�dA �
dd�}|�dtj� |� |� |� � W d � n1 s�0 Y t
�t�� W d � n1 s�0 Y d S )Nza+b����littler r )�listr? �openZ key_filename�seek�io�SEEK_END�int�
from_bytes�read�to_bytesr rK �pytestZraisesr
)r rV rS rT rU r@ r Z corruptedr r r �test_cache_corruption� s
& z)TestRepositoryCache.test_cache_corruptionN)r- r. r/ rb Zfixturer1 r; r r= rA rM rO rP rS rT rU rV rc r r r r r0 H s&
r0 )rH r r\ r+ Z
unittest.mockr rb Zremoter r r r1 r Z
crypto.keyr �compressr Zhelpersr
Z hashindexr rO r
r r0 r r r r �<module> s 6