Support for --depth flag to repo init.

This allows checking out a shallow clone with reduced history.
buildbot · Apr 14, 2014 · b1b07eb · b1b07eb
1 parent d3364bf
commit b1b07eb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/master/buildbot/steps/source/repo.py b/master/buildbot/steps/source/repo.py
@@ -94,7 +94,7 @@ class Repo(Source):
     name = 'repo'
     renderables = ["manifestURL", "manifestBranch", "manifestFile", "tarball", "jobs",
                    "syncAllBranches", "updateTarballAge", "manifestOverrideUrl",
-                   "repoDownloads"]
+                   "repoDownloads", "repoDepth"]
 
     ref_not_found_re = re.compile(r"fatal: Couldn't find remote ref")
     cherry_pick_error_re = re.compile(r"|".join([r"Automatic cherry-pick failed",
@@ -116,6 +116,7 @@ def __init__(self,
                  updateTarballAge=7 * 24.0 * 3600.0,
                  manifestOverrideUrl=None,
                  repoDownloads=None,
+                 repoDepth=0,
                  **kwargs):
         """
         @type  manifestURL: string
@@ -143,6 +144,9 @@ def __init__(self,
         @type repoDownloads: list of strings
         @param repoDownloads: optional repo download to perform after the repo sync
 
+        @type repoDepth: integer
+        @param repoDepth: optional depth parameter to repo init.
+                          If specified, create a shallow clone with given depth.
         """
         self.manifestURL = manifestURL
         self.manifestBranch = manifestBranch
@@ -155,6 +159,7 @@ def __init__(self,
         if repoDownloads is None:
             repoDownloads = []
         self.repoDownloads = repoDownloads
+        self.repoDepth = repoDepth
         Source.__init__(self, **kwargs)
 
         assert self.manifestURL is not None
@@ -273,7 +278,8 @@ def doRepoSync(self, forceClobber=False):
         yield self._repoCmd(['init',
                              '-u', self.manifestURL,
                              '-b', self.manifestBranch,
-                             '-m', self.manifestFile])
+                             '-m', self.manifestFile,
+                             '--depth', str(self.repoDepth)])
 
         if self.manifestOverrideUrl:
             self.stdio_log.addHeader("overriding manifest with %s\n" % (self.manifestOverrideUrl))

diff --git a/master/buildbot/test/unit/test_steps_source_repo.py b/master/buildbot/test/unit/test_steps_source_repo.py
@@ -127,7 +127,7 @@ def expectRepoSync(self, which_fail=-1, breakatfail=False, syncoptions=["-c"], o
                     'bash', '-c', self.step._getCleanupCommand()]),
             self.ExpectShell(
                 command=['repo', 'init', '-u', 'git://myrepo.com/manifest.git',
-                         '-b', 'mb', '-m', 'mf'])
+                         '-b', 'mb', '-m', 'mf', '--depth', '0'])
         ] + override_commands + [
             self.ExpectShell(command=['repo', 'sync'] + syncoptions),
             self.ExpectShell(
@@ -395,7 +395,7 @@ def test_repo_download_manifest(self):
             + 0,
             self.ExpectShell(
                 command=['repo', 'init', '-u', 'git://myrepo.com/manifest.git',
-                         '-b', 'mb', '-m', 'mf'])
+                         '-b', 'mb', '-m', 'mf', '--depth', '0'])
             + 0,
             self.ExpectShell(
                 workdir='wkdir/.repo/manifests',