.Ph0dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZddlmZmZmZmZmZmZmZmZddlmZdZdd Zdd Zd Zd Z dZ!dZ"dZ#ddZ$ddZ%dZ&dS)z*Utility functions common to Dulwich tests.N)SkipTest) commit_tree)CommitFixedShaTag object_class) DELTA_TYPES OFS_DELTA REF_DELTA SHA1Writer create_deltaobj_shawrite_pack_headerwrite_pack_object)RepoicD|tj}tjtjt dddd|}tj||}tj||dt|S)a"Open a copy of a repo in a temporary directory. Use this function for accessing repos in dulwich/tests/data/repos to avoid accidentally or intentionally modifying those repos in place. Use tear_down_repo to delete any temp files created. Args: name: The name of the repository, relative to dulwich/tests/data/repos temp_dir: temporary directory to initialize to. If not provided, a temporary directory will be created. Returns: An initialized Repo object that lives in a temporary directory. Nz..testdatareposT)symlinks) tempfilemkdtempospathjoindirname__file__shutilcopytreer)nametemp_dirrepo_dir temp_repo_dirs S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/dulwich/tests/utils.py open_repor$3s#%%w|| !!4z7DHGLL400M OHmd;;;;   returnc|tj|jtj}t j|dS)zTear down a test repository.N)closerrrrstripseprrmtree)repor s r#tear_down_repor-KsJJJLLLwty//7788H M(r%c Gdd|}d|jz|_|}|D]6\}}|dkrt|fd|_%t |||7|S)aLMake an object for testing and assign some members. This method creates a new subclass to allow arbitrary attribute reassignment, which is not otherwise possible with objects having __slots__. Args: attrs: dict of attributes to set on the new object. Returns: A newly initialized object of type cls. ceZdZdZdS)make_object..TestObjecta Class that inherits from the given class, but without __slots__. Note that classes with __slots__ can't have arbitrary attributes monkey-patched in, so this is a class that is exactly the same only with a __dict__ instead of __slots__. N)__name__ __module__ __qualname____doc__r%r# TestObjectr0^s    r%r6 TestObject_idcSNr5)shasr#zmake_object..mscr%)r1itemsrr;setattr)clsattrsr6objrvaluer;s @r# make_objectrCRs     S   (#,6J *,,C{{}}&& e 4<<5//C!kkkCGG Cu % % % % Jr%c ld}d|dd|ddgdd }||ttfi|S)zMake a Commit object with a default set of members. Args: attrs: dict of attributes to overwrite from the default values. Returns: A newly initialized Commit object. i;=KTest Author rs"Test Committer Test message.s(0000000000000000000000000000000000000000) author author_timeauthor_timezone committer commit_timecommit_timezonemessageparentstree)updaterCr)r@ default_time all_attrss r# make_commitrSss\L4#:##  IU v + + + ++r%c *|j}t|j}tt jt jddd}d|dd||fdd}||ttfi|S)zMake a Tag object with a default set of values. Args: target: object to be tagged (Commit, Blob, Tree, etc) attrs: dict of attributes to overwrite from the default values. Returns: A newly initialized Tag object. irErrFsTest Tag)taggertag_time tag_timezonerMobjectr) r8r type_nameinttimemktimedatetime timetuplerPrCr)targetr@ target_id target_typerQrRs r#make_tagrcs Iv/00Kt{8#4T1a#@#@#J#J#L#LMMNNL4 # * IU s ( (i ( ((r%cdfd }|S)z5Generate a test method that tests the given function.r&Nc |dSr:r5selffuncmethods r#do_testz!functest_builder..do_teststTr%r&Nr5rirhrjs`` r#functest_builderrms. Nr%cdfd }|S)aGenerate a test method that tests the given extension function. This is intended to generate test methods that test both a pure-Python version and an extension version using common test code. The extension test will raise SkipTest if the extension is not found. Sample usage: class MyTest(TestCase); def _do_some_test(self, func_impl): self.assertEqual('foo', func_impl()) test_foo = functest_builder(_do_some_test, foo_py) test_foo_extension = ext_functest_builder(_do_some_test, _foo_c) Args: method: The method to run. It must must two parameters, self and the function implementation to test. func: The function implementation to pass to method. r&Ncxttjstd|dS)Nz extension not found) isinstancetypesBuiltinFunctionTyperrfs r#rjz%ext_functest_builder..do_testsG$ 9:: :d88899 9tTr%rkr5rls`` r#ext_functest_builderrss., Nr%c  t|}t|}t|j|i}i}i}t||krt |D]\}\} } | t vr| | t | | gf||<)| \} } t| tr| |vrH|| \} } } n| | \} } | | t | | gf||<t||kt |D]\}\} }| }| tkr<|\}} |||z } ||\} }} | tt|| f}ny| tkrn|\}} t|tr ||\} }} n(| |\} }t | |} | tt|| f}t|j| |}|||<|||<g}t!|D]I}||\} } }t|dksJ|||| | |||fJ||d|S)aFWrite test pack data from a concise spec. Args: f: A file-like object to write the pack to. objects_spec: A list of (type_num, obj). For non-delta types, obj is the string of that object's data. For delta types, obj is a tuple of (base, data), where: * base can be either an index in objects_spec of the base for that * delta; or for a ref delta, a SHA, in which case the resulting pack * will be thin and the base will be an external ref. * data is a string of the full, non-deltified data for that object. Note that offsets/refs and deltas are computed within this function. store: An optional ObjectStore for looking up external refs. Returns: A list of tuples in the order specified by objects_spec: (offset, type num, data, sha, CRC32) r)r lenrwrite enumerater rrpr[get_rawtellr listr r rrangeappend write_shaseek)f objects_specstoresf num_objects full_objectsoffsetscrc32sitype_numdatabase base_type_num_rAoffset base_index base_database_refcrc32expectedr;s r# build_packrs& ABl##Kbh ,,,LG F l  k ) )#,\#:#:   A${**#+T78dV3L3L"M QJD$$$$ 7|++&24&8# q!!#(==#6#6 q v..LOO l  k ) )$( 55?Hc y " JGJ//D*:6OAy!l9d;;<<=CC  " " NHd(C(( 9%1(%;"9dd+0==+B+B( y}i88l9d;;<<=C!"(Hc:: q H ;  FF*1o$3xx2~~~~XtS&)DEEEELLNNNFF1III Or%c|i}|i}d}ig}|D]g}|d} fd|ddD}n.#t$r!} | j\} td| | d} ~ wwxYwg} ||gD]\} t | dkr| \} }| |t f} | \} }}| | |j|f||]t|| }d| d|||d }| ||itd i|}|d d z}|j|<||||i|S) aBuild a commit graph from a concise specification. Sample usage: >>> c1, c2, c3 = build_commit_graph(store, [[1], [2, 1], [3, 1, 2]]) >>> store[store[c3].parents[0]] == c1 True >>> store[store[c3].parents[1]] == c2 True If not otherwise specified, commits will refer to the empty tree and have commit times increasing in the same order as the commit spec. Args: object_store: An ObjectStore to commit objects to. commit_spec: An iterable of iterables of ints defining the commit graph. Each entry defines one commit, and entries must be in topological order. The first element of each entry is a commit number, and the remaining elements are its parents. The commit numbers are only meaningful for the call to make_commits; since real commit objects are created, they will get created with real, opaque SHAs. trees: An optional dict of commit number -> tree spec for building trees for commits. The tree spec is an iterable of (path, blob, mode) or (path, blob) entries; if mode is omitted, it defaults to the normal file mode (0100644). attrs: A dict of commit number -> (dict of attribute -> value) for assigning additional values to the commits. Returns: The list of commit objects created. Raises: ValueError: If an undefined commit identifier is listed as a parent. Nrc g|] }| Sr5r5).0pnnumss r# z&build_commit_graph..?s888r$r(888r%rUzUnknown parent zCommit ascii)rMrNrOrKrKdr5) KeyErrorargs ValueErrorgetrvFr}r8 add_objectrencoderPrS) object_store commit_spectreesr@rKcommitscommit commit_num parent_idsexcmissing_parentblobsentryrblobmodetree_id commit_attrs commit_objrs @r#build_commit_graphrs@ } }K DG # #AY  J8888VABBZ888JJ J J J # ^?~??@@c I JYYz2.. * *E5zzQ" dtQ$ D$ LL$. / / /  # #D ) ) ) )lE22/*..66w??!&   EIIj"55666 00<00 #=1C7 %=Z +++z"""" Ns5 A AA cZgtjdfd }|t_dfd }|fS)z:Wrap warnings.showwarning with code that records warnings.r&Nc>|ddS)Nr)r})rkwargscaught_warningss r#custom_showwarningz1setup_warning_catcher..custom_showwarningfs!tAw'''''r%c t_dSr:warnings showwarning)original_showwarningsr#restore_showwarningz2setup_warning_catcher..restore_showwarningks3r%rkr)rrrrs @@r#setup_warning_catcherrasfO#/((((((.H444444 / //r%r:rk)NN)'r4r^rrrr\rqrunittestr dulwich.indexrdulwich.objectsrrrr dulwich.packr r r r r rrr dulwich.reporrr$r-rCrSrcrmrsrrrr5r%r#rs,10  %%%%%%????????????                     0B,,,.))).