[Findit] Adding Try Job suspected CLs and result status to analysis

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 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.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):
@@ skipping 76 matching lines @@
                 'status': 'SCHEDULED',
             }
         }
     ]
 
     self._Mock_TriggerTryJobs(responses)
     self._Mock_GetTryJobs('1')
     self._Mock_GetChangeLog('rev2')
 
     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()
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
+    expected_suspected_cl = {
+        'revision': 'rev2',
+        'commit_position': '2',
+        'review_url': 'url_2',
+        'repo_name': 'chromium'
+    }
 
     expected_try_job_results = [
         {
             'report': {
                 'result': {
                     'rev1': 'passed',
                     'rev2': 'failed'
                 },
                 'metadata': {
                     'regression_range_size': 2
                 }
             },
             'url': 'url',
             'try_job_id': '1',
             'culprit': {
-                'compile': {
-                    'revision': 'rev2',
-                    'commit_position': '2',
-                    'review_url': 'url_2'
-                }
+                'compile': expected_suspected_cl
             }
         }
     ]
     self.assertEqual(expected_try_job_results, try_job.compile_results)
+    self.assertEqual(analysis.result_status,
+                     result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls,
+                     [expected_suspected_cl])
 
   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', ['rev2'],
         TryJobType.COMPILE, [])
     root_pipeline._LogUnexpectedAbort(True)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual(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', ['rev2'],
         TryJobType.COMPILE, [])
     root_pipeline._LogUnexpectedAbort(True)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertIsNone(try_job)