[Findit] Adding logic to force try jobs regardless of bailout or previous results

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 datetime import datetime
 from datetime import timedelta
 
 from common.waterfall import failure_type
 from model import analysis_status
 from model.wf_build import WfBuild
@@ skipping 52 matching lines @@
                 'chromium_revision': '223-3'
             }
         }
     }
 
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
-    def _MockShouldBailOutForOutdatedBuild(*_):
-      return False
-    self.mock(
-        try_job_util, '_ShouldBailOutForOutdatedBuild',
-        _MockShouldBailOutForOutdatedBuild)
-
     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
@@ skipping 12 matching lines @@
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)
 
     failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
         failure_info, None, None)
 
     self.assertEqual({}, failure_result_map)
 
   def testBailOutForTryJobWithOutdatedTimestamp(self):
-    master_name = 'master3'
-    builder_name = 'builder3'
+    master_name = 'master1'
+    builder_name = 'builder1'
     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
             }
         },
     }
 
+    yesterday = datetime.utcnow() - timedelta(days=1)
+    build = WfBuild.Create(master_name, builder_name, build_number)
+    build.start_time = yesterday
+    build.put()
+
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     def _MockShouldBailOutForOutdatedBuild(*_):
       return True
+
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)
 
     failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
-        failure_info, None, None)
+        failure_info, None, None, False)
 
     self.assertFalse(_MockRootPipeline.STARTED)
     self.assertEqual({}, failure_result_map)
 
+  def testForceTryJob(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': {
+                'current_failure': 223,
+                'first_failure': 223,
+                'last_pass': 222
+            }
+        },
+        'builds': {
+            '222': {
+                'blame_list': ['222-1'],
+                'chromium_revision': '222-1'
+            },
+            '223': {
+                'blame_list': ['223-1', '223-2', '223-3'],
+                'chromium_revision': '223-3'
+            }
+        },
+        'failure_type': failure_type.COMPILE
+    }
+
+    self.mock(
+        try_job_util.swarming_tasks_to_try_job_pipeline,
+        'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
+    _MockRootPipeline.STARTED = False
+
+    try_job_util.ScheduleTryJobIfNeeded(failure_info, None, None, True)
+
+    try_job = WfTryJob.Get(master_name, builder_name, build_number)
+
+    self.assertTrue(_MockRootPipeline.STARTED)
+    self.assertIsNotNone(try_job)
+
   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 23 matching lines @@
         'failure_type': failure_type.COMPILE
     }
 
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     def _MockShouldBailOutForOutdatedBuild(*_):
       return False
+
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)
 
     try_job_util.ScheduleTryJobIfNeeded(failure_info, None, None)
 
     self.assertFalse(_MockRootPipeline.STARTED)
 
   def testNotNeedANewTryJobIfOneWithResultExists(self):
     master_name = 'm'
@@ skipping 221 matching lines @@
         'failure_type': failure_type.COMPILE
     }
 
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     def _MockShouldBailOutForOutdatedBuild(*_):
       return False
+
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)
 
     try_job_util.ScheduleTryJobIfNeeded(failure_info, None, None)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertTrue(_MockRootPipeline.STARTED)
     self.assertIsNotNone(try_job)
@@ skipping 101 matching lines @@
         try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))
 
   def testShouldBailOutforOutdatedBuild(self):
     yesterday = datetime.utcnow() - timedelta(days=1)
     build = WfBuild.Create('m', 'b', 1)
     build.start_time = yesterday
     self.assertTrue(try_job_util._ShouldBailOutForOutdatedBuild(build))
 
     build.start_time = yesterday + timedelta(hours=1)
     self.assertFalse(try_job_util._ShouldBailOutForOutdatedBuild(build))