From c1fcdb0e00f9163bc3f60069182f231afb83523e Mon Sep 17 00:00:00 2001
From: Russ Cox <rsc@golang.org>
Date: Wed, 6 Aug 2014 15:25:17 -0400
Subject: codereview: handle upload of merge

LGTM=minux
R=minux
CC=golang-codereviews
https://golang.org/cl/118690043
---
 lib/codereview/codereview.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

(limited to 'lib')

diff --git a/lib/codereview/codereview.py b/lib/codereview/codereview.py
index 1b45518890..a64401533c 100644
--- a/lib/codereview/codereview.py
+++ b/lib/codereview/codereview.py
@@ -3478,11 +3478,23 @@ class MercurialVCS(VersionControlSystem):
 		if not err and mqparent != "":
 			self.base_rev = mqparent
 		else:
-			out = RunShell(["hg", "parents", "-q"], silent_ok=True).strip()
+			out = RunShell(["hg", "parents", "-q", "--template={node} {branch}"], silent_ok=True).strip()
 			if not out:
 				# No revisions; use 0 to mean a repository with nothing.
-				out = "0:0"
-			self.base_rev = out.split(':')[1].strip()
+				out = "0:0 default"
+			
+			# Find parent along current branch.
+			branch = repo[None].branch()
+			base = ""
+			for line in out.splitlines():
+				fields = line.strip().split(' ')
+				if fields[1] == branch:
+					base = fields[0]
+					break
+			if base == "":
+				# Use the first parent
+				base = out.strip().split(' ')[0]
+			self.base_rev = base
 
 	def _GetRelPath(self, filename):
 		"""Get relative path of a file according to the current directory,
-- 
cgit v1.2.3-54-g00ecf