1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
# -*- coding: utf-8 -*-
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import, print_function, unicode_literals
import unittest
import mozunit
from datetime import datetime
from taskgraph.util.time import (
InvalidString,
UnknownTimeMeasurement,
value_of,
json_time_from_now,
)
class FromNowTest(unittest.TestCase):
def test_invalid_str(self):
with self.assertRaises(InvalidString):
value_of("wtfs")
def test_missing_unit(self):
with self.assertRaises(InvalidString):
value_of("1")
def test_missing_unknown_unit(self):
with self.assertRaises(UnknownTimeMeasurement):
value_of("1z")
def test_value_of(self):
self.assertEqual(value_of("1s").total_seconds(), 1)
self.assertEqual(value_of("1 second").total_seconds(), 1)
self.assertEqual(value_of("1min").total_seconds(), 60)
self.assertEqual(value_of("1h").total_seconds(), 3600)
self.assertEqual(value_of("1d").total_seconds(), 86400)
self.assertEqual(value_of("1mo").total_seconds(), 2592000)
self.assertEqual(value_of("1 month").total_seconds(), 2592000)
self.assertEqual(value_of("1y").total_seconds(), 31536000)
with self.assertRaises(UnknownTimeMeasurement):
value_of("1m").total_seconds() # ambiguous between minute and month
def test_json_from_now_utc_now(self):
# Just here to ensure we don't raise.
json_time_from_now("1 years")
def test_json_from_now(self):
now = datetime(2014, 1, 1)
self.assertEqual(json_time_from_now("1 years", now), "2015-01-01T00:00:00Z")
self.assertEqual(json_time_from_now("6 days", now), "2014-01-07T00:00:00Z")
if __name__ == "__main__":
mozunit.main()
|
