CrOS: Update Chromite recipe module.

scripts/slave/recipe_modules/chromite/api.py

Index: scripts/slave/recipe_modules/chromite/api.py
diff --git a/scripts/slave/recipe_modules/chromite/api.py b/scripts/slave/recipe_modules/chromite/api.py
index 7149b01211bb75f5f0b89505545150024fa28896..1be029ea56a8ae7b1502ccda2349a6a3e9ad7913 100644
--- a/scripts/slave/recipe_modules/chromite/api.py
+++ b/scripts/slave/recipe_modules/chromite/api.py
@@ -20,6 +20,21 @@ class ChromiteApi(recipe_api.RecipeApi):
   _MANIFEST_CMD_RE = re.compile(r'Automatic:\s+Start\s+([^\s]+)\s+([^\s]+)')
   _BUILD_ID_RE = re.compile(r'CrOS-Build-Id: (.+)')
 
+  def get_config_defaults(self):
+    build_number = self.m.properties.get('buildnumber')
+    if build_number is not None:
+        # On a developer system, it's been noted that when the build number is
+        # zero, it's passed as an empty string in the properties JSON blob.

[luqui, 2015/07/31 19:58:15]

The purpose of this comment is unclear.  So this line is a bug workaround?  File
a crbug and link it if so.

[dnj (Google), 2015/08/01 01:31:34]

Filed https://code.google.com/p/chromium/issues/detail?id=516091

Rather than link, I'm going to just fix this bug:
https://codereview.chromium.org/1254003005

+        build_number = int(build_number or 0)
+
+    return {
+        'CBB_CONFIG': self.m.properties.get('cbb_config'),
+        'CBB_BRANCH': self.m.properties.get('cbb_branch'),
+        'CBB_BUILD_NUMBER': build_number,
+        'CBB_DEBUG': self.m.properties.get('cbb_debug') is not None,
+        'CBB_CLOBBER': 'clobber' in self.m.properties,
+    }
+
   def check_repository(self, repo_type_key, value):
     """Scans through registered repositories for a specified value.
 
@@ -121,6 +136,7 @@ class ChromiteApi(recipe_api.RecipeApi):
       loaded.insert(0, '')
       result.presentation.step_text += '<br/>'.join(loaded)
 
+  @property
   def default_chromite_path(self):
     """Returns: (Path) The default Chromite checkout path."""
     return self.m.path['slave_build'].join(self.chromite_subpath)
@@ -146,6 +162,12 @@ class ChromiteApi(recipe_api.RecipeApi):
     self.m.repo.init(manifest_url, '--repo-url', repo_url)
     self.m.repo.sync()
 
+  @property
+  def old_chromite_layout(self):

[luqui, 2015/07/31 19:58:15]

How about using_old_chromite_layout.  This naming makes me think that it returns
the layout (which I know is nonsense, still confused me though)

[dnj (Google), 2015/08/01 01:31:34]

Done.

+    """Returns (bool): True if we're using old Chromite checkout layout.
+    """
+    return self.c.chromite_branch in self.c.old_chromite_branches
+
   def cbuildbot(self, name, config, args=None, chromite_path=None, **kwargs):
     """Runs the cbuildbot command defined by the arguments.
 
@@ -154,23 +176,22 @@ class ChromiteApi(recipe_api.RecipeApi):
       config: (str) The name of the 'cbuildbot' configuration to invoke.
       args: (list) If not None, addition arguments to pass to 'cbuildbot'.
       chromite_path: (str) The path to the Chromite checkout; if None, the
-          'default_chromite_path()' will be used.
+          'default_chromite_path' will be used.
 
     Returns: (Step) The step that was run.
     """
-    chromite_path = chromite_path or self.default_chromite_path()
+    chromite_path = chromite_path or self.default_chromite_path
     args = (args or [])[:]
     args.append(config)
 
-    cmd = [self.m.path.join(chromite_path, 'bin', 'cbuildbot')] + args
-
-    # TODO(petermayo): Wrap this nested annotation in a stabilizing wrapper.
+    bindir = ('bin' if not self.old_chromite_layout else 'buildbot')

[luqui, 2015/07/31 19:58:15]

This is convoluted to read.  I see what you're going for though, 'bin' is like
the usual one, 'buildbot' is the strange off-case.  Maybe

   bindir = 'bin'
   if self.old_chromite_layout:
       bindir = 'buildbot'

[dnj (Google), 2015/08/01 01:31:34]

Agreed this is more readable. Stylistically I like this better too, so Done!

+    cmd = [self.m.path.join(chromite_path, bindir, 'cbuildbot')] + args
     return self.m.step(name, cmd, allow_subannotations=True, **kwargs)
 
   def cros_sdk(self, name, cmd, args=None, environ=None, chromite_path=None,
                  **kwargs):
     """Return a step to run a command inside the cros_sdk."""
-    chromite_path = chromite_path or self.default_chromite_path()
+    chromite_path = chromite_path or self.default_chromite_path
 
     chroot_cmd = self.m.path.join(chromite_path, 'bin', 'cros_sdk')
 
@@ -216,30 +237,12 @@ class ChromiteApi(recipe_api.RecipeApi):
       for config_name in config_map.get('variants', {}).get(variant, ()):
         self.apply_config(config_name)
 
-    # If a Chromite branch is supplied, use it to override the default Chromite
-    # checkout revision.
-    if properties.get('cbb_branch'):
-      self.c.chromite_branch = properties['cbb_branch']
-
-    # Set the build number if one is defined in the properties.
-    if self.m.properties.get('buildnumber') is not None:
-      # On a developer system, it's been noted that when the build number is
-      # zero, it's passed as an empty string in the properties JSON blob.
-      self.c.cbb.build_number = int(self.m.properties['buildnumber'] or 0)
-
-    # Run a debug build if instructed.
-    if properties.get('cbb_debug'):
-      self.c.cbb.debug = True
-
-    # If a clobber build was requested, set this builder to clobber.
-    if 'clobber' in properties:
-      self.c.cbb.clobber = True
 
     # If a config repo was specified, use it.
     if 'config_repo' in properties:
       self.c.cbb.config_repo = self.m.properties['config_repo']
 
-  def run_cbuildbot(self, config, tryjob=False):
+  def run_cbuildbot(self, tryjob=False):
     """Runs a 'cbuildbot' checkout-and-build workflow.
 
     This workflow uses the registered configuration dictionary to make master-
@@ -258,14 +261,12 @@ class ChromiteApi(recipe_api.RecipeApi):
     - Executes the 'cbuildbot' command.
 
     Args:
-      config (str): The name of the 'cbuildbot' configuration target to build.
       tryjob (bool): If True, load a tryjob description from the source
           repository and augment the cbuildbot command-line with it.
     Returns: (Step) the 'cbuildbot' execution step.
     """
 
     # Assert correct configuration.
-    assert config, 'An empty configuration was specified.'
     assert self.c.cbb.builddir, 'A build directory name must be specified.'
 
     # Load properties from the commit being processed. This requires both a
@@ -328,8 +329,8 @@ class ChromiteApi(recipe_api.RecipeApi):
         force=True)
 
     # Run cbuildbot.
-    return self.cbuildbot(str('cbuildbot [%s]' % (config,)),
-                          config,
+    return self.cbuildbot(str('cbuildbot [%s]' % (self.c.cbb.config,)),
+                          self.c.cbb.config,
                           args=cbb_args,
                           chromite_path=self.m.path['checkout'],
                           cwd=self.m.path['slave_root'])