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

appengine/findit/waterfall/test/swarming_tasks_to_try_job_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.
 
 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 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 import swarming_util
 from waterfall import trigger_swarming_task_pipeline
 from waterfall.swarming_task_request import SwarmingTaskRequest
 from waterfall.swarming_tasks_to_try_job_pipeline import (
     SwarmingTasksToTryJobPipeline)
 from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
 _ISOLATED_SERVER = 'https://isolateserver.appspot.com'
@@ skipping 154 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['rev1'] = MockedChangeLog('1', 'url_1')
       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._MockTriggerTryJobs(responses)
     self._MockGetTryJobs('1')
     self._MockGetChangeLog('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 = SwarmingTasksToTryJobPipeline(
         master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
         TryJobType.COMPILE)
     root_pipeline.start()
     self.execute_queued_tasks()
@@ skipping 116 matching lines @@
             'build': {
                 'id': '2',
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
     self._MockTriggerTryJobs(responses)
     self._MockGetTryJobs('2')
     self._MockGetChangeLog('rev1')
+    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 = SwarmingTasksToTryJobPipeline(
         master_name, builder_name, build_number, 'rev0', 'rev1', ['rev1'],
         TryJobType.TEST, None, targeted_tests)
     root_pipeline.start()
     self.execute_queued_tasks()
@@ skipping 38 matching lines @@
                     }
                 }
             }
         }
     ]
 
     self.assertEqual(expected_try_job_results, try_job.test_results)
     self.assertEqual(analysis.result_status,
                      result_status.FOUND_UNTRIAGED)
     self.assertEqual(analysis.suspected_cls, [expected_suspected_cl])