De-duplicate steps between chromium and chromium_trybot recipes

scripts/slave/recipes/chromium_trybot.py

 # Copyright 2013 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.
 
 DEPS = [
   'bot_update',
   'chromium',
   'gclient',
   'isolate',
   'itertools',
@@ skipping 364 matching lines @@
   conf['enable_swarming'] = True
   BUILDERS[server]['builders'][swarming] = conf
 
 
 add_swarming_builder('linux_chromium_rel', 'linux_chromium_rel_swarming')
 add_swarming_builder('win_chromium_rel', 'win_chromium_rel_swarming')
 add_swarming_builder('mac_chromium_rel', 'mac_chromium_rel_swarming')
 
 
 def GenSteps(api):
-  class CheckdepsTest(api.test_utils.Test):  # pylint: disable=W0232
-    name = 'checkdeps'
-
-    @staticmethod
-    def compile_targets():
-      return []
-
-    @staticmethod
-    def run(suffix):
-      return api.chromium.checkdeps(
-          suffix, can_fail_build=False, always_run=True)
-
-    def has_valid_results(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output is not None
-
-    def failures(self, suffix):
-      results = api.step_history[self._step_name(suffix)].json.output
-      result_set = set()
-      for result in results:
-        for violation in result['violations']:
-          result_set.add((result['dependee_path'], violation['include_path']))
-      return ['%s: %s' % (r[0], r[1]) for r in result_set]
-
-
-  class CheckpermsTest(api.test_utils.Test):  # pylint: disable=W0232
-    name = 'checkperms'
-
-    @staticmethod
-    def compile_targets():
-      return []
-
-    @staticmethod
-    def run(suffix):
-      return api.chromium.checkperms(
-          suffix, can_fail_build=False, always_run=True)
-
-    def has_valid_results(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output is not None
-
-    def failures(self, suffix):
-      results = api.step_history[self._step_name(suffix)].json.output
-      result_set = set()
-      for result in results:
-        result_set.add((result['rel_path'], result['error']))
-      return ['%s: %s' % (r[0], r[1]) for r in result_set]
-
-
-  class ChecklicensesTest(api.test_utils.Test):  # pylint: disable=W0232
-    name = 'checklicenses'
-
-    @staticmethod
-    def compile_targets():
-      return []
-
-    @staticmethod
-    def run(suffix):
-      return api.chromium.checklicenses(
-          suffix, can_fail_build=False, always_run=True)
-
-    def has_valid_results(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output is not None
-
-    def failures(self, suffix):
-      results = api.step_history[self._step_name(suffix)].json.output
-      result_set = set()
-      for result in results:
-        result_set.add((result['filename'], result['license']))
-      return ['%s: %s' % (r[0], r[1]) for r in result_set]
-
-
-  class Deps2GitTest(api.test_utils.Test):  # pylint: disable=W0232
-    name = 'deps2git'
-
-    @staticmethod
-    def compile_targets():
-      return []
-
-    @staticmethod
-    def run(suffix):
-      yield (
-        api.chromium.deps2git(suffix, can_fail_build=False),
-        api.chromium.deps2submodules()
-      )
-
-    def has_valid_results(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output is not None
-
-    def failures(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output
-
-
-  class GTestTest(api.test_utils.Test):
-    def __init__(self, name, args=None, compile_targets=None):
-      api.test_utils.Test.__init__(self)
-      self._name = name
-      self._args = args or []
-
-    @property
-    def name(self):
-      return self._name
-
-    def compile_targets(self):
-      return [self._name]
-
-    def run(self, suffix):
-      def followup_fn(step_result):
-        r = step_result.json.gtest_results
-        p = step_result.presentation
-
-        if r.valid:
-          p.step_text += api.test_utils.format_step_text([
-              ['failures:', r.failures]
-          ])
-
-      # Copy the list because run can be invoked multiple times and we modify
-      # the local copy.
-      args = self._args[:]
-
-      if suffix == 'without patch':
-        args.append(api.chromium.test_launcher_filter(
-                        self.failures('with patch')))
-
-      return api.chromium.runtest(
-          self.name,
-          args,
-          annotate='gtest',
-          test_launcher_summary_output=api.json.gtest_results(
-              add_json_log=False),
-          xvfb=True,
-          name=self._step_name(suffix),
-          parallel=True,
-          can_fail_build=False,
-          followup_fn=followup_fn,
-          step_test_data=lambda: api.json.test_api.canned_gtest_output(True))
-
-    def has_valid_results(self, suffix):
-      step_name = self._step_name(suffix)
-      gtest_results = api.step_history[step_name].json.gtest_results
-      if not gtest_results.valid:  # pragma: no cover
-        return False
-      global_tags = gtest_results.raw.get('global_tags', [])
-      return 'UNRELIABLE_RESULTS' not in global_tags
-
-    def failures(self, suffix):
-      step_name = self._step_name(suffix)
-      return api.step_history[step_name].json.gtest_results.failures
-
-
-  class SwarmingGTestTest(api.test_utils.Test):
-    def __init__(self, name, args=None, shards=1):
-      api.test_utils.Test.__init__(self)
-      self._name = name
-      self._args = args or []
-      self._shards = shards
-      self._tasks = {}
-      self._results = {}
-
-    @property
-    def name(self):
-      return self._name
-
-    def compile_targets(self):
-      # <X>_run target depends on <X>, and then isolates it invoking isolate.py.
-      # It is a convention, not a hard coded rule.
-      return [self._name + '_run']
-
-    def pre_run(self, suffix):
-      """Launches the test on Swarming."""
-      assert suffix not in self._tasks, (
-          'Test %s was already triggered' % self._step_name(suffix))
-
-      # *.isolated may be missing if *_run target is misconfigured. It's a error
-      # in gyp, not a recipe failure. So carry on with recipe execution.
-      isolated_hash = api.isolate.isolated_tests.get(self._name)
-      if not isolated_hash:
-        return api.python.inline(
-            '[error] %s' % self._step_name(suffix),
-            r"""
-            import sys
-            print '*.isolated file for target %s is missing' % sys.argv[1]
-            sys.exit(1)
-            """,
-            args=[self._name],
-            always_run=True)
-
-      # If rerunning without a patch, run only tests that failed.
-      args = self._args[:]
-      if suffix == 'without patch':
-        failed_tests = sorted(self.failures('with patch'))
-        args.append('--gtest_filter=%s' % ':'.join(failed_tests))
-
-      # Trigger the test on swarming.
-      self._tasks[suffix] = api.swarming.gtest_task(
-          title=self._step_name(suffix),
-          isolated_hash=isolated_hash,
-          shards=self._shards,
-          test_launcher_summary_output=api.json.gtest_results(
-              add_json_log=False),
-          extra_args=args)
-      return api.swarming.trigger([self._tasks[suffix]], always_run=True)
-
-    def run(self, suffix):  # pylint: disable=R0201
-      """Not used. All logic in pre_run, post_run."""
-      return []
-
-    def post_run(self, suffix):
-      """Waits for launched test to finish and collect the results."""
-      assert suffix not in self._results, (
-          'Results of %s were already collected' % self._step_name(suffix))
-
-      # Emit error if test wasn't triggered. This happens if *.isolated is not
-      # found. (The build is already red by this moment anyway).
-      if suffix not in self._tasks:
-        return api.python.inline(
-            '[collect error] %s' % self._step_name(suffix),
-            r"""
-            import sys
-            print '%s wasn\'t triggered' % sys.argv[1]
-            sys.exit(1)
-            """,
-            args=[self._name],
-            always_run=True)
-
-      # Update step presentation, store step results in self._results.
-      def followup_fn(step_result):
-        r = step_result.json.gtest_results
-        p = step_result.presentation
-        if r.valid:
-          p.step_text += api.test_utils.format_step_text([
-              ['failures:', r.failures]
-          ])
-        self._results[suffix] = r
-
-      # Wait for test on swarming to finish. If swarming infrastructure is
-      # having issues, this step produces no valid *.json test summary, and
-      # 'has_valid_results' returns False.
-      return api.swarming.collect(
-          [self._tasks[suffix]],
-          always_run=True,
-          can_fail_build=False,
-          followup_fn=followup_fn)
-
-    def has_valid_results(self, suffix):
-      # Test wasn't triggered or wasn't collected.
-      if suffix not in self._tasks or not suffix in self._results:
-        return False
-      # Test ran, but failed to produce valid *.json.
-      gtest_results = self._results[suffix]
-      if not gtest_results.valid:  # pragma: no cover
-        return False
-      global_tags = gtest_results.raw.get('global_tags', [])
-      return 'UNRELIABLE_RESULTS' not in global_tags
-
-    def failures(self, suffix):
-      assert self.has_valid_results(suffix)
-      return self._results[suffix].failures
-
-
-  class NaclIntegrationTest(api.test_utils.Test):  # pylint: disable=W0232
-    name = 'nacl_integration'
-
-    @staticmethod
-    def compile_targets():
-      return ['chrome']
-
-    def run(self, suffix):
-      args = [
-        '--mode', api.chromium.c.build_config_fs,
-        '--json_build_results_output_file', api.json.output(),
-      ]
-      return api.python(
-          self._step_name(suffix),
-          api.path['checkout'].join('chrome',
-                            'test',
-                            'nacl_test_injection',
-                            'buildbot_nacl_integration.py'),
-          args,
-          can_fail_build=False,
-          step_test_data=lambda: api.m.json.test_api.output([]))
-
-    def has_valid_results(self, suffix):
-      return api.step_history[self._step_name(suffix)].json.output is not None
-
-    def failures(self, suffix):
-      failures = api.step_history[self._step_name(suffix)].json.output
-      return [f['raw_name'] for f in failures]
-
-
   def parse_test_spec(test_spec, enable_swarming, should_use_test):
     """Returns a list of tests to run and additional targets to compile.
 
     Uses 'should_use_test' callback to figure out what tests should be skipped.
 
     Returns triple (compile_targets, gtest_tests, swarming_tests) where
       gtest_tests is a list of GTestTest
       swarming_tests is a list of SwarmingGTestTest.
     """
     compile_targets = []
@@ skipping 40 matching lines @@
         if swarming_spec.get('can_use_on_swarming_builders'):
           use_swarming = should_use_test(swarming_spec)
         swarming_shards = swarming_spec.get('shards', 1)
 
       test_args = test_dict.get('args')
       if isinstance(test_args, basestring):
         test_args = [test_args]
 
       if use_swarming:
         swarming_tests.append(
-            SwarmingGTestTest(test_name, test_args, swarming_shards))
+            api.chromium.steps.SwarmingGTestTest(
+                test_name, test_args, swarming_shards))
       else:
-        gtest_tests.append(GTestTest(test_name, test_args))
+        gtest_tests.append(api.chromium.steps.GTestTest(test_name, test_args))
 
     return compile_targets, gtest_tests, swarming_tests
 
 
   mastername = api.properties.get('mastername')
   buildername = api.properties.get('buildername')
   master_dict = BUILDERS.get(mastername, {})
   bot_config = master_dict.get('builders', {}).get(buildername)
   assert bot_config, (
       'Unrecognized builder name %r for master %r.' % (
@@ skipping 69 matching lines @@
   yield api.chromium.runhooks(env=runhooks_env)
 
   # If going to use swarming_client (pinned in src/DEPS), ensure it is
   # compatible with what recipes expect.
   if swarming_tests:
     yield api.swarming.check_client_version()
 
   tests = []
   # TODO(phajdan.jr): Re-enable checkdeps on Windows when it works with git.
   if not api.platform.is_win:
-    tests.append(CheckdepsTest())
+    tests.append(api.chromium.steps.CheckdepsTest())
   if api.platform.is_linux:
     tests.extend([
-        CheckpermsTest(),
-        ChecklicensesTest(),
+        api.chromium.steps.CheckpermsTest(),
+        api.chromium.steps.ChecklicensesTest(),
     ])
-  tests.append(Deps2GitTest())
+  tests.append(api.chromium.steps.Deps2GitTest())
   tests.extend(gtest_tests)
   tests.extend(swarming_tests)
-  tests.append(NaclIntegrationTest())
+  tests.append(api.chromium.steps.NaclIntegrationTest())
 
   compile_targets.extend(bot_config.get('compile_targets', []))
   compile_targets.extend(api.itertools.chain(
-      *[t.compile_targets() for t in tests]))
+      *[t.compile_targets(api) for t in tests]))
   # TODO(phajdan.jr): Also compile 'all' on win, http://crbug.com/368831 .
   # Disabled for now because it takes too long and/or fails on Windows.
   if not api.platform.is_win and not bot_config.get('exclude_compile_all'):
     compile_targets = ['all'] + compile_targets
   yield api.chromium.compile(compile_targets,
                              name='compile (with patch)')
 
   # Do not run tests if the build is already in a failed state.
   if api.step_history.failed:
     return
@@ skipping 21 matching lines @@
       yield api.chromium.taskkill()
     bot_update_json = api.step_history['bot_update'].json.output
     api.gclient.c.solutions[0].revision = str(
         bot_update_json['properties']['got_revision'])
     yield api.bot_update.ensure_checkout(force=True,
                                          patch=False,
                                          always_run=True,
                                          update_presentation=False)
     yield api.chromium.runhooks(always_run=True),
     compile_targets = list(api.itertools.chain(
-                               *[t.compile_targets() for t in failing_tests]))
+        *[t.compile_targets(api) for t in failing_tests]))
     if compile_targets:
       yield api.chromium.compile(compile_targets,
                                  name='compile (without patch)',
                                  abort_on_failure=False,
                                  can_fail_build=False,
                                  always_run=True)
       if api.step_history['compile (without patch)'].retcode != 0:
         yield api.chromium.compile(compile_targets,
                                    name='compile (without patch, clobber)',
                                    force_clobber=True,
                                    always_run=True)
       # Search for *.isolated only if enabled in bot config or if some swarming
       # test is being recompiled.
       failing_swarming_tests = set(failing_tests) & set(swarming_tests)
       if bot_config.get('use_isolate') or failing_swarming_tests:
         yield api.isolate.find_isolated_tests(api.chromium.output_dir,
                                               always_run=True)
 
-  yield api.test_utils.determine_new_failures(tests, deapply_patch_fn)
+  yield api.test_utils.determine_new_failures(api, tests, deapply_patch_fn)
 
 
 def _sanitize_nonalpha(text):
   return ''.join(c if c.isalnum() else '_' for c in text)
 
 
 def GenTests(api):
   canned_checkdeps = {
     True: [],
     False: [
@@ skipping 205 matching lines @@
     ) +
     api.override_step_data(
         'find isolated tests',
         api.isolate.output_json(['base_unittests', 'browser_tests'])) +
     api.override_step_data('[swarming] base_unittests (with patch)',
                            canned_test(passing=False)) +
     api.override_step_data(
         'find isolated tests (2)',
         api.isolate.output_json(['base_unittests']))
   )