-Santos Gallegos <stsewd _at_ proton.me>

-Wenhan Zhu <wzhu.cosmos _at_ gmail.com>

-Eliah Kagan <eliah.kagan _at_ gmail.com>

-Ethan Lin <et.repositories _at_ gmail.com>

Portions derived from other open source works and are clearly marked.

__all__= ("TreeModifier","Tree")

defgit_cmp(t1:TreeCacheTup,t2:TreeCacheTup)->int:

a,b=t1[2],t2[2]

len_a,len_b=len(a),len(b)

min_len=min(len_a,len_b)

min_cmp=cmp(a[:min_len],b[:min_len])

ifmin_cmp:

defmerge_sort(a:List[TreeCacheTup],cmp:Callable[[TreeCacheTup,TreeCacheTup],int])->None:

iflen(a)<2:

mid=len(a)//2

lefthalf=a[:mid]

righthalf=a[mid:]

merge_sort(lefthalf,cmp)

merge_sort(righthalf,cmp)

whilei<len(lefthalf)andj<len(righthalf):

ifcmp(lefthalf[i],righthalf[j])<=0:

a[k]=lefthalf[i]

else:

a[k]=righthalf[j]

whilei<len(lefthalf):

a[k]=lefthalf[i]

whilej<len(righthalf):

a[k]=righthalf[j]

classTreeModifier:

merge_sort(self._cache,git_cmp)

self._cache.sort(key=lambdax: (x[2]+"/")ifx[1]==Tree.tree_id<<12elsex[2])

fromgit.objectsimportTree,Blob

fromtest.libimportTestBase

importos.pathasosp

classTestTree(TestBase):

testtree._deserialize(stream)

deftest_tree_modifier_ordering(self):

defsetup_git_repository_and_get_ordered_files():

os.mkdir("tmp")

os.chdir("tmp")

subprocess.run(["git","init","-q"],check=True)

os.mkdir("file")

forfilenamein [

"bin",

"bin.d",

"file.to",

"file.toml",

"file.toml.bin",

"file0",

"file/a",

]:

open(filename,"a").close()

subprocess.run(["git","add","."],check=True)

subprocess.run(["git","commit","-m","c1"],check=True)

tree_hash=subprocess.check_output(["git","rev-parse","HEAD^{tree}"]).decode().strip()

cat_file_output=subprocess.check_output(["git","cat-file","-p",tree_hash]).decode()

return [line.split()[-1]forlineincat_file_output.split("\n")ifline]

roottree=self.rorepo.tree(hexsha)

blob_mode=Tree.blob_id<<12

tree_mode=Tree.tree_id<<12

files_in_desired_order= [

(blob_mode,"bin"),

(blob_mode,"bin.d"),

(blob_mode,"file.to"),

(blob_mode,"file.toml"),

(blob_mode,"file.toml.bin"),

(blob_mode,"file0"),

(tree_mode,"file"),

]

mod=roottree.cache

forfile_mode,file_nameinfiles_in_desired_order:

mod.add(hexsha,file_mode,file_name)

deffile_names_in_order():

return [t[1]fortinfiles_in_desired_order]

defnames_in_mod_cache():

a= [t[2]fortinmod._cache]

here=file_names_in_order()

return [eforeinaifeinhere]

git_file_names_in_order=setup_git_repository_and_get_ordered_files()

os.chdir("..")

mod.set_done()

assertnames_in_mod_cache()==git_file_names_in_order,"set_done() performs git-sorting"

deftest_traverse(self):

root=self.rorepo.tree("0.1.6")