[Findit] Pipeline change to save suspected cls to data store.

appengine/findit/waterfall/test/trigger_swarming_tasks_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.
 
 from common.pipeline_wrapper import BasePipeline
 from common.pipeline_wrapper import pipeline_handlers
 from common.waterfall import failure_type
+from model.base_build_model import BaseBuildModel
 from model.wf_analysis import WfAnalysis
 from waterfall import trigger_swarming_task_pipeline
 from waterfall import trigger_swarming_tasks_pipeline
 from waterfall.test import wf_testcase
 from waterfall.trigger_swarming_tasks_pipeline import (
     TriggerSwarmingTasksPipeline)
-from waterfall.trigger_swarming_task_pipeline import TriggerSwarmingTaskPipeline
 
 
 class TriggerSwarmingTasksPipelineTest(wf_testcase.WaterfallTestCase):
   app_module = pipeline_handlers._APP
 
   def test_GetStepsThatNeedToTriggerSwarmingTasksNoAnalysis(self):
     result = (
         trigger_swarming_tasks_pipeline._GetStepsThatNeedToTriggerSwarmingTasks(
             'm', 'b', 1, {}))
     self.assertEqual(result, {})
@@ skipping 43 matching lines @@
             }
         },
         'failure_type': failure_type.TEST
     }
 
     expected_result = {
         'abc_test': ['Unittest2.Subtest1']
     }
     expected_failure_result_map = {
         'abc_test': {
-            'Unittest2.Subtest1': '%s/%s/%d' % (
+            'Unittest2.Subtest1': BaseBuildModel.CreateBuildId(
                 master_name, builder_name, build_number),
-            'Unittest3.Subtest2': '%s/%s/%d' % (
+            'Unittest3.Subtest2': BaseBuildModel.CreateBuildId(
                 master_name, builder_name, 1)
         }
     }
 
     result = (
         trigger_swarming_tasks_pipeline._GetStepsThatNeedToTriggerSwarmingTasks(
             master_name, builder_name, build_number, failure_info))
     analysis = WfAnalysis.Get(master_name, builder_name, build_number)
 
     self.assertEqual(result, expected_result)
@@ skipping 178 matching lines @@
       count = 0
       def run(self, *_):
         _MockedTriggerSwarmingTaskPipeline.count += 1
     self.mock(trigger_swarming_task_pipeline, 'TriggerSwarmingTaskPipeline',
               _MockedTriggerSwarmingTaskPipeline)
     pipeline = TriggerSwarmingTasksPipeline(
         master_name, builder_name, build_number, failure_info)
     pipeline.start()
     self.execute_queued_tasks()
     self.assertEqual(_MockedTriggerSwarmingTaskPipeline.count, 1)