[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 25ad562e2b2985cfb6cb9e410c85b62815f61056..94df6a6de44a54e3b31b04a675f6175813e5c37d 100644
--- a/appengine/findit/handlers/test/handlers_util_test.py
+++ b/appengine/findit/handlers/test/handlers_util_test.py
@@ -537,6 +537,34 @@ 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': None
+                }
+            ]
+        }
+    }
+    self.assertEqual(expected_result, result)
+
   def testGetAllTryJobResultsForTestNoSwarmingTaskInfo(self):
     failure_result_map = {
         'step1': {
@@ -910,7 +938,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': {
@@ -950,7 +978,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': {