File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/benchmark.cpython-36.opt-1.pyc
3
y��b/ � @ s� d Z ddlZddlZddlmZmZ ejdd� �Zejddgd �d
d � �Zejd d
dgd�dd� �Z ejddgd �dd� �Z
dd� Z dd� Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� ZdS )%zM
Do benchmarks using pytest-benchmark.
Usage:
py.test --benchmark-only
� N� )� changedir�cmdc c s| |j dd� |j dd� |j dd� |j dd� |j dt|jd ��� |j d
t|jd ��� t|jd ��V |jd
d� d S )NZBORG_PASSPHRASEZ123456Z!BORG_CHECK_I_KNOW_WHAT_I_AM_DOINGZYESZ"BORG_DELETE_I_KNOW_WHAT_I_AM_DOINGZ*BORG_UNKNOWN_UNENCRYPTED_REPO_ACCESS_IS_OK�yesZ
BORG_KEYS_DIR�keysZBORG_CACHE_DIR�cacheZ
repositoryr )�rec)Zsetenv�str�join�remove)�request�tmpdirZ monkeypatch� r �!/usr/lib64/python3.6/benchmark.py�repo_url s r �noneZrepokey)�paramsc C s |dd| j |� |S )NZinitz --encryption)�param)r r r r r r �repo s r Zsession�zeros�random)Zscoper c c s� d\}}|j d�}| j}|dkr*dd� }n|dkr<dd� }ntd ��xBt|�D ]6}tt|jt|���d
��}|j||�� W d Q R X qNW t|�V |jd d � d S )N�
i� �datar c S s d| S )N� 0r )�sizer r r r ) s ztestdata.<locals>.datar c S s
t j| �S )N)�os�urandom)r r r r r , s z&data_type must be 'random' or 'zeros'.�wbr )r �@B )r r ) Zmktempr �
ValueError�range�openr r
�writer ) r Ztmpdir_factory�countr �pZ data_typer �i�fr r r �testdata" s
r' �lz4c C s |d }|dd| j ||� |S )Nz::test�createz
--compression)r )r r r r' Z archive_urlr r r �archive7 s r* c C s" | j |ddd|d |f�\}}d S )Nr) z
--compressionr z::test)�pedantic)� benchmarkr r r' �result�outr r r �test_create_none> s r/ c C s" | j |ddd|d |f�\}}d S )Nr) z
--compressionr( z::test)r+ )r, r r r' r- r. r r r �test_create_lz4C s r0 c C s0 t t|��� | j|d|f�\}}W d Q R X d S )N�extract)r r r+ )r, r r* r
r- r. r r r � test_extractH s r2 c C s | j |d|f�\}}d S )N�delete)r+ )r, r r* r- r. r r r � test_deleteN s r4 c C s | |d|�\}}d S )N�listr )r, r r* r- r. r r r � test_listS s r6 c C s | |d|�\}}d S )N�infor )r, r r* r- r. r r r � test_infoX s r8 c C s" |j d�d }| |d|�\}}d S )Nz::r Zcheck)�split)r, r r* r r- r. r r r �
test_check] s r: c C s | |d�\}}d S )N�helpr )r, r r- r. r r r � test_helpc s r< )�__doc__r ZpytestZarchiverr r Zfixturer r r' r* r/ r0 r2 r4 r6 r8 r: r< r r r r �<module> s