a (Dfq@sdZddlmZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZGddde jZd d Zd d ZdddZdddZdddZddZd ddZGdddZGdddeZdS)!zUtilities used in ODB testing)OStreamN)array)BytesIO)wrapscs(eZdZdZdZefddZZS)TestBasezBase class for all tests TestCase providing access to readonly repositories using the following member variables. * gitrepopath * read-only base path of the git source repository, i.e. .../git/.git ZGITDB_TEST_GIT_REPO_BASEcs~ztWnty Yn0tj|j|_|jsjt d|jtj j }tj |||t d|_|jdszJdS)NzpYou can set the %s environment variable to a .git repository of your choice - defaulting to the gitdb repositoryz.git)super setUpClassAttributeErrorosenvirongetk_env_git_repoZ gitrepopathlogginginfopathdirnamejoin__file__endswith)clsZospd __class__[/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/gitdb/test/lib.pyr(s zTestBase.setUpClass)__name__ __module__ __qualname____doc__r classmethodr __classcell__rrrrrs rcsfdd}j|_|S)zCreate a temporary directory which can be written to, remove it if the test succeeds, but leave it otherwise to aid additional debuggingc stjjd}t|d}zz$||WW|sBtt|St yt j dt |jdjd|dd}Yn0W|stt|n|stt|0dS)N)prefixFzTest .z failed, output is at  T)tempfilemktemprr mkdirgcZcollectshutilrmtree Exceptionsysstderrwritetype)selfrZkeepfuncrrwrapperAs(    (  z"with_rw_directory..wrapperrr0r1rr/rwith_rw_directory=s r4csfdd}j|_|S)zFunction that provides a path into which the packs for testing should be copied. Will pass on the path to the actual function afterwardscs td}t||dd||S)Nzpacks/*T) hard_link_ok) fixture_pathcopy_files_globbed)r.rZ src_pack_globr/rrr1_szwith_packs_rw..wrapperr2r3rr/r with_packs_rw[s r8cCstjtjtd|S)z:return: absolute path into the fixture directory :param relapath: relative path into the fixtures directory, or '' to obtain the fixture directory itselfZfixtures)r rrrr)Zrelapathrrrr6msr6Fc Csvt|D]f}|rdttdrdtj|tj|}zt||Wqpty`t ||Yqp0q t ||q dS)zCopy all files found according to the given source glob into the target directory :param hard_link_ok: if True, hard links will be created if possible. Otherwise the files will be copiedlinkN) globhasattrr rrbasenamer:OSErrorr'copy)Z source_glob target_dirr5Zsrc_filetargetrrrr7ts r7cCs8|d}t|}|r&t|}t|td|}|S)zb:return: string with given size in bytes :param randomize: try to produce a very random streami)rangelistrandomshufflertobytes) size_in_bytes randomizeZ actual_sizeZproducerarrr make_bytess  rLcCsdt|}|d|S)z0:return: bytes resembling an uncompressed objectzblob %iascii)lenencode)r-dataodatarrr make_objects rRcCst||}t|t|fS)z`:return: tuple(size_of_stream, stream) :param randomize: try to produce a very random stream)rLrNr)rIrJdrrrmake_memory_files rTc@s,eZdZddZddZddZddZd S) DummyStreamcCsd|_d|_d|_dS)NFr)was_readbytesclosedr.rrr__init__szDummyStream.__init__cCsd|_||_dSNT)rVrW)r.sizerrrreadszDummyStream.readcCs d|_dSr[)rXrYrrrcloseszDummyStream.closecCs|js JdSN)rVrYrrr_assertszDummyStream._assertN)rrrrZr]r^r`rrrrrUsrUc@seZdZddZddZdS) DeriveTestcOs|d|_||_dS)Nmyarg)poprbargs)r.shar-r\streamrdkwargsrrrrZs zDeriveTest.__init__cCs|js J|jsJdSr_)rdrbrYrrrr`s zDeriveTest._assertN)rrrrZr`rrrrrasra)r9)F)F)F)rZgitdbrr*rFriorr;Zunittestr#r'r r&r functoolsrZTestCaserr4r8r6r7rLrRrTrUrarrrrs.    #