def test_update_job_change_type(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
self.login(AdminFactory())
response = self.put(
url_for('api.job', id=task.id), {
'name': task.name,
'description': task.description,
'task': task.task,
'interval': {
'every': 5,
'period': 'minutes',
}
})
self.assert200(response)
self.assertEqual(response.json['id'], str(task.id))
self.assertEqual(response.json['name'], task.name)
self.assertEqual(response.json['task'], task.task)
self.assertEqual(response.json['description'], task.description)
self.assertEqual(response.json['interval']['every'], 5)
self.assertEqual(response.json['interval']['period'], 'minutes')
self.assertIsNone(response.json['crontab'])
def test_update_job(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
self.login(AdminFactory())
response = self.put(
url_for('api.job', id=task.id), {
'name': task.name,
'description': 'New description',
'task': task.task,
'crontab': task.crontab.to_json()
})
self.assert200(response)
self.assertEqual(response.json['id'], str(task.id))
self.assertEqual(response.json['name'], task.name)
self.assertEqual(response.json['task'], task.task)
self.assertEqual(response.json['description'], 'New description')
self.assertIsNotNone(response.json['crontab'])
self.assertIsNone(response.json['interval'])
def test_delete_job(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
self.login(AdminFactory())
response = self.delete(url_for('api.job', id=task.id))
self.assert204(response)
self.assertIsNone(PeriodicTask.objects(id=task.id).first())
def test_scheduled_jobs_list(self):
@job('a-job')
def test_job():
pass
for i in range(6):
params = {
'name': faker.name(),
'description': faker.sentence(),
'task': 'a-job'
}
if i % 2:
params['crontab'] = PeriodicTask.Crontab(minute=str(i))
else:
params['interval'] = PeriodicTask.Interval(every=i,
period='minutes')
PeriodicTask.objects.create(**params)
response = self.get(url_for('api.jobs'))
self.assert200(response)
def test_unschedule_job(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
cli('job unschedule {0}'.format(JOB_NAME))
assert len(PeriodicTask.objects(task=JOB_NAME)) == 0
def test_unschedule_job(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
cli('job unschedule {0}'.format(JOB_NAME))
assert len(PeriodicTask.objects(task=JOB_NAME)) == 0
def test_unschedule_job_different_parameters(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
result = cli('job unschedule {0} arg'.format(JOB_NAME), check=False)
assert result.exit_code != 0
assert len(PeriodicTask.objects(task=JOB_NAME)) == 1
def test_unschedule_job_different_parameters(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
result = cli('job unschedule {0} arg'.format(JOB_NAME),
check=False)
assert result.exit_code != 0
assert len(PeriodicTask.objects(task=JOB_NAME)) == 1
def test_unschedule_job_with_parameters(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
args=['arg'],
kwargs={'key': 'value'},
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
cli('job unschedule {0} arg key=value'.format(JOB_NAME))
assert len(PeriodicTask.objects(task=JOB_NAME)) == 0
def test_unschedule_job_with_parameters(self, cli):
PeriodicTask.objects.create(
task=JOB_NAME,
name='job',
description='I\'m a scheduled job',
enabled=True,
args=['arg'],
kwargs={'key': 'value'},
crontab=PeriodicTask.Crontab.parse('0 0 0 0 0'),
)
cli('job unschedule {0} arg key=value'.format(JOB_NAME))
assert len(PeriodicTask.objects(task=JOB_NAME)) == 0
def test_delete_job_need_admin(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
self.login()
response = self.delete(url_for('api.job', id=task.id))
self.assert403(response)
def test_schedule_job(self, cli):
cli('job schedule "0 1 2 3 sunday" fake-job')
tasks = PeriodicTask.objects(task=JOB_NAME)
assert len(tasks) == 1
task = tasks[0]
assert task.args == []
assert task.kwargs == {}
assert task.crontab.minute == '0'
assert task.crontab.hour == '1'
assert task.crontab.day_of_month == '2'
assert task.crontab.month_of_year == '3'
assert task.crontab.day_of_week == 'sunday'
assert task.enabled
assert task.name == 'Job {0}'.format(JOB_NAME)
def test_schedule_job(self, cli):
cli('job schedule "0 1 2 3 sunday" fake-job')
tasks = PeriodicTask.objects(task=JOB_NAME)
assert len(tasks) == 1
task = tasks[0]
assert task.args == []
assert task.kwargs == {}
assert task.crontab.minute == '0'
assert task.crontab.hour == '1'
assert task.crontab.day_of_month == '2'
assert task.crontab.month_of_year == '3'
assert task.crontab.day_of_week == 'sunday'
assert task.enabled
assert task.name == 'Job {0}'.format(JOB_NAME)
def test_get_job(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
response = self.get(url_for('api.job', id=task.id))
self.assert200(response)
self.assertEqual(response.json['id'], str(task.id))
self.assertEqual(response.json['name'], task.name)
self.assertEqual(response.json['description'], task.description)
self.assertEqual(response.json['task'], task.task)
def test_delete_job(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5')
)
self.login(AdminFactory())
response = self.delete(url_for('api.job', id=task.id))
self.assert204(response)
self.assertIsNone(PeriodicTask.objects(id=task.id).first())
def test_schedule_job_with_parameters(self, cli):
cli('job schedule "0 1 2 3 sunday" fake-job '
'arg0 arg1 key1=value1 key0=value0')
tasks = PeriodicTask.objects(task=JOB_NAME)
assert len(tasks) == 1
task = tasks[0]
assert task.args == ['arg0', 'arg1'] # Arguments order is preserved
assert task.kwargs == {'key0': 'value0', 'key1': 'value1'}
assert task.crontab.minute == '0'
assert task.crontab.hour == '1'
assert task.crontab.day_of_month == '2'
assert task.crontab.month_of_year == '3'
assert task.crontab.day_of_week == 'sunday'
assert task.enabled
# kwargs are sorted
expected = 'Job {0}(arg0, arg1, key0=value0, key1=value1)'
assert task.name == expected.format(JOB_NAME)
def test_schedule_job_with_parameters(self, cli):
cli('job schedule "0 1 2 3 sunday" fake-job '
'arg0 arg1 key1=value1 key0=value0')
tasks = PeriodicTask.objects(task=JOB_NAME)
assert len(tasks) == 1
task = tasks[0]
assert task.args == ['arg0', 'arg1'] # Arguments order is preserved
assert task.kwargs == {'key0': 'value0', 'key1': 'value1'}
assert task.crontab.minute == '0'
assert task.crontab.hour == '1'
assert task.crontab.day_of_month == '2'
assert task.crontab.month_of_year == '3'
assert task.crontab.day_of_week == 'sunday'
assert task.enabled
# kwargs are sorted
expected = 'Job {0}(arg0, arg1, key0=value0, key1=value1)'
assert task.name == expected.format(JOB_NAME)
def test_update_job_need_admin(self):
@job('a-job')
def test_job():
pass
task = PeriodicTask.objects.create(
name=faker.name(),
description=faker.sentence(),
task='a-job',
crontab=PeriodicTask.Crontab(minute='5'))
self.login()
response = self.put(
url_for('api.job', id=task.id), {
'name': task.name,
'description': 'New description',
'task': task.task,
'crontab': task.crontab.to_json()
})
self.assert403(response)