forked from mirrors/gecko-dev
		
	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
		
	
			
		
			
				
	
	
		
			78 lines
		
	
	
	
		
			3.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			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()
 |