o +ke@sddlZddlZddlZddlTddlmZddlmZmZddl m Z ddl m Z dd l mZejejed Zd d Zd dZddZddZeddZejddgdddZddZeddZddZejje d d!d"d#Z ejje d d!d$d%Z!dS)&N)*) KeyfileKey)AtticRepositoryUpgraderAtticKeyfileKey) get_keys_dir) Repository)are_hardlinks_supportedz attic.tar.gzcsXfdd}t|d}|j|||dWdn1s wYtj|S)zo extract tar archive to , all stuff starting with . return path to . c3s"|D] }|jr|VqdSN)name startswith)memberstarinfowhat;usr/lib/python3.10/site-packages/borg/testsuite/upgrader.pyfiless  zuntar..filesr)rN)tarfileopen extractallospathjoin)Ztarfnamerrrtfrrruntars rcCs>tt|ddd }|WdS1swYdS)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)rstrcheck)r repositoryrrr repo_valid$s$r"cCsTtjttj|}t|}|tj WdS1s#wYdS)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) 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|Sr )r+)tmpdirrrr attic_repoHsr-TF)paramscCs|jSr )param)ZrequestrrrinplaceMsr0cCs|}ttj t|Wdn1swYt|dd}|dd|jD}Wdn1s9wY|j|d|d|j ddt|sRJdS)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) NFrcSsg|]\}}|qSrr).0ifilenamerrr `sz)test_convert_segments..dryrunr0r7) pytestraisesrAtticRepositoryr"rioZsegment_iteratorZconvert_segmentsZ convert_cache)r-r0 repo_pathr!segmentsrrrtest_convert_segmentsRs    r?cCsBttt|d}|d||d||ddtj|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_PASSPHRASEtestr()rr)rZsetenvrrr)r,Z monkeypatchZkeys_dirrrrattic_key_filefs    rBcCsd|}t|rJt|dd}t|}tj|ddWdn1s%wYt|s0JdS)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) Fr1r8N)r'rrZ find_key_fileZconvert_keyfiles)r-rBZ keyfile_pathr!r%rrr test_keys~s  rCzhardlinks not supported)reasoncs|}ttj t|Wdn1swYddfdd}||}t|ddE}tt|j d|d}|rO|dusEJ||j |ksNJn|sSJ||j ||ks^Jd tj vrn|j t@d ksnJWdn1sxwYt |sJt|sJdS) 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) NcSsttj|dddS)Ndata0)rstatrrrrrr stat_segmentsz&test_convert_all..stat_segmentcs |jSr )st_inorHrIrr first_inodes z%test_convert_all..first_inodeFr1r6ZBORG_TESTS_IGNORE_MODESr)r9r:rr;r"rrumaskZ UMASK_DEFAULTupgraderenvironst_moder')r-rBr0r=rLZ orig_inoder!ZbackuprrKrtest_convert_alls*       rQcCst|d}t|d }|dWdn1swYt|d}t||tj|dd|d|sHt|j t|j ksGJnt|j t|j ksVJt|d }| d ksdJWddS1sowYdS) 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)r0rbsbXXX) rrrwriterlinkrZheader_replacerGrJr$)r,r0rRtmprTrrr test_hardlinks    "r[)"rrr9 constantsZ crypto.keyrZupgraderrrhelpersrr!rr rrdirname__file__r)rr"r'r+Zfixturer-r0r?rBrCZmarkZskipifrQr[rrrrs4          )