| 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)
|
|
|