Index: appengine/chromium_try_flakes/status/test/util_test.py |
diff --git a/appengine/chromium_try_flakes/status/test/util_test.py b/appengine/chromium_try_flakes/status/test/util_test.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f3ba974a3397acb8c11a8c4da461559d2085f06c |
--- /dev/null |
+++ b/appengine/chromium_try_flakes/status/test/util_test.py |
@@ -0,0 +1,44 @@ |
+# Copyright 2015 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 unittest |
+ |
+from model.flake import Flake |
+from status import util |
+ |
+ |
+class UtilTestCase(unittest.TestCase): |
+ def test_adds_occurrence_time_to_flake(self): |
+ flake = Flake(name='foo.bar', last_time_seen=datetime.datetime.min) |
+ |
+ now = datetime.datetime.utcnow() |
+ util.add_occurrence_time_to_flake(flake, now) |
+ |
+ self.assertEqual(flake.last_time_seen, now) |
+ self.assertEqual(flake.count_hour, 1) |
+ self.assertEqual(flake.count_day, 1) |
+ self.assertEqual(flake.count_week, 1) |
+ self.assertEqual(flake.count_month, 1) |
+ self.assertEqual(flake.last_hour, True) |
+ self.assertEqual(flake.last_day, True) |
+ self.assertEqual(flake.last_week, True) |
+ self.assertEqual(flake.last_month, True) |
+ |
+ def test_does_not_modify_flake(self): |
+ little_time_ago = datetime.datetime.utcnow() - datetime.timedelta(hours=2) |
+ flake = Flake(name='foo.bar', last_time_seen=little_time_ago) |
+ |
+ long_time_ago = datetime.datetime.utcnow() - datetime.timedelta(days=60) |
+ util.add_occurrence_time_to_flake(flake, long_time_ago) |
+ |
+ self.assertEqual(flake.last_time_seen, little_time_ago) |
+ self.assertEqual(flake.count_hour, 0) |
+ self.assertEqual(flake.count_day, 0) |
+ self.assertEqual(flake.count_week, 0) |
+ self.assertEqual(flake.count_month, 0) |
+ self.assertEqual(flake.last_hour, False) |
+ self.assertEqual(flake.last_day, False) |
+ self.assertEqual(flake.last_week, False) |
+ self.assertEqual(flake.last_month, False) |