[Findit] Adding support for triaging suspected builds from flake analysis

appengine/findit/handlers/flake/check_flake.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.
 
+from google.appengine.api import users
+
 from common import auth_util
 from common import constants
 from common import time_util
 from common.base_handler import BaseHandler
 from common.base_handler import Permission
 from model import analysis_status
+from model import triage_status
 from waterfall.flake import initialize_flake_pipeline
 from waterfall.flake import triggering_sources
 
 
+def _GetSuspectedFlakeAnalysisAndTriageResult(analysis):
+  if analysis.suspected_flake_build_number is not None:
+    return {
+        'build_number': analysis.suspected_flake_build_number,
+        'triage_result': (
+            analysis.triage_history[-1].triage_result if analysis.triage_history
+            else triage_status.UNTRIAGED)
+    }
+
+  return {}
+
+
 class CheckFlake(BaseHandler):
   PERMISSION_LEVEL = Permission.ANYONE
 
   def HandleGet(self):
     master_name = self.request.get('master_name').strip()
     builder_name = self.request.get('builder_name').strip()
     build_number = int(self.request.get('build_number', '0').strip())
     step_name = self.request.get('step_name').strip()
     test_name = self.request.get('test_name').strip()
 
@@ skipping 19 matching lines @@
           'template': 'error.html',
           'data': {
               'error_message':
                   ('You could schedule an analysis for flaky test only after '
                    'you login with google.com account.'),
               'login_url': self.GetLoginUrl(),
           },
           'return_code': 401,
       }
 
+    suspected_flake = _GetSuspectedFlakeAnalysisAndTriageResult(analysis)
+
     data = {
         'pass_rates': [],
         'analysis_status': analysis.status_description,
-        'suspected_flake_build_number': (
-            analysis.suspected_flake_build_number),
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'step_name': step_name,
         'test_name': test_name,
+        'version_number': analysis.version_number,
+        'suspected_flake': suspected_flake,
         'request_time': time_util.FormatDatetime(
             analysis.request_time),
         'task_number': len(analysis.data_points),
         'error': analysis.error_message,
         'iterations_to_rerun': analysis.iterations_to_rerun,
+        'show_debug_info': self._ShowDebugInfo()
     }
 
+    if (users.is_current_user_admin() and analysis.completed and
+        analysis.triage_history):
+      data['triage_history'] = analysis.GetTriageHistory()
+
     data['pending_time'] = time_util.FormatDuration(
         analysis.request_time,
         analysis.start_time or time_util.GetUTCNow())
     if analysis.status != analysis_status.PENDING:
       data['duration'] = time_util.FormatDuration(
           analysis.start_time,
           analysis.end_time or time_util.GetUTCNow())
 
     coordinates = []
     for data_point in analysis.data_points:
       coordinates.append([data_point.build_number, data_point.pass_rate])
 
     # Order by build number from earliest to latest.
     coordinates.sort(key=lambda x: x[0])
 
     data['pass_rates'] = coordinates
     return {
         'template': 'flake/result.html',
         'data': data
     }