File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/remote.cpython-36.opt-1.pyc
3
y��b � @ 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.6/remote.py� expect_write s z)TestSleepingBandwidthLimiter.expect_writec C s t |�S )N)�len)r r r r r r � check_write s z(TestSleepingBandwidthLimiter.check_writec C s4 |j td| j� td�}| jdd� |jdd� d S )N�writer � s test)�setattr�osr r r r )r � monkeypatch�itr r r �test_write_unlimited s z1TestSleepingBandwidthLimiter.test_write_unlimitedc s� |j td| j� |j td� fdd�� |j tddd� � d� td�}| jdd � |jdd � | jdd
� |jdd � | jdd
� |jdd
� � d 7 � | jdd
� |jdd
� � d 7 � | jdd
� |jdd
� � d 7 � | jdd
� |jdd
� d S )Nr Z monotonicc s � S )Nr r )�nowr r �<lambda>% s z9TestSleepingBandwidthLimiter.test_write.<locals>.<lambda>Zsleepc S s d S )Nr )�xr r r r & s �d r s tests 123456s
1234567890�
� 1)r r r �timer r r )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 sn t jjt|�d�| _t| jddd��@}|jtd�d� |jtd�d� |jtd�td �� |V W d Q R X d S )
N�
repositoryT)Z exclusive�creater s 1234r s 5678� r! ) r �path�join�strZrepository_locationr �putr �bytes)r Ztmpdirr* r r r r* I s z TestRepositoryCache.repositoryc C s t |�S )N)r )r r* r r r �cacheR s zTestRepositoryCache.cache)r2 c C s d S )Nr )r r2 r r r � test_simpleV s z TestRepositoryCache.test_simplec sF � fdd�}� j td�td�td�g�}� j}|� _� j� |� _d S )Nc s
d� _ d S )Nr )Z
size_limitr )r2 r r �query_size_limiti s z:TestRepositoryCache.test_backoff.<locals>.query_size_limitr r, r )�get_manyr r4 Zbackoff)r r2 r4 �iteratorZqslr )r2 r � test_backoffh s
z TestRepositoryCache.test_backoffc C sD G dd� d�}|j td�td�td�g�}td|�� W d Q R X 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 )Nr )r �argsr r r �__init__� s z=TestRepositoryCache.test_enospc.<locals>.enospc_open.__init__c S s | S )Nr )r r r r � __enter__� s z>TestRepositoryCache.test_enospc.<locals>.enospc_open.__enter__c S s d S )Nr )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�errnoZENOSPC)r r r r r r � s z:TestRepositoryCache.test_enospc.<locals>.enospc_open.writeNc S s d S )Nr )r �nr r r �truncate� s z=TestRepositoryCache.test_enospc.<locals>.enospc_open.truncate)N)r& r' r( r9 r: r< r r@ r r r r � enospc_open� s
rA r r r, z
builtins.open)r5 r r )r r2 rA r6 r r r � test_enospc� s
z TestRepositoryCache.test_enospcc C s, |j dd� tj|tj� �}td�j|_|S )NZBORG_PASSPHRASEZtestZnone)Zsetenvr r+ r
ZMockArgsr Z
compressor)r r* r �keyr r r rC � s zTestRepositoryCache.keyc C s |j |�}|j||j|�� |S )N)Zid_hashr0 Zencrypt)r rC r* r Zid_r r r �_put_encrypted_object� s
z)TestRepositoryCache._put_encrypted_objectc C s | j ||d�S )Ns 1234)rD )r rC r* r r r �H1� s zTestRepositoryCache.H1c C s | j ||d�S )Ns 5678)rD )r rC r* r r r �H2� s zTestRepositoryCache.H2c C s | j ||td��S )Nr! )rD r1 )r rC r* r r r �H3� s zTestRepositoryCache.H3c C s t ||dd�S )NT)�decrypted_cacheZ force_cache)r )r rC r* r r r rH � s z#TestRepositoryCache.decrypted_cache)rH c C s� t |j|||g�� |j|||g�}t|j|�d��P}|jdtj� tj|j � d�dA j
dd�}|jdtj� |j |� |j � W d Q R X t
jt�� W d Q R X d S )Nza+br �littler ���rJ )�listr5 �openZ key_filename�seek�io�SEEK_END�int�
from_bytes�read�to_bytesr r@ �pytestZraisesr
)r rH rE rF rG r6 r Z corruptedr r r �test_cache_corruption� s
z)TestRepositoryCache.test_cache_corruptionN)r& r' r( rT Zfixturer* r2 r r3 r7 rB rC rD rE rF rG rH rU r r r r r) H s r) )r> r rN r$ Z
unittest.mockr rT Zremoter r r r* r Z
crypto.keyr �compressr Zhelpersr
Z hashindexr rC r
r r) r r r r �<module> s 6