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

appengine/findit/waterfall/test/try_job_util_test.py

Index: appengine/findit/waterfall/test/try_job_util_test.py
diff --git a/appengine/findit/waterfall/test/try_job_util_test.py b/appengine/findit/waterfall/test/try_job_util_test.py
index df855b3d4a30454e116d00c264ab8599fc936249..30b6136237d495578f70779947e2395bed90d5d6 100644
--- a/appengine/findit/waterfall/test/try_job_util_test.py
+++ b/appengine/findit/waterfall/test/try_job_util_test.py
@@ -70,12 +70,6 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
         '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)
 
@@ -108,8 +102,8 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
     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,
@@ -124,6 +118,11 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
         },
     }
 
+    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)
@@ -131,16 +130,57 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
 
     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'
@@ -184,6 +224,7 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
 
     def _MockShouldBailOutForOutdatedBuild(*_):
       return False
+
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)
@@ -425,6 +466,7 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
 
     def _MockShouldBailOutForOutdatedBuild(*_):
       return False
+
     self.mock(
         try_job_util, '_ShouldBailOutForOutdatedBuild',
         _MockShouldBailOutForOutdatedBuild)