def test_acquire_machine(self, setup_machinery, setup_machine_lock):
class mock_machinery(object):
def availables(self):
return True
def acquire(self, machine_id, platform, tags):
class mock_acquire:
name = "mock_mach"
label = "mock_label"
platform = "mock_platform"
return mock_acquire()
class mock_machine:
machine = "mock"
class mock_plat:
platform = "plat"
class mock_tags:
tags = "tags"
setup_machinery(mock_machinery())
mock_task_machine = mock_task()
mock_task_machine.machine = mock_machine()
mock_task_machine.platform = mock_plat()
mock_task_machine.tags = mock_tags()
analysis_man = AnalysisManager(task=mock_task_machine,
error_queue=queue.Queue())
analysis_man.acquire_machine()
assert analysis_man.machine.name == "mock_mach"
def test_acquire_machine(self, setup_machinery, setup_machine_lock):
class mock_machinery:
def availables(self, machine_id, platform, tags, arch):
return True
def acquire(self, machine_id, platform, tags, arch):
class mock_acquire:
name = "mock_mach"
label = "mock_label"
platform = "mock_platform"
arch = "x64"
return mock_acquire()
class mock_machine:
machine = "mock"
class mock_plat:
platform = "plat"
class mock_tags:
class mock_tag:
def __init__(self, name):
self.name = name
tags = [mock_tag("tag1"), mock_tag("tag2")]
def __iter__(self):
for tag in self.tags:
yield tag
class mock_arch:
arch = "x64"
setup_machinery(mock_machinery())
mock_task_machine = mock_task()
mock_task_machine.machine = mock_machine()
mock_task_machine.platform = mock_plat()
mock_task_machine.tags = mock_tags()
mock_task_machine.arch = mock_arch()
analysis_man = AnalysisManager(task=mock_task_machine, error_queue=queue.Queue())
analysis_man.acquire_machine()
assert analysis_man.machine.name == "mock_mach"
