def test_get_tasks_with_no_filters(self):
Task.create_task(self._user.username, 'task-1', 'method')
Task.create_task(self._user.username, 'task-2', 'method')
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), None, None)
self.assertEqual(len(task_list), 5)
def test_get_tasks_with_no_filters(self):
Task.create_task(self._user.username, 'task-1', 'method')
Task.create_task(self._user.username, 'task-2', 'method')
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), None, None)
self.assertEqual(len(task_list), 5)
def test_cancel_task(self):
task_id = Task.create_task(self._user.username, 'task', 'method')
ret = client.cancel_task(_make_request(), task_id)
self.assertIsNone(ret)
t = Task.objects.get(id=task_id)
self.assertEqual(t.state, TASK_STATES['CANCELED'])
def test_resubmit_task(self):
task_id = Task.create_task(self._user.username, 'task', 'method', state=TASK_STATES['TIMEOUT'])
new_id = client.resubmit_task(_make_request(self._user), task_id, force=False)
self.assertTrue(new_id > 0)
self.assertNotEqual(task_id, new_id)
task = Task.objects.get(id=new_id)
self.assertEqual(task.priority, 10)
def test_cancel_task(self):
task_id = Task.create_task(self._user.username, 'task', 'method')
ret = client.cancel_task(_make_request(), task_id)
self.assertIsNone(ret)
t = Task.objects.get(id=task_id)
self.assertEqual(t.state, TASK_STATES['CANCELED'])
def test_resubmit_task_do_not_failed(self):
task_id = Task.create_task(self._user.username,
'task',
'method',
state=TASK_STATES['OPEN'])
with self.assertRaises(Exception):
client.resubmit_task(_make_request(), task_id, force=False)
def test_get_tasks_filter_by_ids(self):
t1 = Task.create_task(self._user.username, 'task-1', 'method')
t2 = Task.create_task(self._user.username, 'task-2', 'method')
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), [t1, t2], None)
self.assertEqual(len(task_list), 2)
self.assertEqual(set([t['id'] for t in task_list]), set([t1, t2]))
def test_get_tasks_filter_by_state(self):
t1 = Task.create_task(self._user.username, 'task-1', 'method', state=TASK_STATES['OPEN'])
t2 = Task.create_task(self._user.username, 'task-2', 'method', state=TASK_STATES['TIMEOUT'])
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), None, [TASK_STATES['OPEN'], TASK_STATES['TIMEOUT']])
self.assertEqual(len(task_list), 2)
self.assertEqual(set([t['id'] for t in task_list]), set([t1, t2]))
def test_get_tasks_filter_by_state(self):
t1 = Task.create_task(self._user.username, 'task-1', 'method', state=TASK_STATES['OPEN'])
t2 = Task.create_task(self._user.username, 'task-2', 'method', state=TASK_STATES['TIMEOUT'])
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), None, [TASK_STATES['OPEN'], TASK_STATES['TIMEOUT']])
self.assertEqual(len(task_list), 2)
self.assertEqual(set([t['id'] for t in task_list]), set([t1, t2]))
def test_get_tasks_filter_by_ids(self):
t1 = Task.create_task(self._user.username, 'task-1', 'method')
t2 = Task.create_task(self._user.username, 'task-2', 'method')
Task.create_task(self._user.username, 'task-3', 'method')
Task.create_task(self._user.username, 'task-4', 'method')
Task.create_task(self._user.username, 'task-5', 'method')
task_list = client.get_tasks(_make_request(), [t1, t2], None)
self.assertEqual(len(task_list), 2)
self.assertEqual(set([t['id'] for t in task_list]), set([t1, t2]))
def test_resubmit_task_force_not_failed(self):
task_id = Task.create_task(self._user.username,
'task',
'method',
state=TASK_STATES['OPEN'])
new_id = client.resubmit_task(_make_request(self._user),
task_id,
force=True)
self.assertTrue(new_id > 0)
self.assertNotEqual(task_id, new_id)
def create_subtask(request, label, method, args, parent_id):
parent_task = Task.objects.get_and_verify(task_id=parent_id, worker=request.worker)
return Task.create_task(
parent_task.owner.username,
label,
method,
args=args,
parent_id=parent_id,
arch_name=parent_task.arch.name,
channel_name=parent_task.channel.name,
)
def create_subtask(request, label, method, args, parent_id):
parent_task = Task.objects.get_and_verify(task_id=parent_id,
worker=request.worker)
return Task.create_task(
parent_task.owner.username,
label,
method,
args=args,
parent_id=parent_id,
arch_name=parent_task.arch.name,
channel_name=parent_task.channel.name,
)
def test_create_task_based_on_task(self):
base_task_id = Task.create_task(self._user.username, 'task', 'method')
clone_task_id = client.create_task(_make_request(self._user), {
'task_id': base_task_id,
})
self.assertTrue(clone_task_id > 0)
self.assertNotEqual(base_task_id, clone_task_id)
task = Task.objects.get(id=clone_task_id)
self.assertEqual(task.resubmitted_by.id, self._user.id)
self.assertEqual(task.resubmitted_from.id, base_task_id)
def create_subtask(request, label, method, args, parent_id):
parent_task = Task.objects.get_and_verify(task_id=parent_id,
worker=request.worker)
# def create_task(cls, owner_name, label, method, args=None, parent_id=None, worker_name=None, arch_name="noarch", channel_name="default", priority=10, weight=1, exclusive=False):
# subtask_id = self.__hub.worker.createSubtask(label, method, args, self.__task_id)
return Task.create_task(parent_task.owner.username,
label,
method,
args=args,
parent_id=parent_id,
arch_name=parent_task.arch.name,
channel_name=parent_task.channel.name
) # priority=priority, weight=weight)
def test_create_task_based_on_task(self):
base_task_id = Task.create_task(self._user.username, 'task', 'method')
clone_task_id = client.create_task(_make_request(self._user), {
'task_id': base_task_id,
})
self.assertTrue(clone_task_id > 0)
self.assertNotEqual(base_task_id, clone_task_id)
task = Task.objects.get(id=clone_task_id)
self.assertEqual(task.resubmitted_by.id, self._user.id)
self.assertEqual(task.resubmitted_from.id, base_task_id)
def test_task_info_flat(self):
task_id = Task.create_task(self._user.username, 'label', 'method')
task_info = client.task_info(_make_request(), task_id, flat=True)
self.assertEqual(task_info['id'], task_id)
def test_resubmit_task_do_not_failed(self):
task_id = Task.create_task(self._user.username, 'task', 'method', state=TASK_STATES['OPEN'])
with self.assertRaises(Exception):
client.resubmit_task(_make_request(), task_id, force=False)
def test_resubmit_task_force_not_failed(self):
task_id = Task.create_task(self._user.username, 'task', 'method', state=TASK_STATES['OPEN'])
new_id = client.resubmit_task(_make_request(self._user), task_id, force=True)
self.assertTrue(new_id > 0)
self.assertNotEqual(task_id, new_id)
def create_subtask(request, label, method, args, parent_id):
parent_task = Task.objects.get_and_verify(task_id=parent_id, worker=request.worker)
# def create_task(cls, owner_name, label, method, args=None, parent_id=None, worker_name=None, arch_name="noarch", channel_name="default", priority=10, weight=1, exclusive=False):
# subtask_id = self.__hub.worker.createSubtask(label, method, args, self.__task_id)
return Task.create_task(parent_task.owner.username, label, method, args=args, parent_id=parent_id, arch_name=parent_task.arch.name, channel_name=parent_task.channel.name)# priority=priority, weight=weight)
def test_task_info_flat(self):
task_id = Task.create_task(self._user.username, 'label', 'method')
task_info = client.task_info(_make_request(), task_id, flat=True)
self.assertEqual(task_info['id'], task_id)