def testWait(self):
task = Task(self.wq)
self.assertEqual(task.is_completed(), True)
job1 = Thread(1, object, 'foo1', None)
job2 = Thread(2, object, 'foo2', None)
task.add_job_id(job1.id)
task.add_job_id(job2.id)
self.assertEqual(task.is_completed(), False)
self.wq.job_succeeded_event(job1)
self.assertEqual(task.is_completed(), False)
self.wq.job_succeeded_event(job2)
self.assertEqual(task.is_completed(), True)
def testWait(self):
task = Task(self.wq)
self.assertEqual(task.is_completed(), True)
job1 = Thread(1, object, 'foo1', None)
job2 = Thread(2, object, 'foo2', None)
task.add_job_id(job1.id)
task.add_job_id(job2.id)
self.assertEqual(task.is_completed(), False)
self.wq.job_succeeded_event(job1)
self.assertEqual(task.is_completed(), False)
self.wq.job_succeeded_event(job2)
self.assertEqual(task.is_completed(), True)
def _run(self, hosts, callback, queue_function, *args):
hosts = to_hosts(hosts, default_domain=self.domain)
self.total += len(hosts)
callback = _prepare_connection(callback)
task = Task(self.workqueue)
for host in hosts:
name = host.get_name()
data = {"host": host}
job_id = queue_function(callback, name, *args, data=data)
if job_id is not None:
task.add_job_id(job_id)
if task.is_completed():
self._dbg(2, "No jobs enqueued.")
return None
self._dbg(2, "All jobs enqueued.")
return task
def _run(self, hosts, callback, queue_function, *args):
hosts = to_hosts(hosts, default_domain=self.domain)
self.total += len(hosts)
callback = _prepare_connection(callback)
task = Task(self.workqueue)
for host in hosts:
name = host.get_name()
data = {'host': host}
job_id = queue_function(callback, name, *args, data=data)
if job_id is not None:
task.add_job_id(job_id)
if task.is_completed():
self._dbg(2, 'No jobs enqueued.')
return None
self._dbg(2, 'All jobs enqueued.')
return task
