[Findit] Refactor Findit pipeline.

appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py

Index: appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py
diff --git a/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py b/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py
index 7fb7c666f7608f33d01151010e39a9f0bf5f0199..627fa392c207bb7ea7fa227ed50c7409763f65d9 100644
--- a/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py
+++ b/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py
@@ -7,6 +7,7 @@ import os
 from common import constants
 from common import chromium_deps
 from common.pipeline_wrapper import pipeline_handlers
+from common.waterfall import failure_type
 from model import analysis_status
 from model.wf_analysis import WfAnalysis
 from waterfall import analyze_build_failure_pipeline
@@ -58,22 +59,21 @@ class AnalyzeBuildFailurePipelineTest(wf_testcase.WaterfallTestCase):
         analyze_build_failure_pipeline.TriggerSwarmingTasksPipeline,
         None,
         expected_args=[
-            master_name, builder_name, build_number,
-            'failure_info'],
+            master_name, builder_name, build_number, 'failure_info'],
         expected_kwargs={})
     self.MockPipeline(
         analyze_build_failure_pipeline.StartTryJobOnDemandPipeline,
-        None,
+        'try_job_result',
         expected_args=[
-            'failure_info', 'signals', False, True,
-            'heuristic_result'],
+            master_name, builder_name, build_number, 'failure_info', 'signals',
+            'heuristic_result', False, False],
         expected_kwargs={})
 
     root_pipeline = AnalyzeBuildFailurePipeline(master_name,
                                                 builder_name,
                                                 build_number,
                                                 False,
-                                                True)
+                                                False)
     root_pipeline.start(queue_name=constants.DEFAULT_QUEUE)
     self.execute_queued_tasks()