#Cc`dZddlmZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZGdde jZd Zd Zd Zdd ZddZddZdZddZGddeZGddeZdS)zUtilities used in ODB testing)OStreamN)array)BytesIOwrapsc6eZdZdZdZefdZxZS)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_BASEc  tt|n#t$rYnwxYwtj|j|_|jsptj d|jtj j }tj |||td|_|jdsJdS)NzpYou can set the %s environment variable to a .git repository of your choice - defaulting to the gitdb repositoryz.git)superr setUpClassAttributeErrorosenvirongetk_env_git_repo gitrepopathlogginginfopathdirnamejoin__file__endswith)clsospd __class__s .lib/python3.11/site-packages/gitdb/test/lib.pyr zTestBase.setUpClass(s  (C + + - - - -    D *..);<< O LCEHEW X X X7?D gll44TT(^^0D0D+E+EvNNCO''///////s '+ 88)__name__ __module__ __qualname____doc__r classmethodr __classcell__)rs@rr r sU0N 0 0 0 0[ 0 0 0 0 0r c<tfd}|S)zAll tests decorated with this one will raise SkipTest when run on travis ci. Use it to workaround difficult to solve issues NOTE: copied from bcore (https://github.com/Byron/bcore)cjdtjvrddl}|d|g|Ri|S)NTRAVISrzCannot run on travis-ci)rrnoseSkipTest)selfargskwargsr)funcs rwrapperz"skip_on_travis_ci..wrapperAsL rz ! ! KKK-- 9:: :tD*4***6***r%rr.r/s` rskip_on_travis_cir1=s5 4[[++++[+ Nr%c*fd}j|_|S)zCreate a temporary directory which can be written to, remove it if the test succeeds, but leave it otherwise to aid additional debuggingctjj}tj|d} |||s(t jtj|SS#t$rOtj d t|jj|d}wxYw#|s(t jtj|wwxYw)N)prefixFz%Test {}.{} failed, output is at {!r} T)tempfilemktemprrmkdirgccollectshutilrmtree Exceptionsysstderrwriteformattype)r+rkeepr.s rr/z"with_rw_directory..wrapperPsdm444  $ tD$'' $  d#### $      !I!P!PQUVZQ[Q[Qdfjfsuy!z!z{{{  $  d#### $s A**ACC,C2rr0s` rwith_rw_directoryrDLs-$$$$$,}G Nr%c*fd}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 afterwardsc^td}t||d||S)Nzpacks/*T) hard_link_ok) fixture_pathcopy_files_globbed)r+r src_pack_globr.s rr/zwith_packs_rw..wrapperns7$Y// =$TBBBBtD$r%rCr0s` r with_packs_rwrKjs-     }G Nr%ctjtjtd|S)z:return: absolute path into the fixture directory :param relapath: relative path into the fixtures directory, or '' to obtain the fixture directory itselffixtures)rrrrr)relapaths rrHrH|s* 7<<11:x H HHr%Fc~tj|D]}|rttdrytj|tj|} tj||m#t$rtj ||YwxYwtj ||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) globhasattrrrrbasenamerQOSErrorr:copy) source_glob target_dirrGsrc_filetargets rrIrIsIk** . .  .GB// .W\\*bg.>.>x.H.HIIF 2&)))) 2 2 2 Hj11111 2 K* - - - - . .s,BB$#B$c|dz}t|}|r#t|}tj|t d|}|S)zb:return: string with given size in bytes :param randomize: try to produce a very random streami)rangelistrandomshufflertobytes) size_in_bytes randomize actual_sizeproduceras r make_bytesrhs[ 1$K[!!H!>>x    c8A 99;;r%cVdt|z}|d|zS)z0:return: bytes resembling an uncompressed objectzblob %iascii)lenencode)rAdataodatas r make_objectros) #d)) #E << 4 ''r%c^t||}t|t|fS)z`:return: tuple(size_of_stream, stream) :param randomize: try to produce a very random stream)rhrkr)rcrdds rmake_memory_filerrs* =),,A q6671:: r%c&eZdZdZdZdZdZdS) DummyStreamc0d|_d|_d|_dS)NFr)was_readbytesclosedr+s r__init__zDummyStream.__init__s   r%c"d|_||_dSNT)rvrw)r+sizes rreadzDummyStream.reads  r%cd|_dSr|)rxrys rclosezDummyStream.closes  r%c|jsJdSN)rvrys r_assertzDummyStream._asserts}r%N)rr r!rzr~rrr%rrtrtsP r%rtceZdZdZdZdS) DeriveTestcH|d|_||_dS)Nmyarg)poprr,)r+sharAr}streamr,r-s rrzzDeriveTest.__init__s ZZ((  r%c*|jsJ|jsJdSr)r,rrys rrzDeriveTest._asserts$yzr%N)rr r!rzrrr%rrrs2r%r)rL)F)r"gitdbrr=r`riorrRunittestr5r:rr8r functoolsrTestCaser r1rDrKrHrIrhrorrobjectrtrrr%rrs $#    00000x 000F   <   $IIII . . . .$    ((( &$r%