o hq@sdZddlmZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZGddde jZd d Zd d ZdddZdddZdddZddZdddZGdddZGdddeZdS)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 GITDB_TEST_GIT_REPO_BASEcs~ztWn tyYnwtj|j|_|js5t d|jtj j }tj |||t d|_|jds=JdS)NzpYou can set the %s environment variable to a .git repository of your choice - defaulting to the gitdb repositoryz.git)super setUpClassAttributeErrorosenvirongetk_env_git_repo gitrepopathlogginginfopathdirnamejoin__file__endswith)clsospd __class__B/var/www/vscode/kcb/lib/python3.10/site-packages/gitdb/test/lib.pyr (s zTestBase.setUpClass)__name__ __module__ __qualname____doc__r classmethodr __classcell__rrrrrs  rcfdd}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}z2z||WW|s"tt|SSt y@t j dt |jdjd|dd}w|sMtt|ww)N)prefixFzTest .z failed, output is at  T)tempfilemktemprr mkdirgccollectshutilrmtree Exceptionsysstderrwritetype)selfrkeepfuncrrwrapperAs&     (  z"with_rw_directory..wrapperrr6r7rr5rwith_rw_directory=s r:cr#)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)r3r src_pack_globr5rrr7_s zwith_packs_rw..wrapperr8r9rr5r with_packs_rw[s r?cCstjtjtd|S)z:return: absolute path into the fixture directory :param relapath: relative path into the fixtures directory, or '' to obtain the fixture directory itselffixtures)r rrrr)relapathrrrr<msr<Fc Cstt|D]2}|r1ttdr1tj|tj|}zt||Wqty0t ||Yqwt ||qdS)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 rrbasenamerCOSErrorr,copy) source_glob target_dirr;src_filetargetrrrr=ts r=cCs8|d}t|}|rt|}t|td|}|S)zb:return: string with given size in bytes :param randomize: try to produce a very random streami)rangelistrandomshufflertobytes) size_in_bytes randomize actual_sizeproducerarrr make_bytess  rYcCsdt|}|d|S)z0:return: bytes resembling an uncompressed objectzblob %iascii)lenencode)r2dataodatarrr make_objects r_cCst||}t|t|fS)z`:return: tuple(size_of_stream, stream) :param randomize: try to produce a very random stream)rYr[r)rTrUdrrrmake_memory_files rac@s,eZdZddZddZddZddZd S) DummyStreamcCsd|_d|_d|_dS)NFr)was_readbytesclosedr3rrr__init__s zDummyStream.__init__cCsd|_||_dSNT)rcrd)r3sizerrrreads zDummyStream.readcCs d|_dSrh)rerfrrrcloses zDummyStream.closecCs|jsJdSN)rcrfrrr_assertszDummyStream._assertN)rrrrgrjrkrmrrrrrbs  rbc@seZdZddZddZdS) DeriveTestcOs|d|_||_dS)Nmyarg)poproargs)r3shar2ristreamrqkwargsrrrrgs  zDeriveTest.__init__cCs|jsJ|js JdSrl)rqrorfrrrrms zDeriveTest._assertN)rrrrgrmrrrrrns rn)r@)F)r gitdbrr/rQriorrDunittestr'r,r r*r functoolsrTestCaserr:r?r<r=rYr_rarbrnrrrrs0    #