File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/chunker_pytest.cpython-39.opt-1.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 d S )N�wbr � H� X)�open�write�seek�os�SEEK_CUR�truncate)�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 �path�joinr! �BSr r � SEEK_HOLE� SEEK_DATA�OSError)�tmpdir�fn�fZ offset_holeZ offset_datar r r �fs_supports_sparseD s
>r0 z fs does not support sparse files)�reasonzfname, sparse_mapZsparse1Zsparse2Z
onlysparseZ notsparsec C s* dd� }dd� }t | | �}t||� 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 r3 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 r5 r )r r r r r �get_sparsemap_fdd s z(test_sparsemap.<locals>.get_sparsemap_fd)�strr! )r- r �
sparse_mapr7 r8 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 )r9 r! r$ )r- r r: r r< r? r. r r r �test_chunkify_sparsen s r@ 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 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�rF r )�min_expr r � <genexpr>� rI z6test_buzhash_chunksize_distribution.<locals>.<genexpr>c 3 s | ]}t |d � k�V qdS rJ rK rM )�max_expr r rO � rI zcount: z min: z max: z min count: z max count: ) r �urandomr r r r= rE �min�max�sum�print) �data�maskr> r/ �chunksZ chunk_sizesZ chunks_countZmin_chunksize_observedZmax_chunksize_observedZ min_countZ max_countr )rP rN r �#test_buzhash_chunksize_distribution� s$
�� rY )r )r )�ior r r& Zpytestr> r r r r r Z constantsr) Z map_sparse1Z map_sparse2Z
map_notsparseZmap_onlysparser! r$ r0 ZmarkZskipifZ parametrizer; r@ rY r r r r �<module> sf � �
�
�