Added test for aggressive_tree_merge

3 files changed, 69 insertions, 65 deletions

diff --git a/lib/git/index/fun.py b/lib/git/index/fun.py
index 79fcfddb..b04d018f 100644
--- a/lib/git/index/fun.py
+++ b/lib/git/index/fun.py
@@ -218,71 +218,72 @@ def aggressive_tree_merge(odb, tree_shas):
 		for entry in traverse_tree_recursive(odb, tree_shas[-1], ''):
 			out_append(_tree_entry_to_baseindexentry(entry, 0))
 		# END for each entry
-	elif len(tree_shas) == 3:
-		for base, ours, theirs in traverse_trees_recursive(odb, tree_shas, ''):
-			if base is not None:
-				# base version exists
-				if ours is not None:
-					# ours exists
-					if theirs is not None:
-						# it exists in all branches, if it was changed in both
-						# its a conflict, otherwise we take the changed version
-						# This should be the most common branch, so it comes first
-						if( base[0] != ours[0] and base[0] != theirs[0] and ours[0] != theirs[0] ) or \
-							( base[1] != ours[1] and base[1] != theirs[1] and ourse[1] != theirs[1] ):
-							# changed by both
-							out_append(_tree_entry_to_baseindexentry(base, 1))
-							out_append(_tree_entry_to_baseindexentry(ours, 2))
-							out_append(_tree_entry_to_baseindexentry(theirs, 3))
-						elif base[0] != ours[0] or base[1] != ours[1]:
-							# only we changed it
-							out_append(_tree_entry_to_baseindexentry(ours, 0))
-						else:
-							# either nobody changed it, or they did. In either
-							# case, use theirs
-							out_append(_tree_entry_to_baseindexentry(theirs, 0))
-						# END handle modification 
+		return out
+	# END handle single tree 
+	
+	if len(tree_shas) > 3:
+		raise ValueError("Cannot handle %i trees at once" % len(tree_shas))
+
+	# three trees
+	for base, ours, theirs in traverse_trees_recursive(odb, tree_shas, ''):
+		if base is not None:
+			# base version exists
+			if ours is not None:
+				# ours exists
+				if theirs is not None:
+					# it exists in all branches, if it was changed in both
+					# its a conflict, otherwise we take the changed version
+					# This should be the most common branch, so it comes first
+					if( base[0] != ours[0] and base[0] != theirs[0] and ours[0] != theirs[0] ) or \
+						( base[1] != ours[1] and base[1] != theirs[1] and ours[1] != theirs[1] ):
+						# changed by both
+						out_append(_tree_entry_to_baseindexentry(base, 1))
+						out_append(_tree_entry_to_baseindexentry(ours, 2))
+						out_append(_tree_entry_to_baseindexentry(theirs, 3))
+					elif base[0] != ours[0] or base[1] != ours[1]:
+						# only we changed it
+						out_append(_tree_entry_to_baseindexentry(ours, 0))
 					else:
-						
-						if ours[0] != base[0] or ours[1] != base[1]:
-							# they deleted it, we changed it, conflict 
-							out_append(_tree_entry_to_baseindexentry(base, 1))
-							out_append(_tree_entry_to_baseindexentry(ours, 2))
-							out_append(_tree_entry_to_baseindexentry(theirs, 3))
-						# else:
-						#	we didn't change it, ignore
-						#	pass
-						# END handle our change
-					# END handle theirs
+						# either nobody changed it, or they did. In either
+						# case, use theirs
+						out_append(_tree_entry_to_baseindexentry(theirs, 0))
+					# END handle modification 
 				else:
-					if theirs is None:
-						# deleted in both, its fine - its out
-						pass
-					else:
-						if theirs[0] != base[0] or theirs[1] != base[1]:
-							# deleted in ours, changed theirs, conflict
-							out_append(_tree_entry_to_baseindexentry(base, 1))
-							out_append(_tree_entry_to_baseindexentry(ours, 2))
-							out_append(_tree_entry_to_baseindexentry(theirs, 3))
-						# END theirs changed
-						#else:
-						# 	theirs didnt change
-						#	pass
-					# END handle theirs
-				# END handle ours
+					
+					if ours[0] != base[0] or ours[1] != base[1]:
+						# they deleted it, we changed it, conflict 
+						out_append(_tree_entry_to_baseindexentry(base, 1))
+						out_append(_tree_entry_to_baseindexentry(ours, 2))
+					# else:
+					#	we didn't change it, ignore
+					#	pass
+					# END handle our change
+				# END handle theirs
 			else:
-				# all three can't be None
-				if ours is None:
-					# added in their branch
-					out_append(_tree_entry_to_baseindexentry(theirs, 0))
-				elif theirs is None:
-					# added in our branch
-					out_append(_tree_entry_to_baseindexentry(ours, 0))
-				# END hanle heads
-			# END handle base exists
-		# END for each entries tuple
-	else:
-		raise ValueError("Cannot handle %i trees at once" % len(tree_shas))
-	# END handle tree shas
-	
+				if theirs is None:
+					# deleted in both, its fine - its out
+					pass
+				else:
+					if theirs[0] != base[0] or theirs[1] != base[1]:
+						# deleted in ours, changed theirs, conflict
+						out_append(_tree_entry_to_baseindexentry(base, 1))
+						out_append(_tree_entry_to_baseindexentry(theirs, 3))
+					# END theirs changed
+					#else:
+					# 	theirs didnt change
+					#	pass
+				# END handle theirs
+			# END handle ours
+		else:
+			# all three can't be None
+			if ours is None:
+				# added in their branch
+				out_append(_tree_entry_to_baseindexentry(theirs, 0))
+			elif theirs is None:
+				# added in our branch
+				out_append(_tree_entry_to_baseindexentry(ours, 0))
+			# END hanle heads
+		# END handle base exists
+	# END for each entries tuple
+
 	return out
diff --git a/lib/git/index/typ.py b/lib/git/index/typ.py
index 6ef1d2f2..d9cafa2e 100644
--- a/lib/git/index/typ.py
+++ b/lib/git/index/typ.py
@@ -56,6 +56,9 @@ class BaseIndexEntry(tuple):
 
 	def __str__(self):
 		return "%o %s %i\t%s" % (self.mode, self.hexsha, self.stage, self.path)
+		
+	def __repr__(self):
+		return "(%o, %s, %i, %s)" % (self.mode, self.hexsha, self.stage, self.path)
 
 	@property
 	def mode(self):
diff --git a/lib/git/repo.py b/lib/git/repo.py
index f97126ea..79275106 100644
--- a/lib/git/repo.py
+++ b/lib/git/repo.py
@@ -742,7 +742,7 @@ class Repo(object):
 		# we at least give a proper error instead of letting git fail
 		prev_cwd = None
 		prev_path = None
-		odbt = kwargs.pop('odbt', GitCmdObjectDB)
+		odbt = kwargs.pop('odbt', type(self.odb))
 		if os.name == 'nt':
 			if '~' in path:
 				raise OSError("Git cannot handle the ~ character in path %r correctly" % path)