File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/upgrader.cpython-36.opt-1.pyc
3
y��b� � @ s d dl Z d dlZd dlZddlT ddlmZ ddlmZmZ ddl m
Z
ddl m Z dd l
mZ e jje jje�d
�Zd d � Zd
d� Zdd� Zdd� Zej� dd� �Zejddgd�dd� �Zdd� Zej� 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 sD � fdd�}t j| d��}|j|||�d� W dQ R X tjj|� �S )zo
extract <tarfname> tar archive to <path>, all stuff starting with <what>.
return path to <what>.
c 3 s$ x | D ]}|j j� �r|V qW d S )N)�name�
startswith)�membersZtarinfo)�what� � /usr/lib64/python3.6/upgrader.py�files s
zuntar.<locals>.files�r)r
N)�tarfile�openZ
extractall�os�path�join)Ztarfnamer r r Ztfr )r r �untar s r c C s& t t| �ddd��
}|j� S Q R X 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 �strZcheck)r �
repositoryr r r �
repo_valid$ s r c C s> t jjt� t jj| ��}t|d��}|j� jtj �S Q R X 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
r 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 )N)r% )�tmpdirr r r �
attic_repoH s r' TF)Zparamsc C s | j S )N)Zparam)Zrequestr r r �inplaceM s r( c C st | }t jtj�� t|� W dQ R X t|dd�}|� dd� |jj� D �}W dQ R X |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>` s z)test_convert_segments.<locals>.<listcomp>)�dryrunr( )r- )
�pytest�raisesr �AtticRepositoryr r �ioZsegment_iteratorZconvert_segmentsZ
convert_cache)r' r( � repo_pathr Zsegmentsr r r �test_convert_segmentsR s r3 c C sB t tt| �d�}|jd|� |jd|� |jdd� tjj|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_PASSPHRASEZtestr" )r r# r Zsetenvr r r )r& Z monkeypatchZkeys_dirr r r �attic_key_filef s
r5 c C s8 |}t | dd�� }tj|�}t j|dd� W dQ R X 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)
F)r )r- N)r r Z
find_key_fileZconvert_keyfiles)r' r5 Z keyfile_pathr r r r r � test_keys~ s
r6 zhardlinks not supported)�reasonc s� | }t jtj�� t|� W dQ R X dd� � � fdd�}||�}t|dd��.}tjt� |j d|d�}|rln
d tj
krvW dQ R X 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 jt jj| 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 )N)�st_ino)r )r; r r � first_inode� s z%test_convert_all.<locals>.first_inodeF)r )r- r( ZBORG_TESTS_IGNORE_MODES) r. r/ r r0 r r r �umaskZ
UMASK_DEFAULTZupgrade�environ)r' r5 r( r2 r= Z
orig_inoder Zbackupr )r; r �test_convert_all� s
r@ c C sz t | jd��}t|d��}|jd� W dQ R X t | jd��}tj||� tj|dd|d� |s`n t|d ��}W dQ R X 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( rA ZtmprC r r r �
test_hardlink� s
rI )"r r r. Z constantsZ
crypto.keyr Zupgraderr r Zhelpersr r r � r
r r �dirname�__file__r# r r r! r% Zfixturer' r( r3 r5 r6 ZmarkZskipifr@ rI r r r r �<module> s(
*