[Findit] Add settings for actions after culprits or suspects are identified.

appengine/findit/waterfall/test/try_job_pipeline_test.py

 # Copyright 2015 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 analysis_status
 from model import result_status
 from model.wf_analysis import WfAnalysis
 from model.wf_try_job import WfTryJob
+from waterfall import send_notification_for_culprit_pipeline
 from waterfall.test import wf_testcase
 from waterfall.try_job_pipeline import TryJobPipeline
 from waterfall.try_job_type import TryJobType
 
 
 class TryJobPipelineTest(wf_testcase.WaterfallTestCase):
   app_module = pipeline_handlers._APP
 
   def _Mock_TriggerTryJobs(self, responses):
     def MockedTriggerTryJobs(*_):
@@ skipping 55 matching lines @@
 
         def __init__(self, commit_position, code_review_url):
           self.commit_position = commit_position
           self.code_review_url = code_review_url
 
       mock_change_logs = {}
       mock_change_logs['rev2'] = MockedChangeLog('2', 'url_2')
       return mock_change_logs.get(revision)
     self.mock(GitRepository, 'GetChangeLog', MockedGetChangeLog)
 
+  def _Mock_SendNotificationForCulpritPipeline(self):
+    class Mocked_Pipeline(BasePipeline):
+      def run(self, *args, **kwargs):  # unused arg - pylint: disable=W0612
+        pass
+    self.mock(send_notification_for_culprit_pipeline,
+              'SendNotificationForCulpritPipeline', Mocked_Pipeline)
+
   def testSuccessfullyScheduleNewTryJobForCompile(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
 
     responses = [
         {
             'build': {
                 'id': '1',
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
 
     self._Mock_TriggerTryJobs(responses)
     self._Mock_GetTryJobs('1')
     self._Mock_GetChangeLog('rev2')
+    self._Mock_SendNotificationForCulpritPipeline()
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
     analysis = WfAnalysis.Create(master_name, builder_name, build_number)
     analysis.put()
 
     root_pipeline = TryJobPipeline(
         master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
         TryJobType.COMPILE, [])
     root_pipeline.start()
     self.execute_queued_tasks()
@@ skipping 52 matching lines @@
     build_number = 1
 
     root_pipeline = TryJobPipeline(
         master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
         TryJobType.COMPILE, [])
     root_pipeline._LogUnexpectedAbort(True)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertIsNone(try_job)