File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/compress.cpython-39.pyc
a
HZ�h�" � @ s� d dl Z d dlZd dlZz d dlZW n ey: dZY n0 d dlZddlmZmZm Z m
Z
m Z m Z m
Z
mZmZ ed�ZdZeddd�Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z dd � Z d d � Z d d � Z d!d"� Z!dS )#� N� ) �get_compressor�
Compressor�CompressionSpec�CNONE�ZLIB�LZ4�LZMA�ZSTD�Autoi s� fooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaarfooooooooobaaaaaaaar�zlib� ��name�levelc C s� t dd�} t| t�sJ �t dd�} t| t�s0J �t dd�} t| t�sHJ �t�t�� t dd� W d � n1 sr0 Y d S )N�none�r �lz4r Zfoobar)r �
isinstancer r r �pytest�raises�KeyError)�c� r �=/usr/lib64/python3.9/site-packages/borg/testsuite/compress.py�test_get_compressor s
r c C sf t dd�} | �t�}t|�tt�ks(J �t|v s4J �t| �|�ksFJ �ttf i t���|�ksbJ �d S )Nr r �r �compress�data�len�
decompressr �params�r �cdatar r r �
test_cnull s
r$ c C sZ t dd�} | �t�}t|�tt�k s(J �t| �|�ks:J �ttf i t���|�ksVJ �d S )Nr r r r" r r r �test_lz4' s
r% c C sl | � tddd� � t�d�d }t|�dks0J �td�}|�|�}t|�t|�ksVJ �||�|�kshJ �d S )NZdecidec S s t S )N)r )Zalways_compressr r r �<lambda>1 � z,test_lz4_buffer_allocation.<locals>.<lambda>i P �
i r )�setattrr �os�urandomr r r r )Z monkeypatchr r r# r r r �test_lz4_buffer_allocation/ s
r, c C sZ t dd�} | �t�}t|�tt�k s(J �t| �|�ks:J �ttf i t���|�ksVJ �d S )Nr r r r" r r r � test_zlib; s
r- c C sl t d u rt�d� tdd�} | �t�}t|�tt�k s:J �t| �|�ksLJ �ttf i t ���|�kshJ �d S )NzNo lzma support found.�lzmar )
r. r �skipr r r r r r r! r" r r r � test_lzmaC s
r0 c C sZ t dd�} | �t�}t|�tt�k s(J �t| �|�ks:J �ttf i t���|�ksVJ �d S )N�zstdr r r" r r r � test_zstdM s
r2 c C s� t �t��$ tf i t���d� W d � n1 s40 Y t �t��$ tf i t���d� W d � n1 sr0 Y d S )Ns ��totalcraps notreallyzlib)r r �
ValueErrorr r! r r r r r �test_autodetect_invalidU s 2 r4 c C sv t d�D ]h} td| d�}|�t�}t�t| �}||ks:J �|�|�}t|ksPJ �tf i t���|�}t|ksJ �qd S )Nr( r r )�ranger r r r r r r! )r r Zcdata1Zcdata2Zdata2r r r �test_zlib_compat\ s
r6 c
C s� t dd�t dd�t ddd�t ddd�t dd d�t dd
d�t dd d�g} trf| t d d d�t d d
d�g7 } | D ]*}tf i |��}t|�|�t��ksjJ �qjd S )
Nr r r r1 � r � r r r
� r. )�dictr. r r r r )Z params_listr! r r r r �test_compressorj s
�
�r; c C s� t d�j} t d�j}t d�j}td�}| �|�}|�|�}|�|�}t|�t|� }|dk rlt�|�tkstntstJ �d}| �|�}t�|�t ks�J �d S )Nz auto,zlib,9r �zlib,9i� g�G�z��?s ���-O�i��1�8\�b��E���hm�~)
r �
compressor�bytesr r r Zdetectr r r ) Zcompressor_auto_zlibZcompressor_lz4Zcompressor_zlibr Zcompressed_auto_zlibZcompressed_lz4Zcompressed_zlibZratio�
compressedr r r � test_auto� s
r@ c s� t d�j� td��� ���} t��d t| � krHt��d d ksNn J �t� �fdd�td�D ��dksrJ �t d �}t|jjt�s�J �|j� td��� ���} d
\}}|t�� d t| � kr�|t�� d d ks�n J �t� �fd d�td�D ��d
k�s J �t d�}t|jjt ��s(J �|j� td��� ���} d
\}}|t�� d t| � k�r�|t�� d d k�s�n J �t� �fdd�td�D ��dk�s�J �t d�}t|jjt
��s�J �|j� td��� ���} d
\}}|t�� d t| � k�r |t�� d d k�s&n J �t� �fdd�td�D ��d
k�sLJ �t d�}t|jjt ��shJ �|j� td��� ���} dt| � k�r�dk�s�n J �td��� ���} dt| � k�r�dk�s�n J �d S )Nzobfuscate,1,nonei' � �e c s h | ]}t � �����qS r �r r ��.0�i�r= r r r � <setcomp>� r' z!test_obfuscate.<locals>.<setcomp>�d �2 zobfuscate,2,lz4)g�������?g����MbP?i� c s h | ]}t � �����qS r rC rD rG r r rH � r' r( zobfuscate,6,zstd,3i��� c s h | ]}t � �����qS r rC rD rG r r rH � r' �Z zobfuscate,2,auto,zstd,10c s h | ]}t � �����qS r rC rD rG r r rH � r' zobfuscate,110,nonei� i� i� iL iT iT ) r r= r> r r r5 r �innerr r
r r )r? �csZ min_compressZ max_compressr rG r �test_obfuscate� sL
2$
:&
>&
>&
"
rN c C sn t �tj�� td� W d � n1 s*0 Y ttd�jt�sHJ �ttd�jt�s\J �td�j} t| t �stJ �| j
dks�J �td�j} t| t �s�J �| j
dks�J �td�j} t| t �s�J �| j
d ks�J �t �tj�� td
� W d � n1 s�0 Y td �j}t|t ��s J �|j
dk�s,J �td �j}t|t ��sFJ �|j
dk�sVJ �td
�j}t|t ��spJ �|j
d k�s�J �td�j}t|t ��s�J �|j
dk�s�J �td�j}t|t ��s�J �|j
dk�s�J �td�j}t|t ��s�J �|j
dk�s�J �t �tj�� td� W d � n1 �s*0 Y t �tj�� td� W d � n1 �s`0 Y d S )N� r r r r
zzlib,0r r< r9 zzlib,9,invalidr. zlzma,0zlzma,9r1 r8 zzstd,1r7 zzstd,22� zlzma,9,invalidZinvalid)
r r �argparse�ArgumentTypeErrorr r r= r r r r r r
)r r. r1 r r r �test_compression_specs� sJ &
&
(rS )"rQ r* r r. � ImportErrorr r r r r r r r r r
r r> �bufferr r: r! r r$ r% r, r- r0 r2 r4 r6 r; r@ rN rS r r r r �<module> s0
,
9