[Findit] Updating flakiness dashboard to be able to triage results

appengine/findit/model/flake/test/master_flake_analysis_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.
 
 from datetime import datetime
 
 import unittest
 
 from model import analysis_status
+from model import result_status
+from model import triage_status
 from model.flake.master_flake_analysis import DataPoint
 from model.flake.master_flake_analysis import MasterFlakeAnalysis
 
 
 class MasterFlakeAnalysisTest(unittest.TestCase):
 
   def testMasterFlakeAnalysisStatusIsCompleted(self):
     for status in (analysis_status.COMPLETED, analysis_status.ERROR):
       analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
       analysis.status = status
@@ skipping 59 matching lines @@
   def testMasterFlakeAnalysisStatusDescriptionError(self):
     analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
     analysis.status = analysis_status.ERROR
     self.assertEqual('Error', analysis.status_description)
 
   def testMasterFlakeAnalysisStepTestName(self):
     analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
     self.assertEqual('s', analysis.step_name)
     self.assertEqual('t', analysis.test_name)
 
+  def testMasterFlakeAnalysisUpdateTriageResultCorrect(self):
+    analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
+    analysis.UpdateTriageResult(
+        triage_status.TRIAGED_CORRECT, {'build_number': 100}, 'test')
+    self.assertEqual(analysis.result_status, result_status.FOUND_CORRECT)
+
+  def testMasterFlakeAnalysisUpdateTriageResultIncorrect(self):
+    analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
+    analysis.UpdateTriageResult(
+        triage_status.TRIAGED_INCORRECT, {'build_number': 100}, 'test')
+    self.assertEqual(analysis.result_status, result_status.FOUND_INCORRECT)
+
   def testReset(self):
     analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
     analysis.swarming_rerun_results = [{}]
     analysis.status = analysis_status.RUNNING
     analysis.correct_regression_range = True
     analysis.correct_culprit = False
     analysis.correct_culprit = None
     analysis.data_points = [DataPoint()]
     analysis.suspected_flake_build_number = 123
     analysis.Reset()
@@ skipping 17 matching lines @@
 
   def testGetIterationsToRerun(self):
     cases = [
         (-1, {}),
         (5, {'key': 'value', 'iterations_to_rerun': 5}),
     ]
     for expected_rerun, algorithm_parameters in cases:
       analysis = MasterFlakeAnalysis.Create('m', 'b', 123, 's', 't')
       analysis.algorithm_parameters = algorithm_parameters
       self.assertEqual(expected_rerun, analysis.iterations_to_rerun)