d0dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZmZmZddlmZmZmZmZmZmZmZmZddlmZd Zdd Zd Zd Z d Z!dZ"dZ#dZ$ddZ%ddZ&dZ'dS)z*Utility functions common to Dulwich tests.N)SkipTestskipIf) 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 W/home/feoh/.local/pipx/venvs/poetry/lib/python3.11/site-packages/dulwich/tests/utils.py open_repor&-s~#%%w||BGOOH55tT:wX\]]HGLL400M OHmd;;;;   c|tj|jtj}t j|dS)zTear down a test repository.N)closerrrrstripseprrmtree)repor"s r%tear_down_repor.CsJJJLLLwty//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% TestObjectr1Vs   r'r7 TestObject_idcSNr6)shasr%zmake_object..gscr')r2itemsrr<setattr)clsattrsr7objr!valuer<s @r% make_objectrDJs     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)updaterDr)rA default_time all_attrss r% make_commitrTms\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. irFrrGsTest Tag)taggertag_time tag_timezonerNobjectr!) r9r type_nameinttimemktimedatetime timetuplerQrDr )targetrA target_id target_typerRrSs r%make_tagrds Iv/00Kt{8#4T1a#@#@#J#J#L#LMMNNL4 # * IU s ( (i ( ((r'cfd}|S)z5Generate a test method that tests the given function.c |dSr;r6selffuncmethods r%do_testz!functest_builder..do_teststTr'r6rjrirks`` r%functest_builderrms) Nr'cfd}|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. cxttjstdz|dS)Nz%s extension not found) isinstancetypesBuiltinFunctionTyperrgs r%rkz%ext_functest_builder..do_testsD$ 9:: <3d:;; ;tTr'r6rls`` 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)rlenrwrite enumerater rrpr\get_rawtellr listrr rrangeappend write_shaseek)f objects_specstoresf num_objects full_objectsoffsetscrc32sitype_numdatabase base_type_num_rBoffset 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| z| d} ~ wwxYwg} ||gD]\} t | dkr| \} }| |t f} | \} }}| | |j|f||]t|| }d|z 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|] }| Sr6r6).0pnnumss r% z&build_commit_graph..8s888r$r(888r'rVzUnknown parent %irz Commit %iascii)rNrOrPrLrLdr6) KeyErrorargs ValueErrorgetrvFr}r9 add_objectrencoderQrT) object_store commit_spectreesrArLcommitscommit commit_num parent_idsexcmissing_parentblobsentryrblobmodetree_id commit_attrs commit_objrs @r%build_commit_graphrs> } }K DG # #AY  L8888VABBZ888JJ L L L # ^0>ABB K LYYz2.. * *E5zzQ" dtQ$ D$ LL$. / / /  # #D ) ) ) )lE22$j088AA!&   EIIj"55666 00<00 #=1C7 %=Z +++z"""" Ns5 A AA cVgtjfd}|t_fd}|fS)z:Wrap warnings.showwarning with code that records warnings.c>|ddS)Nr)r})rkwargscaught_warningss r%custom_showwarningz1setup_warning_catcher..custom_showwarning`s!tAw'''''r'c t_dSr;warnings showwarning)original_showwarningsr%restore_showwarningz2setup_warning_catcher..restore_showwarninges3r'r)rrrrs @@r%setup_warning_catcherrZs^O#/(((((.H44444 / //r'r;)NN)(r5r_rrrr]rqr dulwich.testsrrindexrobjectsrrr r packr r r rrrrrr-rrr&r.rDrTrdrmrsrrrr6r'r%rs*10  ********999999999999'''''''''''''''''''' ,   F,,,.))).