File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/chunker_pytest.cpython-39.pyc
a
HZ�h. � @ s8 d dl mZ d dlZd dlZd dlZddlmZ ddlmZmZm Z m
Z
ddl T dZ d e de d fde de d
fd e d e d fd e d
e d
fgZ
d e de d
fde de d fd e d e d
fd e d
e d fgZd e d e d fgZd e d e d
fgZd"dd�Zd#dd�Zdd� Zejje� dd�ej�dde
fdefdefdefg�dd � ��Zejje� dd�ej�d de
d d
fde
d d fde
e d
fde
e d fded d
fded d fdee d
fdee d fded d
fded d fdee d
fdee d fded d
fded d fdee d
fdee d fg�d d � ��Zd d!� ZdS )$� )�BytesION� )�cf� )�Chunker� ChunkerFixed� sparsemap�
has_seek_hole)�*� TF� � � c C s� t | d��r}d}|r*|�d| � ||7 }|D ]4\}}}|rL|�d| � n|�|tj� ||7 }q.|�|� W d � n1 s�0 Y tj�| �|ks�J �d S )N�wbr � H� X)�open�write�seek�os�SEEK_CUR�truncate�path�getsize)�fnamer � header_size�fd�total�offset�size�is_data� r! �C/usr/lib64/python3.9/site-packages/borg/testsuite/chunker_pytest.py�make_sparsefile% s
(r# c C s\ g }d}|r"|� d| � ||7 }| D ]0\}}}|rD|� d| � n
|� |� ||7 }q&|S )Nr r r )�append)r r �resultr r r r r! r! r" � make_content5 s
r& c
C s� t sdS t�� ��} tj�| d�}t|dtdfttdfg� t|d��^}z |� dtj
�}|� dtj �}W n, t y� Y W d � W d � dS 0 W d � n1 s�0 Y |dko�|tkW d � S 1 s�0 Y d S )NFZ test_sparser T�rb)
r �tempfileZTemporaryDirectoryr r �joinr# �BSr r � SEEK_HOLE� SEEK_DATA�OSError)�tmpdir�fn�fZ offset_holeZ offset_datar! r! r" �fs_supports_sparseD s
>r1 z fs does not support sparse files)�reasonzfname, sparse_mapZsparse1Zsparse2Z
onlysparseZ notsparsec C sJ dd� }dd� }t | | �}t||� ||�|ks6J �||�|ksFJ �d S )Nc S s< t j| t jd�}ztt|d��W t �|� S t �|� 0 d S )N)�flags)�fh)r r �O_RDONLY�listr �close)r r4 r! r! r" �get_sparsemap_fh] s
�z(test_sparsemap.<locals>.get_sparsemap_fhc S s< t | d�� }tt|d��W d � S 1 s.0 Y d S )Nr' )r )r r6 r )r r r! r! r" �get_sparsemap_fdd s z(test_sparsemap.<locals>.get_sparsemap_fd)�strr# )r. r �
sparse_mapr8 r9 r/ r! r! r" �test_sparsemapT s
r< z&fname, sparse_map, header_size, sparsec C s@ dd� }t | | �}t|||d� ||||d�t||d�k d S )Nc S sJ t d||d�}t| d�� }t|�|��W d � S 1 s<0 Y d S )Nr )r �sparser' )r r r �chunkify)r r= r �chunkerr r! r! r" �
get_chunks� s z(test_chunkify_sparse.<locals>.get_chunks)r )r= r )r: r# r& )r. r r; r r= r@ r/ r! r! r" �test_chunkify_sparsen s rA c s t �d�} d\�� }td�� |d�}t| �}t|�|��}|d= dd� |D �}t|�}t|�}t|�}t �fdd �|D ��} t � fd
d �|D ��}
t
d |� d |� d
|� d| � d|
� �
� d| k r�dk s�n J �|d� ks�J �|d� ks�J �| dk s�J �|
dk �s J �d S )Ni )�
� � r i� ���c S s g | ] }t |��qS r! )�len)�.0�chunkr! r! r" �
<listcomp>� � z7test_buzhash_chunksize_distribution.<locals>.<listcomp>c 3 s | ]}t |d � k�V qdS �r N��int�rG r )�min_expr! r" � <genexpr>� rJ z6test_buzhash_chunksize_distribution.<locals>.<genexpr>c 3 s | ]}t |d � k�V qdS rK rL rN )�max_expr! r" rP � rJ zcount: z min: z max: z min count: z max count: � � r rB ) r �urandomr r r r> rF �min�max�sum�print) �data�maskr? r0 �chunksZ chunk_sizesZ chunks_countZmin_chunksize_observedZmax_chunksize_observedZ min_countZ max_countr! )rQ rO r" �#test_buzhash_chunksize_distribution� s,
�� r\ )r )r )�ior r r( Zpytestr? r r r r r Z constantsr* Z map_sparse1Z map_sparse2Z
map_notsparseZmap_onlysparser# r& r1 ZmarkZskipifZ parametrizer< rA r\ r! r! r! r" �<module> sf � �
�
�