gecko-dev/taskcluster/taskgraph/test/test_cron_util.py

67 строки
2.5 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/.
from __future__ import absolute_import, unicode_literals
import datetime
import unittest
from mozunit import main
from taskgraph.cron.util import (
match_utc,
)
class TestMatchUtc(unittest.TestCase):
def test_hour_minute(self):
params = {'time': datetime.datetime(2017, 1, 26, 16, 30, 0)}
self.assertFalse(match_utc(params, hour=4, minute=30))
self.assertTrue(match_utc(params, hour=16, minute=30))
self.assertFalse(match_utc(params, hour=16, minute=0))
def test_hour_only(self):
params = {'time': datetime.datetime(2017, 1, 26, 16, 0, 0)}
self.assertFalse(match_utc(params, hour=0))
self.assertFalse(match_utc(params, hour=4))
self.assertTrue(match_utc(params, hour=16))
params = {'time': datetime.datetime(2017, 1, 26, 16, 15, 0)}
self.assertFalse(match_utc(params, hour=0))
self.assertFalse(match_utc(params, hour=4))
self.assertTrue(match_utc(params, hour=16))
params = {'time': datetime.datetime(2017, 1, 26, 16, 30, 0)}
self.assertFalse(match_utc(params, hour=0))
self.assertFalse(match_utc(params, hour=4))
self.assertTrue(match_utc(params, hour=16))
params = {'time': datetime.datetime(2017, 1, 26, 16, 45, 0)}
self.assertFalse(match_utc(params, hour=0))
self.assertFalse(match_utc(params, hour=4))
self.assertTrue(match_utc(params, hour=16))
def test_minute_only(self):
params = {'time': datetime.datetime(2017, 1, 26, 13, 0, 0)}
self.assertTrue(match_utc(params, minute=0))
self.assertFalse(match_utc(params, minute=15))
self.assertFalse(match_utc(params, minute=30))
self.assertFalse(match_utc(params, minute=45))
def test_zeroes(self):
params = {'time': datetime.datetime(2017, 1, 26, 0, 0, 0)}
self.assertTrue(match_utc(params, minute=0))
self.assertTrue(match_utc(params, hour=0))
self.assertFalse(match_utc(params, hour=1))
self.assertFalse(match_utc(params, minute=15))
self.assertFalse(match_utc(params, minute=30))
self.assertFalse(match_utc(params, minute=45))
def test_invalid_minute(self):
params = {'time': datetime.datetime(2017, 1, 26, 13, 0, 0)}
self.assertRaises(Exception, lambda:
match_utc(params, minute=1))
if __name__ == '__main__':
main()