[Findit] Trigger swarming tasks after detech_first_faliure_pipeline

appengine/findit/waterfall/test/swarming_tasks_to_try_job_pipeline_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.
 
 import json
 
 from common.git_repository import GitRepository
 from common.pipeline_wrapper import BasePipeline
 from common.pipeline_wrapper import pipeline_handlers
 from common.waterfall import buildbucket_client
 from model import result_status
 from model.wf_analysis import WfAnalysis
+from model.wf_swarming_task import WfSwarmingTask
 from model.wf_try_job import WfTryJob
 from waterfall import send_notification_for_culprit_pipeline
 from waterfall import swarming_util
 from waterfall import trigger_swarming_task_pipeline
 from waterfall.swarming_task_request import SwarmingTaskRequest
 from waterfall.swarming_tasks_to_try_job_pipeline import (
     SwarmingTasksToTryJobPipeline)
 from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
@@ skipping 232 matching lines @@
     self.assertEqual(analysis.result_status,
                      result_status.FOUND_UNTRIAGED)
     self.assertEqual(analysis.suspected_cls,
                      [expected_suspected_cl])
 
   def testSuccessfullyScheduleNewTryJobForTest(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     targeted_tests = {
-        'a_test': ['TestSuite1.PRE_test1', 'TestSuite1.test3'],
+        'a_test': ['TestSuite1.test1', 'TestSuite1.test3'],
         'b_test': [],  # Non-swarming test.
     }
 
-    # Mocks for TriggerSwarmingTaskPipeline.
-    def MockedDownloadSwarmingTaskData(*_):
-      return [{'task_id': '1'}, {'task_id': '2'}]
-    self.mock(swarming_util, 'ListSwarmingTasksDataByTags',
-              MockedDownloadSwarmingTaskData)
-
-    def MockedGetSwarmingTaskRequest(ref_task_id, *_):
-      self.assertEqual('1', ref_task_id)
-      return SwarmingTaskRequest.Deserialize({
-          'expiration_secs': 3600,
-          'name': 'ref_task_request',
-          'parent_task_id': 'pti',
-          'priority': 25,
-          'properties': {
-              'command': 'cmd',
-              'dimensions': [{'key': 'k', 'value': 'v'}],
-              'env': [
-                  {'key': 'a', 'value': '1'},
-                  {'key': 'GTEST_SHARD_INDEX', 'value': '1'},
-                  {'key': 'GTEST_TOTAL_SHARDS', 'value': '5'},
-              ],
-              'execution_timeout_secs': 3600,
-              'extra_args': ['--flag=value', '--gtest_filter=d.f'],
-              'grace_period_secs': 30,
-              'idempotent': True,
-              'inputs_ref': {'a': 1},
-              'io_timeout_secs': 1200,
-          },
-          'tags': ['master:a', 'buildername:b'],
-          'user': 'user',
-      })
-    self.mock(swarming_util, 'GetSwarmingTaskRequest',
-              MockedGetSwarmingTaskRequest)
-
-    new_request_json = {}
-    def MockedTriggerSwarmingTask(new_request, *_):
-      self.assertEqual({}, new_request_json)
-      new_request_json.update(new_request.Serialize())
-      return 'task_id1'
-    self.mock(swarming_util, 'TriggerSwarmingTask', MockedTriggerSwarmingTask)
-
-    def MockedGetSwarmingTaskName(*_):
-      return 'new_task_name'
-    self.mock(trigger_swarming_task_pipeline, '_GetSwarmingTaskName',
-              MockedGetSwarmingTaskName)
-
     # Mocks for ProcessSwarmingTaskResultPipeline.
     def MockedGetSwarmingTaskResultById(task_id, _):
       swarming_task_results = {
           'task_id1': {
               'state': 'COMPLETED',
               'outputs_ref': {
                   'isolatedserver': _ISOLATED_SERVER,
                   'namespace': 'default-gzip',
                   'isolated': 'shard1_isolated'
               }
@@ skipping 17 matching lines @@
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
     self._MockTriggerTryJobs(responses)
     self._MockGetTryJobs('2')
     self._MockGetChangeLog('rev1')
     self._Mock_SendNotificationForCulpritPipeline()
 
+    task = WfSwarmingTask.Create(
+        master_name, builder_name, build_number, 'a_test')
+    task.task_id = 'task_id1'
+    task.put()
     WfTryJob.Create(master_name, builder_name, build_number).put()
     analysis = WfAnalysis.Create(master_name, builder_name, build_number)
     analysis.put()
 
     root_pipeline = SwarmingTasksToTryJobPipeline(
         master_name, builder_name, build_number, 'rev0', 'rev1', ['rev1'],
         TryJobType.TEST, None, targeted_tests)
     root_pipeline.start()
     self.execute_queued_tasks()
 
@@ skipping 37 matching lines @@
                     }
                 }
             }
         }
     ]
 
     self.assertEqual(expected_try_job_results, try_job.test_results)
     self.assertEqual(analysis.result_status,
                      result_status.FOUND_UNTRIAGED)
     self.assertEqual(analysis.suspected_cls, [expected_suspected_cl])