def test_apply_async_task_status(self, apply_async):
"""
Assert that apply_async() creates new task status.
"""
args = [1, 'b', 'iii']
kwargs = {'1': 'for the money', 'tags': ['test_tags'], 'queue': RESERVED_WORKER_1}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['queue'], RESERVED_WORKER_1)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
self.assertEqual(new_task_status['error'], None)
self.assertEqual(new_task_status['spawned_tasks'], [])
self.assertEqual(new_task_status['progress_report'], {})
self.assertEqual(new_task_status['task_type'], 'pulp.server.async.tasks.Task')
self.assertEqual(new_task_status['start_time'], None)
self.assertEqual(new_task_status['finish_time'], None)
self.assertEqual(new_task_status['result'], None)
def test_apply_async_task_status(self, apply_async):
"""
Assert that apply_async() creates new task status.
"""
args = [1, 'b', 'iii']
kwargs = {
'1': 'for the money',
'tags': ['test_tags'],
'queue': WORKER_1_QUEUE
}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['queue'], WORKER_1_QUEUE)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
self.assertEqual(new_task_status['error'], None)
self.assertEqual(new_task_status['spawned_tasks'], [])
self.assertEqual(new_task_status['progress_report'], {})
self.assertEqual(new_task_status['task_type'],
'pulp.server.async.tasks.Task')
self.assertEqual(new_task_status['start_time'], None)
self.assertEqual(new_task_status['finish_time'], None)
self.assertEqual(new_task_status['result'], None)
def test_delete_task_status(self):
"""
Test delete_task_status() under normal circumstances.
"""
task_id = self.get_random_uuid()
TaskStatusManager.create_task_status(task_id, 'a_queue')
TaskStatusManager.delete_task_status(task_id)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(0, len(task_statuses))
def test_delete_task_status(self):
"""
Test delete_task_status() under normal circumstances.
"""
task_id = self.get_random_uuid()
TaskStatusManager.create_task_status(task_id, 'a_queue')
TaskStatusManager.delete_task_status(task_id)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(0, len(task_statuses))
def test_saves_passed_in_routing_key_as_worker_name(self, apply_async):
args = [1, 'b', 'iii']
kwargs = {'a': 'for the money', 'tags': ['test_tags'], 'routing_key': 'othername'}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['worker_name'], 'othername')
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_saves_default_routing_key_as_worker_name(self, apply_async):
args = [1, 'b', 'iii']
kwargs = {'a': 'for the money', 'tags': ['test_tags']}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['worker_name'],
defaults.NAMESPACES['CELERY']['DEFAULT_ROUTING_KEY'].default)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_saves_default_routing_key_as_worker_name(self, apply_async):
args = [1, 'b', 'iii']
kwargs = {'a': 'for the money', 'tags': ['test_tags']}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(
new_task_status['worker_name'],
defaults.NAMESPACES['CELERY']['DEFAULT_ROUTING_KEY'].default)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_apply_async_task_status(self, apply_async):
"""
Assert that apply_async() creates new task status.
"""
args = [1, 'b', 'iii']
kwargs = {'1': 'for the money', 'tags': ['test_tags'], 'queue': RESERVED_WORKER_1}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['queue'], RESERVED_WORKER_1)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_saves_passed_in_routing_key_as_worker_name(self, apply_async):
args = [1, 'b', 'iii']
kwargs = {
'a': 'for the money',
'tags': ['test_tags'],
'routing_key': 'othername'
}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['worker_name'], 'othername')
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_apply_async_task_status_default_queue(self, apply_async):
"""
Assert that apply_async() creates new task status when we do not pass the queue kwarg. It
default to the default Celery queue.
"""
args = [1, 'b', 'iii']
kwargs = {'1': 'for the money', 'tags': ['test_tags']}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['queue'],
defaults.NAMESPACES['CELERY']['DEFAULT_QUEUE'].default)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_apply_async_task_status_default_queue(self, apply_async):
"""
Assert that apply_async() creates new task status when we do not pass the queue kwarg. It
default to the default Celery queue.
"""
args = [1, 'b', 'iii']
kwargs = {'1': 'for the money', 'tags': ['test_tags']}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(
new_task_status['queue'],
defaults.NAMESPACES['CELERY']['DEFAULT_QUEUE'].default)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
def test_creates_task_status(self, apply_async):
args = [1, 'b', 'iii']
kwargs = {'a': 'for the money', 'tags': ['test_tags'], 'routing_key': WORKER_1}
apply_async.return_value = celery.result.AsyncResult('test_task_id')
task = tasks.Task()
task.apply_async(*args, **kwargs)
task_statuses = list(TaskStatusManager.find_all())
self.assertEqual(len(task_statuses), 1)
new_task_status = task_statuses[0]
self.assertEqual(new_task_status['task_id'], 'test_task_id')
self.assertEqual(new_task_status['worker_name'], WORKER_1)
self.assertEqual(new_task_status['tags'], kwargs['tags'])
self.assertEqual(new_task_status['state'], 'waiting')
self.assertEqual(new_task_status['error'], None)
self.assertEqual(new_task_status['spawned_tasks'], [])
self.assertEqual(new_task_status['progress_report'], {})
self.assertEqual(new_task_status['task_type'], 'pulp.server.async.tasks.Task')
self.assertEqual(new_task_status['start_time'], None)
self.assertEqual(new_task_status['finish_time'], None)
self.assertEqual(new_task_status['result'], None)
