| Current File : //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/remote.cpython-36.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)� expected_fd�
expected_data)�self�fd�data� r � /usr/lib64/python3.6/remote.py� expect_write s z)TestSleepingBandwidthLimiter.expect_writec C s$ || j kst�|| jks t�t|�S )N)r �AssertionErrorr �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)r5 c C s� |j td��dkst�|jdks$t�|jdks2t�t|jtd�g��dgksPt�|jdks^t�|jdkslt�t|jtd�g��dgks�t�|jdks�t�|jdks�t�|j td��dks�t�|jdks�t�|jdks�t�d S )Nr s 1234r r )�getr r �misses�hits�list�get_many)r r5 r r r � test_simpleV s z TestRepositoryCache.test_simplec s � fdd�}t � jtd�td�g��ddgks2t�� jdks@t�� jdksNt�� jtd�td�td�g�}t|�dkszt�� j}|� _� j� |� _� jdks�t�td�� j ks�t�td�� j ks�t�t|�t
d �ks�t�� j dks�t�t|�dks�t�� j dk�st�d S )
Nc s
d� _ d S )Nr )Z
size_limitr )r5 r r �query_size_limiti s z:TestRepositoryCache.test_backoff.<locals>.query_size_limitr r s 1234s 5678r r/ r$ ) r9 r: r r r7 � evictions�nextr< Zbackoffr5 r4 Z slow_misses)r r5 r< �iteratorZqslr )r5 r � test_backoffh s" & z TestRepositoryCache.test_backoffc
C s� G dd� d�}|j td�td�td�g�}t|�dks:t�td|��2 t|�dksVt�|jdksdt�|jd ksrt�W d Q R X t|�td
�ks�t�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+ rB rC rE r rI r r r r � enospc_open� s
rJ r r r/ s 1234z
builtins.opens 5678r r$ )r: r r> r r Zenospcr= r4 )r r5 rJ r? 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 rL � s zTestRepositoryCache.keyc C s |j |�}|j||j|�� |S )N)Zid_hashr3 Zencrypt)r rL r- r Zid_r r r �_put_encrypted_object� s
z)TestRepositoryCache._put_encrypted_objectc C s | j ||d�S )Ns 1234)rM )r rL r- r r r �H1� s zTestRepositoryCache.H1c C s | j ||d�S )Ns 5678)rM )r rL r- r r r �H2� s zTestRepositoryCache.H2c C s | j ||td��S )Nr$ )rM r4 )r rL r- r r r �H3� s zTestRepositoryCache.H3c C s t ||dd�S )NT)�decrypted_cacheZ force_cache)r )r rL r- r r r rQ � s z#TestRepositoryCache.decrypted_cache)rQ c C s� t |j|||g�� |j|||g�}t|�dks4t�t|j|�d��P}|jd tj� t j
|j � d�dA j dd�}|jd
tj� |j
|� |j� W d Q R X tjt�� t|�d ks�t�W d Q R X d S ) N� � 1234za+br �littler � 5678)rR rS ���rV )rR rU )r9 r: r> r �openZ key_filename�seek�io�SEEK_END�int�
from_bytes�read�to_bytesr rI �pytestZraisesr
)r rQ rN rO rP r? r Z corruptedr r r �test_cache_corruption� s
z)TestRepositoryCache.test_cache_corruptionN)r) r* r+ r_ Zfixturer- r5 r r; r@ rK rL rM rN rO rP rQ r` r r r r r, H s r, )rG r rY r' Z
unittest.mockr r_ Zremoter r r r- r Z
crypto.keyr �compressr Zhelpersr
Z hashindexr rL r
r r, r r r r �<module> s 6