def test_init_worker_wo_callback(self):
worker = Worker(
opts=self.opts,
events=self.events,
worker_num=self.worker_num,
group_id=self.group_id,
)
assert worker.callback
def test_init_worker_wo_callback(self):
worker = Worker(
opts=self.opts,
frontend_client=self.frontend_client,
worker_num=self.worker_num,
group_id=self.group_id,
)
worker.vmm = MagicMock()
def init_worker(self):
self.worker = Worker(
opts=self.opts,
events=self.events,
worker_num=self.worker_num,
group_id=self.group_id,
callback=self.worker_callback,
)
def set_ip(*args, **kwargs):
self.worker.vm_ip = self.vm_ip
def erase_ip(*args, **kwargs):
self.worker.vm_ip = None
self.set_ip = set_ip
def init_worker(self):
self.worker = Worker(
opts=self.opts,
frontend_client=self.frontend_client,
worker_num=self.worker_num,
group_id=self.group_id,
)
self.worker.vmm = MagicMock()
def set_ip(*args, **kwargs):
self.worker.vm_ip = self.vm_ip
def erase_ip(*args, **kwargs):
self.worker.vm_ip = None
self.set_ip = set_ip
self.erase_ip = erase_ip
