[Findit] Adding support for disabling test try jobs for given master/builder

appengine/findit/waterfall/test/try_job_util_test.py

 # Copyright 2015 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.
 
 from model import analysis_status
 from model.wf_try_job import WfTryJob
 from waterfall import try_job_util
 from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
 class _MockRootPipeline(object):
   STARTED = False
 
   def __init__(self, *_):
     pass
 
   def start(self, *_, **__):
     _MockRootPipeline.STARTED = True
 
   @property
   def pipeline_status_path(self):
     return 'path'
 
 
 class TryJobUtilTest(wf_testcase.WaterfallTestCase):
 
   def testNotNeedANewTryJobIfBuilderIsNotSupportedYet(self):
-    master_name = 'master2'
-    builder_name = 'builder2'
+    master_name = 'master3'
+    builder_name = 'builder3'
     build_number = 223
     failure_info = {
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'failed_steps': {
             'compile': {
                 'current_failure': 221,
                 'first_failure': 221,
                 'last_pass': 220
@@ skipping 23 matching lines @@
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
         failure_info, None, None)
 
     self.assertFalse(_MockRootPipeline.STARTED)
     self.assertEqual({}, failure_result_map)
 
+  def testBailOutForTestTryJob(self):
+    master_name = 'master2'
+    builder_name = 'builder2'
+    build_number = 223
+    failure_info = {
+        'master_name': master_name,
+        'builder_name': builder_name,
+        'build_number': build_number,
+        'failed_steps': {
+            'a_test': {}
+        },
+    }
+
+    failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
+        failure_info, None, None)
+
+    self.assertEqual({}, failure_result_map)
+
   def testNotNeedANewTryJobIfNotFirstTimeFailure(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 223
     failure_info = {
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'failed_steps': {
             'compile': {
@@ skipping 319 matching lines @@
             {
                 'step_name': 'steps'
             },
         ]
     }
     expected_suspected_revisions = ['r1', 'r2']
     self.assertEqual(
         expected_suspected_revisions,
         try_job_util._GetSuspectsForCompileFailureFromHeuristicResult(
             heuristic_result))