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

appengine/findit/model/test/base_triaged_model_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 mock
+import unittest
+
+from google.appengine.api import users
+
+from model.base_triaged_model import TriagedModel
+from model.base_triaged_model import TriageResult
+
+
+class _DummyModel(TriagedModel):
+  completed = True
+
+
+class TriagedModelTest(unittest.TestCase):
+
+  def testUpdateTriageResult(self):
+    triage_result = 1
+    suspect_info = 'abcd'
+    user_name = 'test'
+
+    model = _DummyModel()
+    model.UpdateTriageResult(1, suspect_info, user_name)
+    self.assertEqual(len(model.triage_history), 1)
+    self.assertEqual(model.triage_history[0].triage_result, triage_result)
+    self.assertEqual(model.triage_history[0].suspect_info, suspect_info)
+    self.assertEqual(model.triage_history[0].user_name, user_name)
+
+  @mock.patch.object(users, 'is_current_user_admin', return_value=True)
+  def testGetTriageHistory(self, _):
+    suspect_info = 'abcd'
+    user_name = 'test'
+
+    model = _DummyModel()
+    result = TriageResult()
+    result.triage_result = 1
+    result.user_name = user_name
+    result.suspect_info = suspect_info
+    model.triage_history.append(result)
+
+    triage_history = TriagedModel.GetTriageHistory(model)
+
+    # Because TriageResult's triage_time uses auto_now=True, a direct dict
+    # comparison will always fail. Instead compare each relevant field
+    # individually.
+    self.assertEqual(len(triage_history), 1)
+    self.assertEqual(triage_history[0].get('user_name'), user_name)
+    self.assertEqual(triage_history[0].get('triage_result'), 'Incorrect')
+    self.assertEqual(triage_history[0].get('suspect_info'), suspect_info)
+