fune/taskcluster/taskgraph/test/test_cron_util.py
Simon Fraser a589ef9901 Bug 1453067 Support for days (of week or month) in taskcluster cron r=dustin
Summary:
We need to run things less often than once a day, so adding support for
days to taskcluster cron. 'day' is the day of the month, 'weekday' is used
as a datetime.weekday (not isoweekday), or a string comparable to strftime('%A')
or strftime('%a')

Reviewers: dustin

Reviewed By: dustin

Bug #: 1453067

Differential Revision: https://phabricator.services.mozilla.com/D903

--HG--
extra : rebase_source : 65be900f383a2c4fd6240d4b8fb0eebef613281f
2018-04-12 12:39:20 +01:00

78 lines
3.4 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}))
def test_day_hour_minute(self):
params = {'time': datetime.datetime(2017, 1, 26, 16, 30, 0)}
self.assertFalse(match_utc(params, {'day': 25, 'hour': 16, 'minute': 30}))
self.assertTrue(match_utc(params, {'day': 26, 'hour': 16, 'minute': 30}))
self.assertFalse(match_utc(params, {'day': 26, 'hour': 16, 'minute': 0}))
def test_weekday_hour_minute(self):
params = {'time': datetime.datetime(2017, 1, 26, 16, 30, 0)}
self.assertFalse(match_utc(params, {'weekday': 'Wednesday', 'hour': 16, 'minute': 30}))
self.assertTrue(match_utc(params, {'weekday': 'Thursday', 'hour': 16, 'minute': 30}))
self.assertFalse(match_utc(params, {'weekday': 'Thursday', 'hour': 16, 'minute': 0}))
if __name__ == '__main__':
main()