forked from mirrors/gecko-dev
d024a31639
This patch was generated via: $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f) $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f) The same set of commands are being applied to standalone taskgraph as well so they remain in sync. Differential Revision: https://phabricator.services.mozilla.com/D123234
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# 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/.
|
|
|
|
|
|
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()
|