DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (1aeaa33a64f9)

VCS Links

Line Code
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 58 59 60
# -*- 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()