Make the recipe roller tryjob accept tryjobs.

scripts/slave/recipes/infra/recipe_roll_tryjob.py

 # Copyright (c) 2014 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.
 
 """Rolls recipes.cfg dependencies."""
 
 DEPS = [
   'depot_tools/bot_update',
   'depot_tools/gclient',
   'file',
@@ skipping 163 matching lines @@
   for project in fetch_result['projects']:
     mapping[project['id']] = project
     mapping[project['id']] = dict(project)
   return mapping
 
 
 RietveldPatch = collections.namedtuple(
     'RietveldPatch', 'project server issue patchset')
 
 
-def parse_patches(api, patches_raw):
+def parse_patches(api, patches_raw, rietveld, issue, patchset, patch_project):
   """
   gives mapping of project to patch
     expect input of
     project1:https://a.b.c/1342342#ps1,project2:https://d.ce.f/1231231#ps1
   """
+  result = {}
+
+  if rietveld and issue and patchset and patch_project:
+    result[patch_project] = RietveldPatch(
+        patch_project, rietveld, issue, patchset)
+
   if not patches_raw:
-    return {}
-
-  result = {}
+    return result
 
   for patch_raw in patches_raw.split(','):
     project, url = patch_raw.split(':', 1)
     server, issue_and_patchset = url.rsplit('/', 1)
     issue, patchset = issue_and_patchset.split('#')
     patchset = patchset[2:]
 
     if project in result:
       api.python.failing_step(
           "Invalid patchset list",
@@ skipping 107 matching lines @@
   'depot_tools',
 ]
 
 PROPERTIES = {
   'patches': Property(kind=str, param_name='patches_raw', default="",
                       help="Patches to apply. Format is"
                       "project1:https://url.to.codereview/123456#ps01 where"
                       "url.to.codereview is the address of the code review site"
                       ", 123456 is the issue number, and ps01 is the patchset"
                       "number"),
+  # This recipe can be used as a tryjob by setting the rietveld, issue, and
+  # patchset properties, like a normal tryjob. If those are set, it will use
+  # those, as well as any data sent in the regular properties, as patches to
+  # apply.
+  "rietveld": Property(kind=str, default="",
+                       help="The code review site the tryjob patch isfrom"),
+  "issue": Property(kind=int, default=None,
+                    help="The issue number this patch is from"),
+  "patchset": Property(kind=int, default=None,
+                       help="The patchset to apply from this patch"),
+  "patch_project": Property(
+      kind=str, default=None,
+      help="The luci-config name of the project this patch belongs to"),
 }
 
-def RunSteps(api, patches_raw):
+def RunSteps(api, patches_raw, rietveld, issue, patchset, patch_project):
   auth_token = get_auth_token(api)
   headers = {'Authorization': 'Bearer %s' % auth_token}
 
-  patches = parse_patches(api, patches_raw)
+  patches = parse_patches(
+      api, patches_raw, rietveld, issue, patchset, patch_project)
 
   root_dir = api.path['slave_build']
 
   url_mapping = get_url_mapping(api, auth_token)
 
   # luci config project name to recipe config namedtuple
   recipe_configs = {}
 
   # List of all the projects we care about testing. luci-config names
   all_projects = set(p for p in url_mapping if p in PROJECTS_TO_TRY)
 
   recipe_configs = {
       p: get_project_config(api, p, headers) for p in all_projects}
 
-  downstream_projects, deps = get_deps_info(all_projects, recipe_configs)
+  deps, downstream_projects = get_deps_info(all_projects, recipe_configs)
 
   projs_to_test, locations = checkout_projects(
       api, all_projects, url_mapping, downstream_projects, root_dir, patches)
 
   with api.step.defer_results():
     for proj in projs_to_test:
       deps_locs = {dep: locations[dep] for dep in deps[proj]}
 
       simulation_test(
           api, proj, recipe_configs[proj], locations[proj], deps_locs)
@@ skipping 48 matching lines @@
       api.properties(
           patches="build:https://f.e.w/1#ps1,build:https://f.e.w/1#ps1")
   )
 
   yield (
       api.test('deps') +
       api.step_data("Get build deps", project('build', ['recipe_engine'])) +
       api.step_data("Get recipe_engine deps", project('recipe_engine'))
   )
 
+  yield (
+      api.test('tryjob') +
+      api.properties(
+        rietveld="https://fake.code.review",
+        issue=12345678,
+        patchset=1,
+        patch_project="build",
+      ) +
+      api.step_data("Get build deps", project('build', ['recipe_engine'])) +
+      api.step_data("Get recipe_engine deps", project('recipe_engine'))
+  )
+