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

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 import buildbucket_client
 from common.git_repository import GitRepository
+from model import wf_analysis_result_status
+from model.wf_analysis import WfAnalysis
 from model.wf_try_job import WfTryJob
 from pipeline_wrapper import pipeline_handlers
 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
 
@@ skipping 175 matching lines @@
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
     self._MockTriggerTryJobs(responses)
     self._MockGetTryJobs('1')
     self._MockGetChangeLog('rev2')
 
     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()
 
     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,
+                     wf_analysis_result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls,
+                     [expected_suspected_cl])
 
   def testSuccessfullyScheduleNewTryJobForTest(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     targeted_tests = {
         'a_test': ['TestSuite1.test1', 'TestSuite1.test3'],
         'b_test': [],  # Non-swarming test.
     }
 
@@ skipping 73 matching lines @@
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
     self._MockTriggerTryJobs(responses)
     self._MockGetTryJobs('2')
     self._MockGetChangeLog('rev1')
 
     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()
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
+    expected_suspected_cl = {
+        'revision': 'rev1',
+        'commit_position': '1',
+        'review_url': 'url_1',
+        'repo_name': 'chromium'
+    }
+
     expected_try_job_results = [
         {
             'report': {
                 'result': {
                     'rev1': {
                         'a_test': {
                             'status': 'failed',
                             'valid': True,
                             'failures': ['TestSuite1.test3']
                         },
                         'b_test': {
                             'status': 'passed',
                             'valid': True,
                             'failures': [],
                         },
                     }
                 },
                 'metadata': {
                     'regression_range_size': 2
                 }
             },
             'url': 'url',
             'try_job_id': '2',
             'culprit': {
                 'a_test': {
                     'tests': {
-                        'TestSuite1.test3': {
-                            'revision': 'rev1',
-                            'commit_position': '1',
-                            'review_url': 'url_1'
-                        }
+                        'TestSuite1.test3': expected_suspected_cl
                     }
                 }
             }
         }
     ]
 
     self.assertEqual(expected_try_job_results, try_job.test_results)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [expected_suspected_cl])