o U h0@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZddlmZmZmZmZmZmZmZmZddlmZdZdd d Zdd d ZddZddZ ddZ!ddZ"ddZ#dddZ$d ddZ%ddZ&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)RepoicCsT|durt}tjtjtdddd|}tj||}tj||ddt |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_dirr!g/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/dulwich/tests/utils.py open_repo3sr#returncCs,|tj|jtj}t|dS)zTear down a test repository.N)closerrrrstripseprrmtree)reporr!r!r"tear_down_repoKsr*c sdGddd|}d|j|_|}|D]\}}|dkr)t|fdd|_qt|||q|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. c@seZdZdZdS)zmake_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!r!r" TestObject^sr/ TestObject_idcsSNr!r!shar!r"mszmake_object..)r+itemsrr4setattr)clsattrsr/objrvaluer!r3r" make_objectRs r<c Ks6d}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)updater<r)r9 default_time all_attrsr!r!r" make_commitss rKcKs\|j}t|j}tttddd}d|dd||fdd}||t t fi|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. ir=rr>sTest Tag)taggertag_time tag_timezonerEobjectr) r1r type_nameinttimemktimedatetime timetuplerHr<r)targetr9 target_id target_typerIrJr!r!r"make_tags  rZcdfdd }|S)z5Generate a test method that tests the given function.r$Ncs|dSr2r!selffuncmethodr!r"do_testsz!functest_builder..do_testr$Nr!r`r_rar!r^r"functest_buildersrdcr[)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$Ncs(ttjs td|dS)Nz extension not found) isinstancetypesBuiltinFunctionTyperr\r^r!r"ras z%ext_functest_builder..do_testrbr!rcr!r^r"ext_functest_buildersrhc Cst|}t|}t|j|i}i}i}t||krct|D]>\}\} } | tvr4| | t| | gf||<q| \} } t| trJ| |vrBq|| \} } } n| | \} } | | t| | gf||<qt||kst|D]b\}\} }| }| t kr|\}} |||} ||\} }} | t t || f}n*| tkr|\}} t|tr||\} }} n | |\} }t| |} | t t || f}t|j| |}|||<|||<qgg}t|D]}||\} } }t|dksJ|||| | |||fq||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 enumeraterr rerRget_rawtellr listr r rrangeappend write_shaseek)f objects_specstoresf num_objects full_objectsoffsetscrc32sitype_numdatabase base_type_num_r:offset base_index base_database_refcrc32expectedr4r!r!r" build_packs^           rc sR|duri}|dur i}d}ig}|D]}|d}zfdd|ddD}WntyA} z | j\} td| | d} ~ wwg} ||gD]$} t| dkr[| \} }| |tf} | \} }}| | |j|f||qJt || }d| d |||d }| ||it d i|}|d d }|j|<||||q|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. Nrcsg|]}|qSr!r!).0pnnumsr!r" ?sz&build_commit_graph..rLzUnknown parent zCommit ascii)rErFrGrCrCdr!) KeyErrorargs ValueErrorgetrjFrqr1 add_objectrencoderHrK) object_store commit_spectreesr9rCcommitscommit commit_num parent_idsexcmissing_parentblobsentryrblobmodetree_id commit_attrs commit_objr!rr"build_commit_graphsJ          rcs4gtjdfdd }|t_dfdd }|fS)z:Wrap warnings.showwarning with code that records warnings.r$Ncs|ddS)Nr)rq)rkwargs)caught_warningsr!r"custom_showwarningfsz1setup_warning_catcher..custom_showwarningcs t_dSr2warnings showwarningr!)original_showwarningr!r"restore_showwarningks z2setup_warning_catcher..restore_showwarningrbr)rrr!)rrr"setup_warning_catcheras rr2rb)NN)'r.rUrrrrSrfrunittestr dulwich.indexrdulwich.objectsrrrr dulwich.packrr r r r r rr dulwich.reporrr#r*r<rKrZrdrhrrrr!r!r!r"s0  (   !  L M