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

appengine/findit/handlers/flake/test/check_flake_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 datetime
+import mock
 import re
 
 import webapp2
 import webtest
 
+from google.appengine.api import users
+
 from handlers.flake import check_flake
 from model.flake.master_flake_analysis import DataPoint
 from model.flake.master_flake_analysis import MasterFlakeAnalysis
 from model import analysis_status
 from model.analysis_status import STATUS_TO_DESCRIPTION
 from waterfall.test import wf_testcase
 
 
 class CheckFlakeTest(wf_testcase.WaterfallTestCase):
   app_module = webapp2.WSGIApplication([
@@ skipping 76 matching lines @@
         'builder_name': builder_name,
         'build_number': int(build_number),
         'step_name': step_name,
         'test_name': test_name,
         'request_time': '2016-10-01 12:10:00 UTC',
         'task_number': 1,
         'error': None,
         'iterations_to_rerun': 100,
         'pending_time': '00:00:05',
         'duration': '00:59:55',
-        'suspected_flake_build_number': 100,
+        'suspected_flake': {
+            'build_number': 100,
+            'triage_result': 0
+        },
+        'version_number': 1,
+        'triage_history': None,
+        'show_debug_info': False
     }
 
     self.assertEquals(200, response.status_int)
     self.assertEqual(expected_check_flake_result, response.json_body)
+
+  @mock.patch.object(users, 'is_current_user_admin', return_value=True)
+  def testGetTriageHistory(self, _):
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = '123'
+    step_name = 's'
+    test_name = 't'
+    suspected_flake_build_number = '100'
+    triage_result = 2
+    user_name = 'test'
+
+    analysis = MasterFlakeAnalysis.Create(
+        master_name, builder_name, build_number, step_name, test_name)
+    analysis.status = analysis_status.COMPLETED
+    analysis.suspected_flake_build_number = 100
+    analysis.Save()
+    analysis.UpdateTriageResult(
+        triage_result, suspected_flake_build_number, 'test')
+
+    response = self.test_app.get('/waterfall/check-flake', params={
+        'master_name': master_name,
+        'builder_name': builder_name,
+        'build_number': build_number,
+        'step_name': step_name,
+        'test_name': test_name,
+        'format': 'json'})
+
+    # Because TriagedResult uses auto_now=True, a direct dict comparison will
+    # always fail. Instead only compare the relevant fields for trige_history.
+    triage_history = response.json_body.get('triage_history')
+    self.assertEqual(len(triage_history), 1)
+    self.assertEqual(triage_history[0].get('triage_result'), 'Correct')
+    self.assertEqual(triage_history[0].get('user_name'), user_name)
+    self.assertEqual(triage_history[0].get('suspect_info'),
+                     suspected_flake_build_number)