def test_start_good_queued_call_collection(self):
request_1 = call.CallRequest(dummy_call)
request_2 = call.CallRequest(dummy_call)
request_1.group_id = request_2.group_id = 'my-group'
queued_request_1 = QueuedCall(request_1)
queued_request_2 = QueuedCall(request_2)
self.queued_call_collection.insert(queued_request_1)
self.queued_call_collection.insert(queued_request_2)
self.coordinator.start()
self.assertEqual(
self.coordinator.execute_call_asynchronously.call_count, 0)
self.assertEqual(self.coordinator.execute_multiple_calls.call_count, 1)
def test_start_good_queued_call(self):
request = call.CallRequest(dummy_call)
queued_request = QueuedCall(request)
self.queued_call_collection.insert(queued_request)
self.coordinator.start()
self.assertEqual(
self.coordinator.execute_call_asynchronously.call_count, 1)
self.assertEqual(self.coordinator.execute_multiple_calls.call_count, 0)
def enqueue(self, task):
"""
Enqueue (i.e. add) a task to the task queue.
@param task: task to be run
@type task: pulp.server.dispatch.task.Task
"""
self.__lock.acquire()
try:
queued_call = QueuedCall(task.call_request)
task.queued_call_id = queued_call['_id']
self.queued_call_collection.save(queued_call, safe=True)
task.complete_callback = self._complete
self._validate_call_request_dependencies(task)
self.__waiting_tasks.append(task)
task.call_life_cycle_callbacks(
dispatch_constants.CALL_ENQUEUE_LIFE_CYCLE_CALLBACK)
self.__condition.notify()
finally:
self.__lock.release()
