[Findit] Modify tryjob pipelines to trigger try jobs for test failure.

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 testing_utils import testing
 
 from common import buildbucket_client
+from common.git_repository import GitRepository
 from model import wf_analysis_status
 from model.wf_try_job import WfTryJob
 from pipeline_wrapper import pipeline_handlers
 from waterfall import waterfall_config
 from waterfall.try_job_pipeline import TryJobPipeline
 
 
 class TryJobPipelineTest(testing.AppengineTestCase):
   app_module = pipeline_handlers._APP
 
   def _Mock_GetTrybotForWaterfallBuilder(self, *_):
     def Mocked_GetTrybotForWaterfallBuilder(*_):
       return 'linux_chromium_variable', 'master.tryserver.chromium.linux'
     self.mock(waterfall_config, 'GetTrybotForWaterfallBuilder',
               Mocked_GetTrybotForWaterfallBuilder)
 
   def _Mock_TriggerTryJobs(self, responses):
     def Mocked_TriggerTryJobs(*_):
-      compile_results = []
+      try_job_results = []
       for response in responses:
         if response.get('error'):  # pragma: no cover
-          compile_results.append((
+          try_job_results.append((
               buildbucket_client.BuildbucketError(response['error']), None))
         else:
-          compile_results.append((
+          try_job_results.append((
               None, buildbucket_client.BuildbucketBuild(response['build'])))
-      return compile_results
+      return try_job_results
     self.mock(buildbucket_client, 'TriggerTryJobs', Mocked_TriggerTryJobs)
 
   def _Mock_GetTryJobs(self, build_id):
     def Mocked_GetTryJobs(*_):
       data = {
           '1': {
               'build': {
                   'id': '1',
                   'url': 'url',
                   'status': 'COMPLETED',
@@ skipping 12 matching lines @@
                   })
               }
           },
           '3': {
               'error': {
                   'reason': 'BUILD_NOT_FOUND',
                   'message': 'message',
               }
           }
       }
-      compile_results = []
+      try_job_results = []
       build_error = data.get(build_id)
       if build_error.get('error'):  # pragma: no cover
-        compile_results.append((
+        try_job_results.append((
             buildbucket_client.BuildbucketError(build_error['error']), None))
       else:
-        compile_results.append((
+        try_job_results.append((
             None, buildbucket_client.BuildbucketBuild(build_error['build'])))
-      return compile_results
+      return try_job_results
     self.mock(buildbucket_client, 'GetTryJobs', Mocked_GetTryJobs)
 
+  def _Mock_GetChangeLog(self, revision):
+    def Mocked_GetChangeLog(*_):
+      class MockedChangeLog(object):
+        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', Mocked_GetChangeLog)
+
   def testSuccessfullyScheduleNewTryJobForCompile(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
-    good_revision = 'rev1'
-    bad_revision = 'rev2'
 
     responses = [
         {
             'build': {
                 'id': '1',
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
     self._Mock_GetTrybotForWaterfallBuilder(master_name, builder_name)
     self._Mock_TriggerTryJobs(responses)
     self._Mock_GetTryJobs('1')
+    self._Mock_GetChangeLog('rev2')
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
 
     root_pipeline = TryJobPipeline(
-        master_name, builder_name, build_number, good_revision, bad_revision,
-        [])
+        master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
+        'compile', [])
     root_pipeline.start()
     self.execute_queued_tasks()
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
-    expected_compile_results = [
+    expected_try_job_results = [
         {
             'report': {
                 'result': {
                     'rev1': 'passed',
                     'rev2': 'failed'
                 },
                 'metadata': {
                     'regression_range_size': 2
                 }
             },
             'url': 'url',
             'try_job_id': '1',
+            'culprit': {
+                'revision': 'rev2',
+                'commit_position': '2',
+                'review_url': 'url_2'
+            }
         }
     ]
-    self.assertEqual(expected_compile_results, try_job.compile_results)
+    self.assertEqual(expected_try_job_results, try_job.compile_results)
 
   def testPipelineAbortedWithTryJobResult(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
 
     root_pipeline = TryJobPipeline(
-        master_name, builder_name, build_number, 'rev1', 'rev2', [])
+        master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
+        'compile', [])
     root_pipeline._LogUnexpectedAbort(True)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual(wf_analysis_status.ERROR, try_job.status)
 
   def testPipelineAbortedWithOutTryJobResult(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
 
     root_pipeline = TryJobPipeline(
-        master_name, builder_name, build_number, 'rev1', 'rev2', [])
+        master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'],
+        'compile', [])
     root_pipeline._LogUnexpectedAbort(True)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertIsNone(try_job)