[Findit] UI change and triage change for cl level trige.

appengine/findit/handlers/test/build_failure_test.py

 # Copyright 2014 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 os
 import re
 
 from google.appengine.ext import testbed
 
 import webapp2
 import webtest
 
 from handlers import build_failure
 from handlers import handlers_util
 from handlers import result_status
+from model import analysis_approach_type
 from model.base_build_model import BaseBuildModel
 from model.wf_analysis import WfAnalysis
 from model.wf_try_job import WfTryJob
 from model import analysis_status
+from model import suspected_cl_status
+from model.base_build_model import BaseBuildModel
 from model.wf_analysis import WfAnalysis
+from model.wf_suspected_cl import WfSuspectedCL
+from model.wf_try_job import WfTryJob
 from waterfall import buildbot
 from waterfall.test import wf_testcase
 
 # Root directory appengine/findit.
 ROOT_DIR = os.path.join(os.path.dirname(__file__),
                         os.path.pardir, os.path.pardir)
 
 SAMPLE_TRY_JOB_INFO = {
     'm/b/119': {
         'step1 on platform': {
@@ skipping 103 matching lines @@
     analysis.triage_history = [
         {
             'triage_timestamp': 1438380761,
             'user_name': 'test',
             'result_status': 'dummy status',
             'version': 'dummy version',
         }
     ]
     self.assertIsNone(build_failure._GetTriageHistory(analysis))
 
+  def testGetCLDict(self):
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev2',
+            'url': 'url',
+            'commit_position': 123
+        }
+    ]
+    analysis.put()
+    cl_info = 'chromium/rev1'
+    self.assertEqual({}, build_failure._GetCLDict(analysis, cl_info))
+
+  def testGetCLDictNone(self):
+    self.assertEqual({}, build_failure._GetCLDict(None, None))
+
   def testGetTriageHistoryWhenUserIsAdmin(self):
     analysis = WfAnalysis.Create('m', 'b', 1)
     analysis.status = analysis_status.COMPLETED
+    analysis.suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev1',
+            'url': 'url',
+            'commit_position': 123
+        }
+    ]
     analysis.triage_history = [
         {
             'triage_timestamp': 1438380761,
             'user_name': 'test',
             'result_status': 'dummy status',
             'version': 'dummy version',
+            'triaged_cl': 'chromium/rev1'
         }
     ]
     self.mock_current_user(user_email='test@chromium.org', is_admin=True)
     self.assertEqual(1, len(build_failure._GetTriageHistory(analysis)))
 
   def testInvalidBuildUrl(self):
     build_url = 'abc'
     self.assertRaisesRegexp(
         webtest.app.AppError,
         re.compile('.*501 Not Implemented.*Url "%s" '
                    'is not pointing to a build.*' % build_url,
                    re.MULTILINE | re.DOTALL),
         self.test_app.get, '/build-failure', params={'url': build_url})
 
   def testNonAdminCanViewAnalysisOfFailureOnUnsupportedMaster(self):
     master_name = 'm2'
     builder_name = 'b 1'
     build_number = 123
     build_url = buildbot.CreateBuildUrl(
         master_name, builder_name, build_number)
 
     analysis = WfAnalysis.Create(master_name, builder_name, build_number)
     analysis.status = analysis_status.COMPLETED
+    analysis.suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'r99_2',
+            'commit_position': None,
+            'url': None,
+        }
+    ]
     analysis.put()
 
     response = self.test_app.get('/build-failure',
                                  params={'url': build_url})
     self.assertEquals(200, response.status_int)
     self.assertEqual(0, len(self.taskqueue_stub.get_filtered_tasks()))
 
   def testNonAdminCannotRequestAnalysisOfFailureOnUnsupportedMaster(self):
     master_name = 'm2'
     builder_name = 'b 1'
@@ skipping 546 matching lines @@
             },
             {
                 'step_name': 'steps',
             },
         ]
     }
     analysis.failure_result_map = {
         'compile': 'm/b/122',
     }
     analysis.status = analysis_status.COMPLETED
+    analysis.suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev',
+            'commit_position': 122,
+            'url': None
+        }
+    ]
     analysis.put()
 
     try_job = WfTryJob.Create('m', 'b', 122)
     try_job.status = analysis_status.COMPLETED
     try_job.compile_results = [
         {
             'url': 'build/url',
             'culprit': {
                 'compile': {
                     'revision': 'rev',
                 }
             }
         }
     ]
     try_job.put()
 
+    suspected_cl = WfSuspectedCL.Create('chromium', 'rev', 122)
+    suspected_cl.builds = {
+        'm/b/123': {
+            'failure_type': 'compile',
+            'failures': None,
+            'status': suspected_cl_status.CORRECT,
+            'approach': analysis_approach_type.HEURISTIC,
+            'top_score': 5,
+            'Confidence': 97.9
+        }
+    }
+    suspected_cl.put()
+
     expected_try_job_result = {
         'status': 'completed',
         'url': 'build/url',
         'completed': True,
         'culprit': {
             'revision': 'rev',
         },
         'failed': False,
     }
 
+    expected_suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev',
+            'commit_position': 122,
+            'url': None,
+            'status': suspected_cl_status.CORRECT
+        }
+    ]
+
     build_url = buildbot.CreateBuildUrl('m', 'b', 123)
     response = self.test_app.get('/build-failure',
                                  params={'url': build_url, 'format': 'json'})
 
     self.assertEquals(200, response.status_int)
     self.assertEqual(expected_try_job_result, response.json_body['try_job'])
+    self.assertEqual(
+        expected_suspected_cls, response.json_body['suspected_cls'])
+
+  def testGetAllSuspectedCLsAndCheckStatus(self):
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = 123
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev',
+            'commit_position': 122,
+            'url': None
+        }
+    ]
+    analysis.put()
+    suspected_cl = WfSuspectedCL.Create('chromium', 'rev', 122)
+    suspected_cl.builds = {
+        'm/b/122': {
+            'failure_type': 'compile',
+            'failures': None,
+            'status': suspected_cl_status.CORRECT,
+            'approach': analysis_approach_type.HEURISTIC,
+            'top_score': 5,
+            'Confidence': 97.9
+        }
+    }
+    suspected_cl.put()
+
+    expected_suspected_cls = [
+        {
+            'repo_name': 'chromium',
+            'revision': 'rev',
+            'commit_position': 122,
+            'url': None,
+            'status': None
+        }
+    ]
+
+    suspected_cls = build_failure._GetAllSuspectedCLsAndCheckStatus(
+        master_name, builder_name, build_number, analysis)
+    self.assertEqual(
+        expected_suspected_cls, suspected_cls)
+