[Findit] Cloud Endpoints API to receive reports on flakes.

appengine/findit/model/flake/test/flake_analysis_request_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.flake.flake_analysis_request import BuildStep
+from model.flake.flake_analysis_request import FlakeAnalysisRequest
+
+
+class FlakeAnalysisRequestTest(unittest.TestCase):
+
+  def testStripMasterPrefix(self):
+    cases = {
+        'master.tryserver.chromium.linux': 'tryserver.chromium.linux',
+        'chromium.linux': 'chromium.linux',
+    }
+    for original_name, expected_name in cases.iteritems():
+      self.assertEqual(expected_name,
+                       BuildStep._StripMasterPrefix(original_name))
+
+  def testAddBuildStep(self):
+    t1 = datetime(2016, 10, 1, 0, 0, 0)
+    t2 = datetime(2016, 10, 2, 0, 0, 0)
+    t3 = datetime(2016, 10, 2, 1, 0, 0)
+    t4 = datetime(2016, 10, 2, 0, 30, 0)
+    request = FlakeAnalysisRequest.Create('flaky_test', False, 123)
+    self.assertTrue(request.AddBuildStep('m', 'b1', 1, 's', t1))
+    self.assertTrue(request.AddBuildStep('m', 'b2', 10, 's', t2))
+    self.assertFalse(request.AddBuildStep('m', 'b2', 11, 's', t3))
+    self.assertTrue(request.AddBuildStep('m', 'b2', 9, 's', t4))
+    self.assertEqual(2, len(request.build_steps), request.build_steps)
+    self.assertEqual(BuildStep.Create('m', 'b1', 1, 's', t1),
+                     request.build_steps[0])
+    self.assertEqual(BuildStep.Create('m', 'b2', 9, 's', t4),
+                     request.build_steps[1])