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

appengine/findit/model/flake/test/master_flake_analysis_test.py

Index: appengine/findit/model/flake/test/master_flake_analysis_test.py
diff --git a/appengine/findit/model/flake/test/master_flake_analysis_test.py b/appengine/findit/model/flake/test/master_flake_analysis_test.py
index aff35faee5957634641f974dfcb6433c4f7569b5..edc5e946dc0f51c58ac10160454a1782632e7d5f 100644
--- a/appengine/findit/model/flake/test/master_flake_analysis_test.py
+++ b/appengine/findit/model/flake/test/master_flake_analysis_test.py
@@ -7,6 +7,8 @@ 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
 
@@ -86,6 +88,18 @@ class MasterFlakeAnalysisTest(unittest.TestCase):
     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 = [{}]