class TestJobCollection(TestCase):
@setup
def setup_collection(self):
self.collection = JobCollection()
def test_load_from_config(self):
autospec_method(self.collection.jobs.filter_by_name)
autospec_method(self.collection.add)
factory = mock.create_autospec(JobSchedulerFactory)
job_configs = {'a': mock.Mock(), 'b': mock.Mock()}
result = self.collection.load_from_config(job_configs, factory, True)
result = list(result)
self.collection.jobs.filter_by_name.assert_called_with(job_configs)
expected_calls = [mock.call(v) for v in job_configs.values()]
assert_mock_calls(expected_calls, factory.build.mock_calls)
assert_length(self.collection.add.mock_calls, len(job_configs) * 2)
assert_length(result, len(job_configs))
job_schedulers = [
call[1][0] for call in self.collection.add.mock_calls[::2]
]
for job_scheduler in job_schedulers:
job_scheduler.schedule.assert_called_with()
job_scheduler.get_job.assert_called_with()
def test_move_running_job(self):
with mock.patch(
'tron.core.job_collection.JobCollection.get_by_name',
autospec=None
) as mock_scheduler:
mock_scheduler.return_value.get_job.return_value.status = Job.STATUS_RUNNING
result = self.collection.move('old.test', 'new.test')
assert 'Job is still running.' in result
def test_move(self):
with mock.patch(
'tron.core.job_collection.JobCollection.get_by_name',
autospec=None
) as mock_scheduler:
mock_scheduler.return_value.get_job.return_value.status = Job.STATUS_ENABLED
mock_scheduler.get_name.return_value = 'old.test'
self.collection.add(mock_scheduler)
result = self.collection.move('old.test', 'new.test')
assert 'succeeded' in result
def test_update(self):
mock_scheduler = mock.create_autospec(JobScheduler)
existing_scheduler = mock.create_autospec(JobScheduler)
autospec_method(
self.collection.get_by_name,
return_value=existing_scheduler,
)
assert self.collection.update(mock_scheduler)
self.collection.get_by_name.assert_called_with(
mock_scheduler.get_name(),
)
existing_scheduler.update_from_job_scheduler.assert_called_with(
mock_scheduler
)
existing_scheduler.schedule_reconfigured.assert_called_with()
class TestJobCollection(TestCase):
@setup
def setup_collection(self):
self.collection = JobCollection()
def test_load_from_config(self):
autospec_method(self.collection.jobs.filter_by_name)
autospec_method(self.collection.add)
factory = mock.create_autospec(JobSchedulerFactory)
job_configs = {'a': mock.Mock(), 'b': mock.Mock()}
result = self.collection.load_from_config(job_configs, factory, True)
result = list(result)
self.collection.jobs.filter_by_name.assert_called_with(job_configs)
expected_calls = [mock.call(v) for v in job_configs.values()]
assert_mock_calls(expected_calls, factory.build.mock_calls)
assert_length(self.collection.add.mock_calls, len(job_configs) * 2)
assert_length(result, len(job_configs))
job_schedulers = [
call[1][0] for call in self.collection.add.mock_calls[::2]
]
for job_scheduler in job_schedulers:
job_scheduler.schedule.assert_called_with()
job_scheduler.get_job.assert_called_with()
def test_move_running_job(self):
with mock.patch('tron.core.job_collection.JobCollection.get_by_name',
autospec=None) as mock_scheduler:
mock_scheduler.return_value.get_job.return_value.status = Job.STATUS_RUNNING
result = self.collection.move('old.test', 'new.test')
assert 'Job is still running.' in result
def test_move(self):
with mock.patch('tron.core.job_collection.JobCollection.get_by_name',
autospec=None) as mock_scheduler:
mock_scheduler.return_value.get_job.return_value.status = Job.STATUS_ENABLED
mock_scheduler.get_name.return_value = 'old.test'
self.collection.add(mock_scheduler)
result = self.collection.move('old.test', 'new.test')
assert 'succeeded' in result
def test_update(self):
mock_scheduler = mock.create_autospec(JobScheduler)
existing_scheduler = mock.create_autospec(JobScheduler)
autospec_method(
self.collection.get_by_name,
return_value=existing_scheduler,
)
assert self.collection.update(mock_scheduler)
self.collection.get_by_name.assert_called_with(
mock_scheduler.get_name(), )
existing_scheduler.update_from_job_scheduler.assert_called_with(
mock_scheduler)
existing_scheduler.schedule_reconfigured.assert_called_with()
class TestJobCollection(TestCase):
@setup
def setup_collection(self):
self.collection = JobCollection()
def test_load_from_config(self):
autospec_method(self.collection.jobs.filter_by_name)
autospec_method(self.collection.add)
factory = mock.create_autospec(JobSchedulerFactory)
job_configs = {'a': mock.Mock(), 'b': mock.Mock()}
result = self.collection.load_from_config(job_configs, factory, True)
result = list(result)
self.collection.jobs.filter_by_name.assert_called_with(job_configs)
expected_calls = [mock.call(v) for v in six.itervalues(job_configs)]
assert_mock_calls(expected_calls, factory.build.mock_calls)
assert_length(self.collection.add.mock_calls, len(job_configs) * 2)
assert_length(result, len(job_configs))
job_schedulers = [
call[1][0] for call in self.collection.add.mock_calls[::2]
]
for job_scheduler in job_schedulers:
job_scheduler.schedule.assert_called_with()
job_scheduler.get_job.assert_called_with()
def test_update(self):
mock_scheduler = mock.create_autospec(JobScheduler)
existing_scheduler = mock.create_autospec(JobScheduler)
autospec_method(
self.collection.get_by_name,
return_value=existing_scheduler,
)
assert self.collection.update(mock_scheduler)
self.collection.get_by_name.assert_called_with(
mock_scheduler.get_name(),
)
existing_scheduler.get_job().update_from_job.assert_called_with(
mock_scheduler.get_job.return_value,
)
existing_scheduler.schedule_reconfigured.assert_called_with()