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

appengine/findit/handlers/test/handlers_util_test.py

Index: appengine/findit/handlers/test/handlers_util_test.py
diff --git a/appengine/findit/handlers/test/handlers_util_test.py b/appengine/findit/handlers/test/handlers_util_test.py
index 2e692c511ba2d930a527957928f900f4fb1b39bb..2e49000cd126db6713313baa43ca977285e7ac8c 100644
--- a/appengine/findit/handlers/test/handlers_util_test.py
+++ b/appengine/findit/handlers/test/handlers_util_test.py
@@ -550,6 +550,35 @@ class HandlersUtilResultTest(wf_testcase.WaterfallTestCase):
     }
     self.assertEqual(expected_result, result)
 
+  def testGetAllTryJobResultsForTestNonSwarmingForcedTryJob(self):
+    tasks_info = {
+        'step1': {
+            'swarming_tasks': {
+                'm/b/119': {
+                    'task_info': {
+                        'status': result_status.NON_SWARMING_NO_RERUN
+                    },
+                    'all_tests': ['test1']
+                },
+            }
+        }
+    }
+    result = handlers_util._GetAllTryJobResultsForTest(
+        {'step1': 'm/b/119'}, tasks_info, True)
+
+    expected_result = {
+        'step1': {
+            'try_jobs': [
+                {
+                    'try_job_key': 'm/b/119',
+                    'ref_name': 'step1'
+                }
+            ]
+        }
+    }
+
+    self.assertEqual(expected_result, result)
+
   def testGetAllTryJobResultsForTestNoSwarmingTaskInfo(self):
     failure_result_map = {
         'step1': {
@@ -923,7 +952,7 @@ class HandlersUtilResultTest(wf_testcase.WaterfallTestCase):
     }
     WfTryJob.Create('m', 'b', '119').put()
     handlers_util._GetCulpritInfoForTryJobResultForTest(
-        try_job_key, culprits_info)
+        try_job_key, culprits_info, False)
 
     expected_culprits_info = {
         'step1 on platform': {
@@ -963,7 +992,7 @@ class HandlersUtilResultTest(wf_testcase.WaterfallTestCase):
     ]
     try_job.put()
     handlers_util._GetCulpritInfoForTryJobResultForTest(
-        try_job_key, culprits_info)
+        try_job_key, culprits_info, False)
 
     expected_culprits_info = {
         'step1 on platform': {