Make gclient_scm_test much more silent

scm.py

 # Copyright (c) 2006-2009 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """SCM-specific utility classes."""
 
 import cStringIO
 import glob
 import os
 import re
@@ skipping 48 matching lines @@
   for line in file_content:
     data.write('+')
     data.write(line)
   result = data.getvalue()
   data.close()
   return result
 
 
 class GIT(object):
   @staticmethod
-  def Capture(args, in_directory=None, print_error=True, error_ok=False):
-    """Runs git, capturing output sent to stdout as a string.
-
-    Args:
-      args: A sequence of command line parameters to be passed to git.
-      in_directory: The directory where git is to be run.
-
-    Returns:
-      The output sent to stdout as a string.
-    """
-    try:
-      return gclient_utils.CheckCall(['git'] + args, in_directory, print_error)
-    except gclient_utils.CheckCallError:
-      if error_ok:
-        return ('', '')
-      raise
+  def Capture(args, **kwargs):
+    return gclient_utils.CheckCall(['git'] + args, print_error=False,
+        **kwargs)[0]
 
   @staticmethod
   def CaptureStatus(files, upstream_branch=None):
     """Returns git status.
 
     @files can be a string (one file) or a list of files.
 
     Returns an array of (status, file) tuples."""
     if upstream_branch is None:
       upstream_branch = GIT.GetUpstreamBranch(os.getcwd())
       if upstream_branch is None:
-        raise Exception("Cannot determine upstream branch")
-    command = ["diff", "--name-status", "-r", "%s..." % upstream_branch]
+        raise gclient_utils.Error('Cannot determine upstream branch')
+    command = ['diff', '--name-status', '-r', '%s...' % upstream_branch]
     if not files:
       pass
     elif isinstance(files, basestring):
       command.append(files)
     else:
       command.extend(files)
-
-    status = GIT.Capture(command)[0].rstrip()
+    status = GIT.Capture(command).rstrip()
     results = []
     if status:
-      for statusline in status.split('\n'):
+      for statusline in status.splitlines():
         m = re.match('^(\w)\t(.+)$', statusline)
         if not m:
-          raise Exception("status currently unsupported: %s" % statusline)
+          raise gclient_utils.Error(
+              'status currently unsupported: %s' % statusline)
         results.append(('%s      ' % m.group(1), m.group(2)))
     return results
 
   @staticmethod
-  def GetEmail(repo_root):
+  def GetEmail(cwd):
     """Retrieves the user email address if known."""
     # We could want to look at the svn cred when it has a svn remote but it
     # should be fine for now, users should simply configure their git settings.
-    return GIT.Capture(['config', 'user.email'],
-                       repo_root, error_ok=True)[0].strip()
+    try:
+      return GIT.Capture(['config', 'user.email'], cwd=cwd).strip()
+    except gclient_utils.CheckCallError:
+      return ''
 
   @staticmethod
   def ShortBranchName(branch):
     """Converts a name like 'refs/heads/foo' to just 'foo'."""
     return branch.replace('refs/heads/', '')
 
   @staticmethod
   def GetBranchRef(cwd):
     """Returns the full branch reference, e.g. 'refs/heads/master'."""
-    return GIT.Capture(['symbolic-ref', 'HEAD'], cwd)[0].strip()
+    return GIT.Capture(['symbolic-ref', 'HEAD'], cwd=cwd).strip()
 
   @staticmethod
   def GetBranch(cwd):
     """Returns the short branch name, e.g. 'master'."""
     return GIT.ShortBranchName(GIT.GetBranchRef(cwd))
 
   @staticmethod
   def IsGitSvn(cwd):
     """Returns true if this repo looks like it's using git-svn."""
     # If you have any "svn-remote.*" config keys, we think you're using svn.
     try:
-      GIT.Capture(['config', '--get-regexp', r'^svn-remote\.'], cwd)
+      GIT.Capture(['config', '--get-regexp', r'^svn-remote\.'], cwd=cwd)
       return True
     except gclient_utils.CheckCallError:
       return False
 
   @staticmethod
   def GetSVNBranch(cwd):
     """Returns the svn branch name if found."""
     # Try to figure out which remote branch we're based on.
     # Strategy:
     # 1) find all git-svn branches and note their svn URLs.
     # 2) iterate through our branch history and match up the URLs.
 
     # regexp matching the git-svn line that contains the URL.
     git_svn_re = re.compile(r'^\s*git-svn-id: (\S+)@', re.MULTILINE)
 
     # Get the refname and svn url for all refs/remotes/*.
     remotes = GIT.Capture(
         ['for-each-ref', '--format=%(refname)', 'refs/remotes'],
-        cwd)[0].splitlines()
+        cwd=cwd).splitlines()
     svn_refs = {}
     for ref in remotes:
       match = git_svn_re.search(
-          GIT.Capture(['cat-file', '-p', ref], cwd)[0])
+          GIT.Capture(['cat-file', '-p', ref], cwd=cwd))
       # Prefer origin/HEAD over all others.
       if match and (match.group(1) not in svn_refs or
                     ref == "refs/remotes/origin/HEAD"):
         svn_refs[match.group(1)] = ref
 
     svn_branch = ''
     if len(svn_refs) == 1:
       # Only one svn branch exists -- seems like a good candidate.
       svn_branch = svn_refs.values()[0]
     elif len(svn_refs) > 1:
@@ skipping 14 matching lines @@
     return svn_branch
 
   @staticmethod
   def FetchUpstreamTuple(cwd):
     """Returns a tuple containg remote and remote ref,
        e.g. 'origin', 'refs/heads/master'
        Tries to be intelligent and understand git-svn.
     """
     remote = '.'
     branch = GIT.GetBranch(cwd)
-    upstream_branch = None
-    upstream_branch = GIT.Capture(
-        ['config', 'branch.%s.merge' % branch], in_directory=cwd,
-        error_ok=True)[0].strip()
+    try:
+      upstream_branch = GIT.Capture(
+          ['config', 'branch.%s.merge' % branch], cwd=cwd).strip()
+    except gclient_utils.Error:
+      upstream_branch = None
     if upstream_branch:
-      remote = GIT.Capture(
-          ['config', 'branch.%s.remote' % branch],
-          in_directory=cwd, error_ok=True)[0].strip()
+      try:
+        remote = GIT.Capture(
+            ['config', 'branch.%s.remote' % branch], cwd=cwd).strip()
+      except gclient_utils.Error:
+        pass
     else:
       # Fall back on trying a git-svn upstream branch.
       if GIT.IsGitSvn(cwd):
         upstream_branch = GIT.GetSVNBranch(cwd)
       else:
         # Else, try to guess the origin remote.
-        remote_branches = GIT.Capture(
-            ['branch', '-r'], in_directory=cwd)[0].split()
+        remote_branches = GIT.Capture(['branch', '-r'], cwd=cwd).split()
         if 'origin/master' in remote_branches:
           # Fall back on origin/master if it exits.
           remote = 'origin'
           upstream_branch = 'refs/heads/master'
         elif 'origin/trunk' in remote_branches:
           # Fall back on origin/trunk if it exists. Generally a shared
           # git-svn clone
           remote = 'origin'
           upstream_branch = 'refs/heads/trunk'
         else:
@@ skipping 20 matching lines @@
     if not branch:
       branch = GIT.GetUpstreamBranch(cwd)
     command = ['diff', '-p', '--no-prefix', '--no-ext-diff',
                branch + "..." + branch_head]
     if not full_move:
       command.append('-C')
     # TODO(maruel): --binary support.
     if files:
       command.append('--')
       command.extend(files)
-    diff = GIT.Capture(command, cwd)[0].splitlines(True)
+    diff = GIT.Capture(command, cwd=cwd).splitlines(True)
     for i in range(len(diff)):
       # In the case of added files, replace /dev/null with the path to the
       # file being added.
       if diff[i].startswith('--- /dev/null'):
         diff[i] = '--- %s' % diff[i+1][4:]
     return ''.join(diff)
 
   @staticmethod
   def GetDifferentFiles(cwd, branch=None, branch_head='HEAD'):
     """Returns the list of modified files between two branches."""
     if not branch:
       branch = GIT.GetUpstreamBranch(cwd)
     command = ['diff', '--name-only', branch + "..." + branch_head]
-    return GIT.Capture(command, cwd)[0].splitlines(False)
+    return GIT.Capture(command, cwd=cwd).splitlines(False)
 
   @staticmethod
   def GetPatchName(cwd):
     """Constructs a name for this patch."""
-    short_sha = GIT.Capture(['rev-parse', '--short=4', 'HEAD'], cwd)[0].strip()
+    short_sha = GIT.Capture(['rev-parse', '--short=4', 'HEAD'], cwd=cwd).strip()
     return "%s#%s" % (GIT.GetBranch(cwd), short_sha)
 
   @staticmethod
-  def GetCheckoutRoot(path):
+  def GetCheckoutRoot(cwd):
     """Returns the top level directory of a git checkout as an absolute path.
     """
-    root = GIT.Capture(['rev-parse', '--show-cdup'], path)[0].strip()
-    return os.path.abspath(os.path.join(path, root))
+    root = GIT.Capture(['rev-parse', '--show-cdup'], cwd=cwd).strip()
+    return os.path.abspath(os.path.join(cwd, root))
 
   @staticmethod
   def AssertVersion(min_version):
     """Asserts git's version is at least min_version."""
     def only_int(val):
       if val.isdigit():
         return int(val)
       else:
         return 0
-    current_version =  GIT.Capture(['--version'])[0].split()[-1]
+    current_version =  GIT.Capture(['--version']).split()[-1]
     current_version_list = map(only_int, current_version.split('.'))
     for min_ver in map(int, min_version.split('.')):
       ver = current_version_list.pop(0)
       if ver < min_ver:
         return (False, current_version)
       elif ver > min_ver:
         return (True, current_version)
     return (True, current_version)
 
 
@@ skipping 517 matching lines @@
     if not SVN.current_version:
       SVN.current_version = SVN.Capture(['--version']).split()[2]
     current_version_list = map(only_int, SVN.current_version.split('.'))
     for min_ver in map(int, min_version.split('.')):
       ver = current_version_list.pop(0)
       if ver < min_ver:
         return (False, SVN.current_version)
       elif ver > min_ver:
         return (True, SVN.current_version)
     return (True, SVN.current_version)