­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 yb@sddlZddlZddlZddlTddlmZddlmZmZddl m Z ddl m Z dd l mZejjejjed Zd d Zd dZddZddZejddZejddgdddZddZejddZddZejje d d!d"d#Z ejje d d!d$d%Z!dS)&N)*) KeyfileKey)AtticRepositoryUpgraderAtticKeyfileKey) get_keys_dir) Repository)are_hardlinks_supportedz attic.tar.gzc sDfdd}tj|d}|j|||dWdQRXtjj|S)zo extract tar archive to , all stuff starting with . return path to . c3s$x|D]}|jjr|VqWdS)N)name startswith)membersZtarinfo)what /usr/lib64/python3.6/upgrader.pyfiless  zuntar..filesr)r N)tarfileopenZ extractallospathjoin)ZtarfnamerrrZtfr)rruntars rc Cs&tt|ddd }|jSQRXdS)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 exclusivecreateN)rstrZcheck)r repositoryrrr repo_valid$src Cs>tjjttjj|}t|d}|jjtj SQRXdS)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 rN) rrrrbasenamerreadr rZFILE_ID)rkeyfilefrrr key_valid0s  r!cCsttt|dS)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)dirrrrmake_attic_repo=sr%cCst|S)N)r%)tmpdirrrr attic_repoHsr'TF)ZparamscCs|jS)N)Zparam)ZrequestrrrinplaceMsr(cCs|}tjtjt|WdQRXt|dd}|dd|jjD}WdQRX|j|d|d|j ddt|s|t dS)atest 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)rcSsg|] \}}|qSrr).0ifilenamerrr `sz)test_convert_segments..)dryrunr()r-) pytestraisesrAtticRepositoryrrioZsegment_iteratorZconvert_segmentsZ convert_cacheAssertionError)r'r( repo_pathrZsegmentsrrrtest_convert_segmentsRs   r4cCsBttt|d}|jd||jd||jddtjj|dS)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")rr#rZsetenvrrr)r&Z monkeypatchZkeys_dirrrrattic_key_filefs    r6c CsR|}t| stt|dd}tj|}tj|ddWdQRXt|sNtdS)autest 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!r2rrZ find_key_fileZconvert_keyfiles)r'r6Z keyfile_pathrrrrr test_keys~s  r7zhardlinks not supported)reasoncs|}tjtjt|WdQRXddfdd}||}t|dd}tjt|j d|d}|r|dksvt ||j |kst n>|st ||j ||kst d tj krȈ|j t@d kst WdQRXt|st t|st 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) NcSstjtjj|dddS)Ndata0)rstatrr)rrrr stat_segmentsz&test_convert_all..stat_segmentcs |jS)N)st_ino)r)r<rr first_inodesz%test_convert_all..first_inodeF)r)r-r(ZBORG_TESTS_IGNORE_MODESr)r.r/rr0rrrumaskZ UMASK_DEFAULTZupgrader2renvironst_moder!)r'r6r(r3r>Z orig_inoderZbackupr)r<rtest_convert_alls$       rBcCst|jd}t|d}|jdWdQRXt|jd}tj||tj|dd|d|s|tj|j tj|j kst ntj|j tj|j kst t|d }|j d kst WdQRXdS) abtest 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.awbsaXXXNbab)r(rbsbXXX) rrrwriterlinkrZheader_replacer;r=r2r)r&r(rCZtmprErrr test_hardlinks    rK)"rrr.Z constantsZ crypto.keyrZupgraderrrZhelpersrrrr rrdirname__file__r#rrr!r%Zfixturer'r(r4r6r7ZmarkZskipifrBrKrrrrs(       *