mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 17:33:54 +00:00
944 lines
30 KiB
Python
944 lines
30 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
import copy
|
|
import datetime
|
|
import logging
|
|
import os
|
|
import random
|
|
import time
|
|
|
|
import dateutil.parser as dateutil_parser
|
|
import datetime
|
|
|
|
# Import Salt Testing libs
|
|
from tests.support.case import ModuleCase
|
|
from tests.support.mixins import SaltReturnAssertsMixin
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mock import MagicMock, patch
|
|
from tests.support.unit import skipIf
|
|
import tests.integration as integration
|
|
|
|
# Import Salt libs
|
|
import salt.utils.schedule
|
|
import salt.utils.platform
|
|
|
|
from salt.modules.test import ping as ping
|
|
|
|
try:
|
|
import croniter # pylint: disable=W0611
|
|
HAS_CRONITER = True
|
|
except ImportError:
|
|
HAS_CRONITER = False
|
|
|
|
log = logging.getLogger(__name__)
|
|
ROOT_DIR = os.path.join(integration.TMP, 'schedule-unit-tests')
|
|
SOCK_DIR = os.path.join(ROOT_DIR, 'test-socks')
|
|
|
|
DEFAULT_CONFIG = salt.config.minion_config(None)
|
|
DEFAULT_CONFIG['conf_dir'] = ROOT_DIR
|
|
DEFAULT_CONFIG['root_dir'] = ROOT_DIR
|
|
DEFAULT_CONFIG['sock_dir'] = SOCK_DIR
|
|
DEFAULT_CONFIG['pki_dir'] = os.path.join(ROOT_DIR, 'pki')
|
|
DEFAULT_CONFIG['cachedir'] = os.path.join(ROOT_DIR, 'cache')
|
|
|
|
|
|
class SchedulerEvalTest(ModuleCase, SaltReturnAssertsMixin):
|
|
'''
|
|
Validate the pkg module
|
|
'''
|
|
def setUp(self):
|
|
with patch('salt.utils.schedule.clean_proc_dir', MagicMock(return_value=None)):
|
|
functions = {'test.ping': ping}
|
|
self.schedule = salt.utils.schedule.Schedule(copy.deepcopy(DEFAULT_CONFIG), functions, returners={})
|
|
self.schedule.opts['loop_interval'] = 1
|
|
|
|
self.schedule.opts['grains']['whens'] = {'tea time': '11/29/2017 12:00pm'}
|
|
|
|
def tearDown(self):
|
|
self.schedule.reset()
|
|
|
|
def test_eval(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
}
|
|
}
|
|
}
|
|
run_time2 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
run_time1 = run_time2 - datetime.timedelta(seconds=1)
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second before the run time
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time2)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time2)
|
|
|
|
def test_eval_multiple_whens(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_multiple_whens'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': [
|
|
'11/29/2017 4:00pm',
|
|
'11/29/2017 5:00pm',
|
|
],
|
|
}
|
|
}
|
|
}
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
run_time2 = dateutil_parser.parse('11/29/2017 5:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate run time1
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time1)
|
|
|
|
time.sleep(2)
|
|
|
|
# Evaluate run time2
|
|
self.schedule.eval(now=run_time2)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time2)
|
|
|
|
def test_eval_whens(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_whens'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': 'tea time',
|
|
}
|
|
}
|
|
}
|
|
run_time = dateutil_parser.parse('11/29/2017 12:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate run time1
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_loop_interval(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_loop_interval'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
}
|
|
}
|
|
}
|
|
# 30 second loop interval
|
|
LOOP_INTERVAL = random.randint(30, 59)
|
|
self.schedule.opts['loop_interval'] = LOOP_INTERVAL
|
|
|
|
run_time2 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time2 + datetime.timedelta(seconds=LOOP_INTERVAL))
|
|
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time2 + datetime.timedelta(seconds=LOOP_INTERVAL))
|
|
|
|
def test_eval_multiple_whens_loop_interval(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_multiple_whens_loop_interval'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': [
|
|
'11/29/2017 4:00pm',
|
|
'11/29/2017 5:00pm',
|
|
],
|
|
}
|
|
}
|
|
}
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# 30 second loop interval
|
|
LOOP_INTERVAL = random.randint(30, 59)
|
|
self.schedule.opts['loop_interval'] = LOOP_INTERVAL
|
|
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm') + datetime.timedelta(seconds=LOOP_INTERVAL)
|
|
run_time2 = dateutil_parser.parse('11/29/2017 5:00pm') + datetime.timedelta(seconds=LOOP_INTERVAL)
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time1)
|
|
|
|
time.sleep(2)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time2)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time2)
|
|
|
|
def test_eval_once(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_once'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'once': '2017-12-13T13:00:00',
|
|
}
|
|
}
|
|
}
|
|
run_time = dateutil_parser.parse('12/13/2017 1:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts['schedule'] = {}
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_once_loop_interval(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_once_loop_interval'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'once': '2017-12-13T13:00:00',
|
|
}
|
|
}
|
|
}
|
|
# Randomn second loop interval
|
|
LOOP_INTERVAL = random.randint(0, 59)
|
|
self.schedule.opts['loop_interval'] = LOOP_INTERVAL
|
|
|
|
# Run the job at the right plus LOOP_INTERVAL
|
|
run_time = dateutil_parser.parse('12/13/2017 1:00pm') + datetime.timedelta(seconds=LOOP_INTERVAL)
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate at the run time
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
@skipIf(not HAS_CRONITER, 'Cannot find croniter python module')
|
|
def test_eval_cron(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_cron'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'cron': '0 16 29 11 *',
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
run_time = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
with patch('croniter.croniter.get_next', MagicMock(return_value=run_time)):
|
|
self.schedule.eval(now=run_time)
|
|
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
@skipIf(not HAS_CRONITER, 'Cannot find croniter python module')
|
|
def test_eval_cron_loop_interval(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_cron_loop_interval'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'cron': '0 16 29 11 *',
|
|
}
|
|
}
|
|
}
|
|
# Randomn second loop interval
|
|
LOOP_INTERVAL = random.randint(0, 59)
|
|
self.schedule.opts['loop_interval'] = LOOP_INTERVAL
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
run_time = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
with patch('croniter.croniter.get_next', MagicMock(return_value=run_time)):
|
|
self.schedule.eval(now=run_time)
|
|
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_until(self):
|
|
'''
|
|
verify that scheduled job is skipped once the current
|
|
time reaches the specified until time
|
|
'''
|
|
job_name = 'test_eval_until'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'hours': '1',
|
|
'until': '11/29/2017 5:00pm',
|
|
}
|
|
}
|
|
}
|
|
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# Add job to schedule
|
|
self.schedule.delete_job('test_eval_until')
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# eval at 3:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 4:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 5:00pm, will not run
|
|
run_time = dateutil_parser.parse('11/29/2017 5:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_skip_reason'], 'until_passed')
|
|
self.assertEqual(ret['_skipped_time'], run_time)
|
|
|
|
def test_eval_after(self):
|
|
'''
|
|
verify that scheduled job is skipped until after the specified
|
|
time has been reached.
|
|
'''
|
|
job_name = 'test_eval_after'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'hours': '1',
|
|
'after': '11/29/2017 5:00pm',
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# eval at 3:00pm, will not run.
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_skip_reason'], 'after_not_passed')
|
|
self.assertEqual(ret['_skipped_time'], run_time)
|
|
|
|
# eval at 4:00pm, will not run.
|
|
run_time = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_skip_reason'], 'after_not_passed')
|
|
self.assertEqual(ret['_skipped_time'], run_time)
|
|
|
|
# eval at 5:00pm, will not run
|
|
run_time = dateutil_parser.parse('11/29/2017 5:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_skip_reason'], 'after_not_passed')
|
|
self.assertEqual(ret['_skipped_time'], run_time)
|
|
|
|
# eval at 6:00pm, will run
|
|
run_time = dateutil_parser.parse('11/29/2017 6:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_enabled(self):
|
|
'''
|
|
verify that scheduled job does not run
|
|
'''
|
|
job_name = 'test_eval_enabled'
|
|
job = {
|
|
'schedule': {
|
|
'enabled': True,
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
}
|
|
}
|
|
}
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time1)
|
|
|
|
def test_eval_enabled_key(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
when the enabled key is in place
|
|
https://github.com/saltstack/salt/issues/47695
|
|
'''
|
|
job_name = 'test_eval_enabled_key'
|
|
job = {
|
|
'schedule': {
|
|
'enabled': True,
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
}
|
|
}
|
|
}
|
|
run_time2 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
run_time1 = run_time2 - datetime.timedelta(seconds=1)
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second before the run time
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status('test_eval_enabled_key')
|
|
self.assertNotIn('_last_run', ret)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time2)
|
|
ret = self.schedule.job_status('test_eval_enabled_key')
|
|
self.assertEqual(ret['_last_run'], run_time2)
|
|
|
|
def test_eval_disabled(self):
|
|
'''
|
|
verify that scheduled job does not run
|
|
'''
|
|
job_name = 'test_eval_disabled'
|
|
job = {
|
|
'schedule': {
|
|
'enabled': False,
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
}
|
|
}
|
|
}
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate 1 second at the run time
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
self.assertEqual(ret['_skip_reason'], 'disabled')
|
|
|
|
def test_eval_run_on_start(self):
|
|
'''
|
|
verify that scheduled job is run when minion starts
|
|
'''
|
|
job_name = 'test_eval_run_on_start'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'hours': '1',
|
|
'run_on_start': True,
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
# eval at 3:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
def test_eval_splay(self):
|
|
'''
|
|
verify that scheduled job runs with splayed time
|
|
'''
|
|
job_name = 'job_eval_splay'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'seconds': '30',
|
|
'splay': '10',
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
with patch('random.randint', MagicMock(return_value=10)):
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# eval at 2:00:40pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:40pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_splay_range(self):
|
|
'''
|
|
verify that scheduled job runs with splayed time
|
|
'''
|
|
job_name = 'job_eval_splay_range'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'seconds': '30',
|
|
'splay': {'start': 5, 'end': 10},
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
with patch('random.randint', MagicMock(return_value=10)):
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# eval at 2:00:40pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:40pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_splay_global(self):
|
|
'''
|
|
verify that scheduled job runs with splayed time
|
|
'''
|
|
job_name = 'job_eval_splay_global'
|
|
job = {
|
|
'schedule': {
|
|
'splay': {'start': 5, 'end': 10},
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'seconds': '30',
|
|
}
|
|
}
|
|
}
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
with patch('random.randint', MagicMock(return_value=10)):
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# eval at 2:00:40pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:40pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_seconds(self):
|
|
'''
|
|
verify that scheduled job run mutiple times with seconds
|
|
'''
|
|
job_name = 'job_eval_seconds'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'seconds': '30',
|
|
}
|
|
}
|
|
}
|
|
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
next_run_time = run_time + datetime.timedelta(seconds=30)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 2:00:01pm, will not run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:01pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 2:00:30pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:30pm')
|
|
next_run_time = run_time + datetime.timedelta(seconds=30)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 2:01:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:01:00pm')
|
|
next_run_time = run_time + datetime.timedelta(seconds=30)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 2:01:30pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:01:30pm')
|
|
next_run_time = run_time + datetime.timedelta(seconds=30)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
def test_eval_minutes(self):
|
|
'''
|
|
verify that scheduled job run mutiple times with minutes
|
|
'''
|
|
job_name = 'job_eval_minutes'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'minutes': '30',
|
|
}
|
|
}
|
|
}
|
|
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
next_run_time = run_time + datetime.timedelta(minutes=30)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 2:00:01pm, will not run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:01pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 2:30:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:30:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 3:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 3:30:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 3:30:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_hours(self):
|
|
'''
|
|
verify that scheduled job run mutiple times with hours
|
|
'''
|
|
job_name = 'job_eval_hours'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'hours': '2',
|
|
}
|
|
}
|
|
}
|
|
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00pm')
|
|
next_run_time = run_time + datetime.timedelta(hours=2)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 2:00:01pm, will not run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:01pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 4:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 4:00:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 6:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 6:00:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 8:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 8:00:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
|
|
def test_eval_days(self):
|
|
'''
|
|
verify that scheduled job run mutiple times with days
|
|
'''
|
|
job_name = 'job_eval_days'
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'days': '2',
|
|
'dry_run': True
|
|
}
|
|
}
|
|
}
|
|
|
|
if salt.utils.platform.is_darwin():
|
|
job['schedule'][job_name]['dry_run'] = True
|
|
|
|
# Add job to schedule
|
|
self.schedule.opts.update(job)
|
|
|
|
# eval at 11/23/2017 2:00pm to prime, simulate minion start up.
|
|
run_time = dateutil_parser.parse('11/23/2017 2:00pm')
|
|
next_run_time = run_time + datetime.timedelta(days=2)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 11/25/2017 2:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/25/2017 2:00:00pm')
|
|
next_run_time = run_time + datetime.timedelta(days=2)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
# eval at 11/26/2017 2:00:00pm, will not run.
|
|
run_time = dateutil_parser.parse('11/26/2017 2:00:00pm')
|
|
last_run_time = run_time - datetime.timedelta(days=1)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], last_run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 11/27/2017 2:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/27/2017 2:00:00pm')
|
|
next_run_time = run_time + datetime.timedelta(days=2)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 11/28/2017 2:00:00pm, will not run.
|
|
run_time = dateutil_parser.parse('11/28/2017 2:00:00pm')
|
|
last_run_time = run_time - datetime.timedelta(days=1)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], last_run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
time.sleep(2)
|
|
|
|
# eval at 11/29/2017 2:00:00pm, will run.
|
|
run_time = dateutil_parser.parse('11/29/2017 2:00:00pm')
|
|
next_run_time = run_time + datetime.timedelta(days=2)
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time)
|
|
self.assertEqual(ret['_next_fire_time'], next_run_time)
|
|
|
|
def test_eval_when_splay(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_when_splay'
|
|
splay = 300
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': '11/29/2017 4:00pm',
|
|
'splay': splay
|
|
}
|
|
}
|
|
}
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
run_time2 = run_time1 + datetime.timedelta(seconds=splay)
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
with patch('random.randint', MagicMock(return_value=splay)):
|
|
# Evaluate to prime
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# Evaluate at expected runtime1, should not run
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
|
|
# Evaluate at expected runtime2, should run
|
|
self.schedule.eval(now=run_time2)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertEqual(ret['_last_run'], run_time2)
|
|
|
|
def test_eval_when_splay_in_past(self):
|
|
'''
|
|
verify that scheduled job runs
|
|
'''
|
|
job_name = 'test_eval_when_splay_in_past'
|
|
splay = 300
|
|
job = {
|
|
'schedule': {
|
|
job_name: {
|
|
'function': 'test.ping',
|
|
'when': ['11/29/2017 6:00am'],
|
|
'splay': splay
|
|
}
|
|
}
|
|
}
|
|
run_time1 = dateutil_parser.parse('11/29/2017 4:00pm')
|
|
|
|
# Add the job to the scheduler
|
|
self.schedule.opts.update(job)
|
|
|
|
# Evaluate to prime
|
|
run_time = dateutil_parser.parse('11/29/2017 3:00pm')
|
|
self.schedule.eval(now=run_time)
|
|
ret = self.schedule.job_status(job_name)
|
|
|
|
# Evaluate at expected runtime1, should not run
|
|
# and _next_fire_time should be None
|
|
self.schedule.eval(now=run_time1)
|
|
ret = self.schedule.job_status(job_name)
|
|
self.assertNotIn('_last_run', ret)
|
|
self.assertEqual(ret['_next_fire_time'], None)
|