gqdZddlmZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZGdde j"Zd Zd Zdd Zdd Zdd ZdZddZGddZGddeZy)zUtilities used in ODB testing)OStreamN)array)BytesIO)wrapsc0eZdZdZdZefdZxZS)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  t|tjj |j |_|jsutjd|j tjj}tjj|||td|_|jjdsJy#t$rYwxYw)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__s C/var/www/openai/venv/lib/python3.12/site-packages/gitdb/test/lib.pyr zTestBase.setUpClass(s  G  **..););< LLCEHEWEW X77??D ggll4T(^0D+EvNCO''///   sC C*)C*)__name__ __module__ __qualname____doc__r classmethodr __classcell__)rs@rrrs#0N 0 0rc4fd}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 tjj}tj|d} |||s*t j tj|SS#t$rJtjjdt|jdjd|dd}wxYw#|s*t j tj|wwxYw)N)prefixFzTest .z failed, output is at  T)tempfilemktemprrmkdirgccollectshutilrmtree Exceptionsysstderrwritetype)selfrkeepfuncs rwrapperz"with_rw_directory..wrapperAsdmm4  $ D$'  d#    5d)<)<(=Qt}}oMcdhckkm!no   d#sA00ACC.C4rr8r9s` rwith_rw_directoryr<=s$,}}G Nr$c4fd}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 afterwardscHtd}t||d||S)Nzpacks/*T) hard_link_ok) fixture_pathcopy_files_globbed)r6r src_pack_globr8s rr9zwith_packs_rw..wrapper_s&$Y/ =$TBD$r$r:r;s` r with_packs_rwrC[s }}G Nr$ctjjtjjtd|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 rr@r@ms( 77<<1:x HHr$c|tj|D]}|rettdrUtjj |tjj |} tj ||jtj||y#t$rtj||YwxYw)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) globhasattrrrrbasenamerHOSErrorr/copy) source_glob target_dirr?src_filetargets rrArAtsIIk* GB/WW\\*bgg.>.>x.HIF 2&) KK* -+  2 Hj1 2s)BB;:B;c|dz}t|}|r t|}tj|t d|}|j 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_bytesr_sH 1$K[!H>x  c8A 99;r$cFdt|z}|jd|zS)z0:return: bytes resembling an uncompressed objectzblob %iascii)lenencode)r5dataodatas r make_objectrfs$ #d) #E << 4 ''r$cFt||}t|t|fS)z`:return: tuple(size_of_stream, stream) :param randomize: try to produce a very random stream)r_rbr)rZr[ds rmake_memory_fileris# =),A q671: r$c$eZdZdZdZdZdZy) DummyStreamc.d|_d|_d|_y)NFr)was_readbytesclosedr6s r__init__zDummyStream.__init__s   r$c d|_||_yNT)rmrn)r6sizes rreadzDummyStream.reads  r$cd|_yrs)rorps rclosezDummyStream.closes  r$c |jsJyN)rmrps r_assertzDummyStream._asserts}}}r$N)rrr rqrurwrzr$rrkrks r$rkceZdZdZdZy) DeriveTestc>|jd|_||_y)Nmyarg)poprargs)r6shar5rtstreamrkwargss rrqzDeriveTest.__init__sZZ(  r$c<|jsJ|jsJyry)rrrps rrzzDeriveTest._assertsyyyzzzr$N)rrr rqrzr{r$rr}r}s r$r}))F)r!gitdbrr2rWriorrIunittestr*r/rr-r functoolsrTestCaserr<rCr@rAr_rfrirkr}r{r$rrs~ $    0x  0F< $I .$ ( $r$