U 楡cw@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_BASEcsztWntk r"YnXtj|j|_|jslt d|jtj j }tj |||t d|_|jds|tdS)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__endswithAssertionError)clsZospd __class__2/tmp/pip-unpacked-wheel-jjg5dubb/gitdb/test/lib.pyr(szTestBase.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}zXz||WWHSt k rpt j dt |jdjd|dd}YnXW5|stt|XdS)N)prefixFzTest .z failed, output is at  T)tempfilemktemprr mkdirgcZcollectshutilrmtree Exceptionsysstderrwritetype)selfrZkeepfuncrrwrapperAs ( z"with_rw_directory..wrapperrr1r2rr0rwith_rw_directory=s r5csfdd}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_globr0rrr2_szwith_packs_rw..wrapperr3r4rr0r with_packs_rw[s r9cCstjtjtd|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)Zrelapathrrrr7msr7Fc Csxt|D]h}|rfttdrftj|tj|}zt||Wqrtk rbt ||YqrXq 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_dirr6Zsrc_filetargetrrrr8tsr8cCs8|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  rMcCsdt|}|d|S)z0:return: bytes resembling an uncompressed objectzblob %iascii)lenencode)r.dataodatarrr make_objects rScCst||}t|t|fS)z`:return: tuple(size_of_stream, stream) :param randomize: try to produce a very random stream)rMrOr)rJrKdrrrmake_memory_files rUc@s,eZdZddZddZddZddZd S) DummyStreamcCsd|_d|_d|_dS)NFr)was_readbytesclosedr/rrr__init__szDummyStream.__init__cCsd|_||_dSNT)rWrX)r/sizerrrreadszDummyStream.readcCs d|_dSr\)rYrZrrrcloseszDummyStream.closecCs|js tdSN)rWrrZrrr_assertszDummyStream._assertN)rrrr[r^r_rarrrrrVsrVc@seZdZddZddZdS) DeriveTestcOs|d|_||_dS)Nmyarg)poprcargs)r/shar.r]streamrekwargsrrrr[s zDeriveTest.__init__cCs|js t|jstdSr`)rerrcrZrrrras zDeriveTest._assertN)rrrr[rarrrrrbsrb)r:)F)F)F)rZgitdbrr+rGriorr<Zunittestr$r(r r'r functoolsrZTestCaserr5r9r7r8rMrSrUrVrbrrrrs.    #