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

appengine/findit/handlers/test/handlers_util_test.py

 # Copyright 2016 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 handlers import handlers_util
 from handlers import result_status
 from model import analysis_status
 from model.wf_analysis import WfAnalysis
 from model.wf_swarming_task import WfSwarmingTask
 from model.wf_try_job import WfTryJob
@@ skipping 532 matching lines @@
                     'try_job_key': 'm/b/119',
                     'status': result_status.NO_TRY_JOB_REASON_MAP.get(
                         result_status.NON_SWARMING_NO_RERUN),
                     'tests': ['test1']
                 }
             ]
         }
     }
     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': {
             'test3': 'm/b/119'
         }
     }
 
     tasks_info = {}
 
     result = handlers_util._GetAllTryJobResultsForTest(
@@ skipping 353 matching lines @@
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3']
                 }
             ]
         }
     }
     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': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3'],
                     'status': analysis_status.PENDING
                 }
@@ skipping 19 matching lines @@
     try_job.status = analysis_status.RUNNING
     try_job.test_results = [
         {
             'url': ('http://build.chromium.org/p/tryserver.chromium.'
                     'linux/builders/linux_chromium_variable/builds/121'),
             'try_job_id': '121'
         }
     ]
     try_job.put()
     handlers_util._GetCulpritInfoForTryJobResultForTest(
-        try_job_key, culprits_info)
+        try_job_key, culprits_info, False)
 
     expected_culprits_info = {
         'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3'],
                     'status': analysis_status.RUNNING,
                     'try_job_url': (
                         'http://build.chromium.org/p/tryserver.chromium.'
                         'linux/builders/linux_chromium_variable/builds/121'),
                     'try_job_build_number': 121
                 }
             ]
         }
     }
     self.assertEqual(expected_culprits_info, culprits_info)