File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/upgrader.cpython-39.opt-1.pyc
a
HZ�h� � @ s d dl Z d dlZd dlZddlT ddlmZ ddlmZmZ ddl m
Z
ddl m Z dd l
mZ e j�e j�e�d
�Zd d � Zd
d� Zdd� Zdd� Ze�� dd� �Zejddgd�dd� �Zdd� Ze�� d d � �Z d d � Z ej j e� d d!�d"d#� �Z ej j e� d d!�d$d%� �Z!dS )&� N� )�*)�
KeyfileKey)�AtticRepositoryUpgrader�AtticKeyfileKey)� get_keys_dir)�
Repository� )�are_hardlinks_supportedz attic.tar.gzc sX � fdd�}t �| d��"}|j|||�d� W d � n1 s@0 Y tj�|� �S )zo
extract <tarfname> tar archive to <path>, all stuff starting with <what>.
return path to <what>.
c 3 s | D ]}|j �� �r|V qd S �N)�name�
startswith)�members�tarinfo��what� �=/usr/lib64/python3.9/site-packages/borg/testsuite/upgrader.py�files s zuntar.<locals>.files�r)r N)�tarfile�open�
extractall�os�path�join)Ztarfnamer r r Ztfr r r �untar s 0r c C s> t t| �ddd��}|�� W d � S 1 s00 Y dS )z�
utility function to check if borg can open a repository
:param path: the path to the repository
:returns: if borg can check the repository
TF)Z exclusive�createN)r �str�check)r �
repositoryr r r �
repo_valid$ s r! c C sT t j�t� t j�| ��}t|�� }|�� �tj �W d � S 1 sF0 Y dS )z�
check that the new keyfile is alright
:param path: the path to the key file
:returns: if the file starts with the borg magic string
N)
r r r r �basenamer �readr
r ZFILE_ID)r �keyfile�fr r r � key_valid0 s
�
r& c C s t tt| �d�S )z�
create an attic repo with some stuff in it
:param dir: path to the repository to be created
:returns: path to attic repository
�repo)r � ATTIC_TARr )�dirr r r �make_attic_repo= s r* c C s t | �S r )r* )�tmpdirr r r �
attic_repoH s r, TF)�paramsc C s | j S r )Zparam)Zrequestr r r �inplaceM s r. c C s� | }t �tj�� t|� W d � n1 s.0 Y t|dd�}|�$ dd� |j�� D �}W d � n1 sr0 Y |j|d|d� |j dd� dS )a test segment conversion
this will load the given attic repository, list all the segments
then convert them one at a time. we need to close the repo before
conversion otherwise we have errors from borg
:param attic_repo: a populated attic repository (fixture)
NF�r c S s g | ] \}}|�qS r r )�.0�i�filenamer r r �
<listcomp>` � z)test_convert_segments.<locals>.<listcomp>��dryrunr. �r6 )
�pytest�raisesr �AtticRepositoryr! r �ioZsegment_iteratorZconvert_segmentsZ
convert_cache)r, r. � repo_pathr �segmentsr r r �test_convert_segmentsR s & 2 r> c C sB t tt| �d�}|�d|� |�d|� |�dd� tj�|d�S )z�
create an attic key file from the given repo, in the keys
subdirectory of the given tmpdir
:param tmpdir: a temporary directory (a builtin fixture)
:returns: path to key file
�keysZATTIC_KEYS_DIRZ
BORG_KEYS_DIRZATTIC_PASSPHRASE�testr' )r r( r Zsetenvr r r )r+ Z monkeypatchZkeys_dirr r r �attic_key_filef s
rA c C sL |}t | dd��(}t�|�}t j|dd� W d � n1 s>0 Y dS )au test key conversion
test that we can convert the given key to a properly formatted
borg key. assumes that the ATTIC_KEYS_DIR and BORG_KEYS_DIR have
been properly populated by the attic_key_file fixture.
:param attic_repo: path to an attic repository (fixture defined above)
:param attic_key_file: path to an attic key file (fixture defined above)
Fr/ r7 N)r r Z
find_key_fileZconvert_keyfiles)r, rA Z keyfile_pathr r$ r r r � test_keys~ s
,rB zhardlinks not supported)�reasonc s� | }t �tj�� t|� W d � n1 s.0 Y dd� � � fdd�}||�}t|dd��8}t�t� |j d|d�}|r�n
d tj
vr�W d � n1 s�0 Y dS )
a[ test all conversion steps
this runs everything. mostly redundant test, since everything is
done above. yet we expect a NotImplementedError because we do not
convert caches yet.
:param attic_repo: path to an attic repository (fixture defined above)
:param attic_key_file: path to an attic key file (fixture defined above)
Nc S s t �t j�| ddd��S )N�data�0)r �statr r �r r r r � stat_segment� s z&test_convert_all.<locals>.stat_segmentc s
� | �j S r )�st_inorG �rH r r � first_inode� s z%test_convert_all.<locals>.first_inodeFr/ r5 ZBORG_TESTS_IGNORE_MODES) r8 r9 r r: r! r r �umaskZ
UMASK_DEFAULTZupgrade�environ)r, rA r. r< rK Z
orig_inoder Zbackupr rJ r �test_convert_all� s &
rN c C s� t | �d��}t|d��}|�d� W d � n1 s80 Y t | �d��}t�||� tj|dd|d� |stn t|d ��}W d � n1 s�0 Y dS )
ab test that we handle hard links properly
that is, if we are in "inplace" mode, hardlinks should *not*
change (ie. we write to the file directly, so we do not rewrite the
whole file, and we do not re-create the file).
if we are *not* in inplace mode, then the inode should change, as
we are supposed to leave the original inode alone.�a�wbs aXXXN�b� a� b)r. �rb)r r r �writer �linkr Zheader_replace)r+ r. rO �tmprQ r r r �
test_hardlink� s
( rX )"r r r8 Z constantsZ
crypto.keyr Zupgraderr r Zhelpersr r r � r
r r �dirname�__file__r( r r! r& r* Zfixturer, r. r> rA rB ZmarkZskipifrN rX r r r r �<module> s2
)