[Findit] Trigger swarming tasks after detech_first_faliure_pipeline

appengine/findit/waterfall/process_swarming_task_result_pipeline.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 collections import defaultdict
 import datetime
 import logging
 import time
 
 from common.http_client_appengine import HttpClientAppengine as HttpClient
@@ skipping 44 matching lines @@
 
 class ProcessSwarmingTaskResultPipeline(BasePipeline):
   """A pipeline for monitoring swarming task and processing task result.
 
   This pipeline waits for result for a swarming task and processes the result to
   generate a dict for statuses for each test run.
   """
 
   HTTP_CLIENT = HttpClient()
   # Arguments number differs from overridden method - pylint: disable=W0221
-  def run(self, master_name, builder_name, build_number, step_name, task_id):
+  def run(self, master_name, builder_name, build_number, step_name):
     """
     Args:
       master_name (str): The master name.
       builder_name (str): The builder name.
       build_number (str): The build number.
       step_name (str): The failed test step name.
-      task_id (str): Id for the swarming task which is triggered by Findit.
 
     Returns:
       A dict of lists for reliable/flaky tests.
     """
 
-    assert task_id
     timeout_hours = waterfall_config.GetSwarmingSettings().get(
         'task_timeout_hours')
     deadline = time.time() + timeout_hours * 60 * 60
     server_query_interval_seconds = waterfall_config.GetSwarmingSettings().get(
         'server_query_interval_seconds')
 
     task_started = False
     task_completed = False
     tests_statuses = {}
     step_name_no_platform = None
 
+    task = WfSwarmingTask.Get(
+        master_name, builder_name, build_number, step_name)
+    task_id = task.task_id
     while not task_completed:
       # Keeps monitoring the swarming task, waits for it to complete.
       data = swarming_util.GetSwarmingTaskResultById(
           task_id, self.HTTP_CLIENT)
       task_state = data['state']
       step_name_no_platform = swarming_util.GetTagValue(
           data.get('tags', {}), 'ref_name')
       if task_state not in swarming_util.STATES_RUNNING:
         task_completed = True
         task = WfSwarmingTask.Get(
@@ skipping 37 matching lines @@
 
     # Update swarming task metadate.
     task = WfSwarmingTask.Get(
         master_name, builder_name, build_number, step_name)
     task.created_time = _ConvertDateTime(data.get('created_ts'))
     task.started_time = _ConvertDateTime(data.get('started_ts'))
     task.completed_time = _ConvertDateTime(data.get('completed_ts'))
     task.put()
 
     return step_name, (step_name_no_platform, task.classified_tests)