[Findit] Trigger swarming tasks after detech_first_faliure_pipeline

appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py

Index: appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py
diff --git a/appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py b/appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py
index b6b82afc27403fab912f3b15bbe129f7c776d538..1bd527fa40442da05f38c1e36f9bf0e5c09ab6d9 100644
--- a/appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py
+++ b/appengine/findit/waterfall/test/process_swarming_task_result_pipeline_test.py
@@ -156,19 +156,20 @@ class ProcessSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     return _SAMPLE_FAILURE_LOG
 
   def testProcessSwarmingTaskResultPipeline(self):
-    task_id = 'task_id1'
 
     self.mock(swarming_util, 'GetSwarmingTaskFailureLog',
               self._MockedGetSwarmingTaskFailureLog)
 
-    WfSwarmingTask.Create(
+    task = WfSwarmingTask.Create(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name).put()
+        self.build_number, self.step_name)
+    task.task_id = 'task_id1'
+    task.put()
 
     pipeline = ProcessSwarmingTaskResultPipeline()
     step_name, task_info = pipeline.run(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name, task_id)
+        self.build_number, self.step_name)
 
     self.assertEqual(self.step_name, step_name)
     self.assertEqual('abc_tests', task_info[0])
@@ -188,16 +189,17 @@ class ProcessSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
                      task.completed_time)
 
   def testProcessSwarmingTaskResultPipelineTaskNotRunning(self):
-    task_id = 'task_id2'
 
-    WfSwarmingTask.Create(
+    task = WfSwarmingTask.Create(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name).put()
+        self.build_number, self.step_name)
+    task.task_id = 'task_id2'
+    task.put()
 
     pipeline = ProcessSwarmingTaskResultPipeline()
     step_name, task_info = pipeline.run(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name, task_id)
+        self.build_number, self.step_name)
 
     self.assertEqual(self.step_name, step_name)
     self.assertIsNone(task_info[0])
@@ -211,7 +213,6 @@ class ProcessSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     self.assertEqual({}, task.classified_tests)
 
   def testProcessSwarmingTaskResultPipelineTaskTimeOut(self):
-    task_id = 'task_id1'
 
     # Override swarming config settings to force a timeout.
     override_swarming_settings = {
@@ -220,14 +221,16 @@ class ProcessSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     self.UpdateUnitTestConfigSettings(
         'swarming_settings', override_swarming_settings)
 
-    WfSwarmingTask.Create(
+    task = WfSwarmingTask.Create(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name).put()
+        self.build_number, self.step_name)
+    task.task_id = 'task_id1'
+    task.put()
 
     pipeline = ProcessSwarmingTaskResultPipeline()
     step_name, task_info = pipeline.run(
         self.master_name, self.builder_name,
-        self.build_number, self.step_name, task_id)
+        self.build_number, self.step_name)
 
     self.assertEqual(self.step_name, step_name)
     self.assertEqual('abc_tests', task_info[0])